REDROOM

Navigation

Breadcrumbs

/ / lib / python3 / dist-packages / DistUpgrade

Create

Create File | Create Folder | Upload Files | Run Command

Edit File Size: 9.14 KB

//lib/python3/dist-packages/DistUpgrade/DistUpgradeMain.py

Text Base64

# DistUpgradeMain.py 
#  
#  Copyright (c) 2004-2008 Canonical
#  
#  Author: Michael Vogt <michael.vogt@ubuntu.com>
# 
#  This program is free software; you can redistribute it and/or 
#  modify it under the terms of the GNU General Public License as 
#  published by the Free Software Foundation; either version 2 of the
#  License, or (at your option) any later version.
# 
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
# 
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
#  USA

import apt
import atexit
import gettext
import glob
import logging
import os
import shutil
import subprocess
import sys

from datetime import datetime
from optparse import OptionParser
from gettext import gettext as _

# dirs that the packages will touch, this is needed for the coherence check
# before the upgrade
SYSTEM_DIRS = ["/bin",
              "/boot",
              "/etc",
              "/initrd",
              "/lib",
              "/lib32", # ???
              "/lib64", # ???
              "/sbin",
              "/usr",
              "/var",
              ]

from .DistUpgradeConfigParser import DistUpgradeConfig

def do_commandline():
    " setup option parser and parse the commandline "
    parser = OptionParser()
    parser.add_option("--have-prerequists", dest="havePrerequists",
                      action="store_true", default=False)
    parser.add_option("--with-network", dest="withNetwork",action="store_true")
    parser.add_option("--without-network", dest="withNetwork",action="store_false")
    parser.add_option("--frontend", dest="frontend",default=None,
                      help=_("Use frontend. Currently available: \n"\
                             "DistUpgradeViewText, DistUpgradeViewGtk, DistUpgradeViewKDE"))
    parser.add_option("--mode", dest="mode",default="desktop",
                      help=_("*DEPRECATED* this option will be ignored"))
    parser.add_option("--partial", dest="partial", default=False,
                      action="store_true", 
                      help=_("Perform a partial upgrade only (no sources.list rewriting)"))
    parser.add_option("--disable-gnu-screen", action="store_true", 
                      default=False,
                      help=_("Disable GNU screen support"))
    parser.add_option("--datadir", dest="datadir", default=".",
                      help=_("Set datadir"))
    parser.add_option("--devel-release", action="store_true",
                      dest="devel_release", default=False,
                      help=_("Upgrade to the development release"))
    return parser.parse_args()

def setup_logging(options, config):
    " setup the logging "
    logdir = config.getWithDefault("Files","LogDir","/var/log/dist-upgrade/")
    if not os.path.exists(logdir):
        os.mkdir(logdir)
    # check if logs exists and move logs into place
    if glob.glob(logdir+"/*.log"):
        now = datetime.now()
        backup_dir = logdir+"/%04i%02i%02i-%02i%02i" % (now.year,now.month,now.day,now.hour,now.minute)
        if not os.path.exists(backup_dir):
            os.mkdir(backup_dir)
        for f in glob.glob(logdir+"/*.log"):
            shutil.move(f, os.path.join(backup_dir,os.path.basename(f)))
    fname = os.path.join(logdir,"main.log")
    # do not overwrite the default main.log
    if options.partial:
        fname += ".partial"
    with open(fname, "a"):
        pass
    logging.basicConfig(level=logging.DEBUG,
                        filename=fname,
                        format='%(asctime)s %(levelname)s %(message)s',
                        filemode='w')
    # log what config files are in use here to detect user
    # changes
    logging.info("Using config files '%s'" % config.config_files)
    logging.info("uname information: '%s'" % " ".join(os.uname()))
    cache = apt.apt_pkg.Cache(None)
    apt_version = cache['apt'].current_ver.ver_str
    logging.info("apt version: '%s'" % apt_version)
    logging.info("python version: '%s'" % sys.version)
    return logdir

def save_system_state(logdir):
    # save package state to be able to re-create failures
    try:
        from .apt_clone import AptClone
    except ImportError:
        logging.error("failed to import AptClone")
        return
    target = os.path.join(logdir, "apt-clone_system_state.tar.gz")
    logging.debug("creating statefile: '%s'" % target)
    # this file may contain sensitive data so ensure we create with the
    # right umask
    old_umask = os.umask(0o0066)
    clone = AptClone()
    clone.save_state(sourcedir="/", target=target, with_dpkg_status=True,
        scrub_sources=True)
    # reset umask
    os.umask(old_umask)
    # lspci output
    try:
        s=subprocess.Popen(["lspci","-nn"], stdout=subprocess.PIPE,
                           universal_newlines=True).communicate()[0]
        with open(os.path.join(logdir, "lspci.txt"), "w") as f:
            f.write(s)
    except OSError as e:
        logging.debug("lspci failed: %s" % e)
    
def setup_view(options, config, logdir):
    " setup view based on the config and commandline "

# the commandline overwrites the configfile
    for requested_view in [options.frontend]+config.getlist("View","View"):
        if not requested_view:
            continue
        try:
            # this should work with py3 and py2.7
            from importlib import import_module
            # use relative imports
            view_modul = import_module("."+requested_view, "DistUpgrade")
            # won't work with py3
            #view_modul = __import__(requested_view, globals())
            view_class = getattr(view_modul, requested_view)
            instance = view_class(logdir=logdir, datadir=options.datadir)
            break
        except Exception as e:
            logging.warning("can't import view '%s' (%s)" % (requested_view,e))
            print("can't load %s (%s)" % (requested_view, e))
    else:
        logging.error("No view can be imported, aborting")
        print("No view can be imported, aborting")
        sys.exit(1)
    return instance

