File Content: DistUpgradeMain.py # 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 Submit