Current File : //usr/lib/python3/dist-packages/DistUpgrade/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
Mini Shell