def run_new_gnu_screen_window_or_reattach():
    """ check if there is a upgrade already running inside gnu screen,
        if so, reattach
        if not, create new screen window
    """
    SCREENNAME = "ubuntu-release-upgrade-screen-window"
    # get the active screen sockets
    try:
        out = subprocess.Popen(
            ["screen","-ls"], stdout=subprocess.PIPE,
            universal_newlines=True).communicate()[0]
        logging.debug("screen returned: '%s'" % out)
    except OSError:
        logging.info("screen could not be run")
        return
    # check if a release upgrade is among them
    if SCREENNAME in out:
        logging.info("found active screen session, re-attaching")
        # if we have it, attach to it
        os.execv("/usr/bin/screen",  ["screen", "-d", "-r", "-p", SCREENNAME])
    # otherwise re-exec inside screen with (-L) for logging enabled
    os.environ["RELEASE_UPGRADER_NO_SCREEN"]="1"
    # unset escape key to avoid confusing people who are not used to
    # screen. people who already run screen will not be affected by this
    # unset escape key with -e, enable log with -L, set name with -S
    cmd = ["screen", 
           "-e", "\\0\\0",
           "-c", "screenrc",
           "-S", SCREENNAME]+sys.argv
    logging.info("re-exec inside screen: '%s'" % cmd)
    os.execv("/usr/bin/screen", cmd)

def main():
    """ main method """

# commandline setup and config
    (options, args) = do_commandline()
    config = DistUpgradeConfig(options.datadir)
    logdir = setup_logging(options, config)

from .DistUpgradeVersion import VERSION
    logging.info("release-upgrader version '%s' started" % VERSION)
    # ensure that DistUpgradeView translations are displayed
    gettext.textdomain("ubuntu-release-upgrader")
    if options.datadir is None or options.datadir == '.':
        localedir = os.path.join(os.getcwd(), "mo")
        gettext.bindtextdomain("ubuntu-release-upgrader", localedir)

# create view and app objects
    view = setup_view(options, config, logdir)

# gnu screen support
    if (view.needs_screen and
        not "RELEASE_UPGRADER_NO_SCREEN" in os.environ and
        not options.disable_gnu_screen):
        run_new_gnu_screen_window_or_reattach()

# A reboot is required at the end of the upgrade,
    # so there is no need to run needrestart during upgrade.
    if not os.getenv('NEEDRESTART_SUSPEND'):
        os.environ['NEEDRESTART_SUSPEND'] = 'y'

from .DistUpgradeController import DistUpgradeController
    app = DistUpgradeController(view, options, datadir=options.datadir)
    atexit.register(app._enableAptCronJob)

# partial upgrade only
    if options.partial:
        if not app.doPartialUpgrade():
            sys.exit(1)
        sys.exit(0)

# save system state (only if not doing just a partial upgrade)
    save_system_state(logdir)

# full upgrade, return error code for success/failure
    if app.run():
        return 0
    return 1

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Directory Contents

Dirs: 1 × Files: 33

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-02-07 06:05:46	Edit Download recursive recursive
apt_btrfs_snapshot.py	9.68 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
apt_clone.py	33.75 KB	lrw-r--r--	2024-03-15 15:46:01	Edit Download recursive recursive
dist-upgrade.py	129 B	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
distro.py	23.80 KB	lrw-r--r--	2025-07-25 15:18:03	Edit Download recursive recursive
DistUpgradeApport.py	4.94 KB	lrw-r--r--	2026-01-19 14:20:45	Edit Download recursive recursive
DistUpgradeCache.py	53.81 KB	lrw-r--r--	2026-01-19 14:20:45	Edit Download recursive recursive
DistUpgradeConfigParser.py	3.76 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeController.py	125.34 KB	lrw-r--r--	2026-01-19 14:20:45	Edit Download recursive recursive
DistUpgradeFetcher.py	5.87 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeFetcherCore.py	12.45 KB	lrw-r--r--	2026-01-19 14:20:45	Edit Download recursive recursive
DistUpgradeFetcherKDE.py	9.74 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeFetcherSelf.py	1.91 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeGettext.py	2.98 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeMain.py	9.14 KB	lrw-r--r--	2026-01-19 14:20:45	Edit Download recursive recursive
DistUpgradePatcher.py	3.92 KB	lrw-r--r--	2026-01-19 14:20:45	Edit Download recursive recursive
DistUpgradeQuirks.py	87.91 KB	lrw-r--r--	2026-01-19 14:22:43	Edit Download recursive recursive
DistUpgradeVersion.py	21 B	lrw-r--r--	2026-01-19 15:31:41	Edit Download recursive recursive
DistUpgradeView.py	16.94 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeViewGtk3.py	34.16 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeViewKDE.py	41.19 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
DistUpgradeViewNonInteractive.py	13.54 KB	lrw-r--r--	2026-01-19 14:20:45	Edit Download recursive recursive
DistUpgradeViewText.py	12.53 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
GtkProgress.py	3.93 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
MetaRelease.py	16.99 KB	lrw-r--r--	2025-04-25 13:16:13	Edit Download recursive recursive
QUrlOpener.py	3.24 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
ReleaseNotesViewer.py	7.48 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
ReleaseNotesViewerWebkit.py	2.91 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
SimpleGtk3builderApp.py	2.01 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
SimpleGtkbuilderApp.py	1.99 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
telemetry.py	3.40 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
utils.py	17.37 KB	lrw-r--r--	2025-04-25 13:16:13	Edit Download recursive recursive
xorg_fix_proprietary.py	3.99 KB	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive
__init__.py	0 B	lrw-r--r--	2024-10-10 14:56:28	Edit Download recursive recursive