U
    ٪zc                    @   s  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlmZ d dlmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZ d dlmZ d dlm Z  dd	l!m"Z" dd
l#m$Z$ ddl%m&Z& ddl'm'Z' ddl(m)Z) d dl*m+Z+ d dl*m,Z, e+j-e,_-d dl.m/Z/m0Z0 ddl1m2Z2m3Z3 ddl4m5Z6 ddl4m7Z7 d dl5Z5ddl#m8Z8m9Z9m:Z: ddl;m<Z< dZ=dd Z>G dd de?Z@G dd deAZBeCdkrddlDmDZD ejEejFd eD ZGeBeGZHeHI  eHJ  dS )     N)NoOptionError)ConfigParser   )get)	country_mirrorurl_downloadablecheck_and_fix_xbitget_archiptables_activeinside_chroot)get_string_with_no_auth_from_source_entryis_child_of_process_nameinhibit_sleep)Template)urlsplit)Step)MyCache)DistUpgradeConfig)DistUpgradeQuirks)AptCdrom)distinfo)sourceslist)SourcesList	is_mirror)
get_distroNoDistroTemplateException)gettext)ngettext)CacheExceptionDpkgInterruptedCacheExceptionLockingFailedNotEnoughFreeSpaceError)
run_apportz/var/run/reboot-requiredc                 C   s>   ddddg}z| | W S  tk
r8   t|d  Y S X dS )zB key() function for sorted to ensure "correct" component ordering main
restrictedZuniverseZ
multiverser   N)index
ValueErrorlen)aZordering r(   C/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeController.pycomponent_ordering_keyO   s
    r*   c                   @   s   e Zd ZdS )NoBackportsFoundExceptionN)__name__
__module____qualname__r(   r(   r(   r)   r+   Z   s   r+   c                   @   s  e Zd ZdZd_ddZd`ddZd	d
 Zdd Zdd Zdd Z	dd Z
dd Zdd ZdaddZdd Zdd Zdd Zdbdd Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Z d?d@ Z!dAdB Z"dCdD Z#dEdF Z$dGdH Z%dIdJ Z&dKdL Z'dcdMdNZ(dddOdPZ)dQdR Z*dSdT Z+dUdV Z,dWdX Z-dYdZ Z.d[d\ Z/d]d^ Z0dS )eDistUpgradeControllerz3 this is the controller that does most of the work Nc                 C   s  d}|d ks|dkr*t  }t j|d}|| _|| _td| td t	
d|jj  || _| jtd d | _d | _| jr| jjd krd| _n
| jj| _|r|j}nd }t||| _t|| _d| jdd	 | _| jd
 | jd
dt| j | jd
dd | jr2| jjr2| jd
dd | jdd| _| jdd| _ | jdd| _!t" | _#dt j$krt	
d t%jdd d| _&t'| | j| _(t) | _*dt j$d< dt j$d< dt  d t j$d f t j$d< t+d | j,dd }t%jd!t| t%jd"td# t%jd$td% t%jd&d' t%jd(d' t%jd)d* | j-d+d,| _.| j/dd-| _0g | _1| j2d.rd/d0 | j3d.D | _1d1| _4d2| _5i | _6d S )3Nz/usr/share/locale/.Zmozubuntu-release-upgraderzUsing '%s' viewzReading cacheTZFilesZ	BackupExtOptionswithNetworkZ
devReleaseFalseTrueSourcesZFromZToZValidOriginZ"RELEASE_UPGRADE_NO_FORCE_OVERWRITEzenable dpkg --force-overwritezDPkg::Options::z--force-overwriteF1ZRELEASE_UPGRADE_IN_PROGRESSZPYCENTRAL_FORCE_OVERWRITEz%s:%sz	/importedPATHz./imported/invoke-rc.dNetwork
MaxRetrieszAcquire::RetrieszDpkg::MaxArgsi   zDpkg::MaxArgBytesi   zAcquire::http::TimeoutZ20zAcquire::ftp::TimeoutzApt::Get::List-CleanupfalseDistroForcedObsoletesZValidMirrorsZThirdPartyMirrorsc                 S   s   g | ]}|d  qS )r   r(   ).0Zpairr(   r(   r)   
<listcomp>   s     z2DistUpgradeController.__init__.<locals>.<listcomp>i   )7osgetcwdpathjoindatadiroptionsr   bindtextdomain
textdomainloggingdebug	__class__r,   _viewupdateStatus_cachefetcherr2   
useNetworkZ	cdromPathr   aptcdromr   configr   sources_backup_extZadd_sectionsetstrdevel_releasefromDisttoDistoriginr	   archenvironapt_pkg_partialUpgrader   quirksr   Zinhibitor_fdr   getintgetlistforced_obsoletesZgetListFromFilevalid_mirrorsvalid_3p_mirrorshas_sectionitems_aptCronJobPerms_uid	_user_env)selfZdistUpgradeViewrE   rD   Z	localedir	cdrompath
maxRetriesr(   r(   r)   __init__a   sz    







zDistUpgradeController.__init__TFc              
   C   s   t d | jd kr"| jd n| j  | j  d}d}z| |W S  tk
r } zr|d7 }| j	
  td t d|  ||krt d|  | j	td	td
 |r|   n
td W 5 d }~X Y q>X q>d S )NzopenCache()ZPreCacheOpen   r   r   皙?z'failed to lock the cache, retrying (%i)zCache can not be locked (%s)zUnable to get exclusive lockzThis usually means that another package management application (like apt-get or aptitude) already running. Please close that application first.)rH   rI   rN   r^   runrelease_lockunlock_lists_dir
_openCacher   rK   processEventstimesleeperrorrM   abortsysexit)ri   lockrestore_sources_list_on_failZMAX_LOCK_RETRIESZ
lock_retryer(   r(   r)   	openCache   s0    






zDistUpgradeController.openCachec              	   C   s   z*t | j| j| j| j || _| j| _W nn tk
r   t	d dddg}t
jddkrh|d | j | t | j| j| j| j | _Y nX | j| j_tdt| j  d S )	Nz-dpkg interrupted, calling dpkg --configure -a/usr/bin/dpkg--configure-aDEBIAN_FRONTENDnoninteractive--force-confoldz/openCache(), new cache size %i)r   rR   rK   r^   getOpCacheProgressrN   Z	apt_cacher   rH   warningr@   r[   r   appendgetTerminalcallr]   ZpartialUpgraderI   r&   )ri   rz   cmdr(   r(   r)   rr      s*    



z DistUpgradeController._openCachec                 C   s$   | j dd}| jjj|kr dS dS )z
      Returns True if this view support upgrades over ssh.
      In theory all views should support it, but for savety
      we do only allow text ssh upgrades (see LP: #322482)
      ViewZ
SupportSSHTF)rR   r`   rK   rJ   r,   )ri   Z	supportedr(   r(   r)   _viewSupportsSSH  s    z&DistUpgradeController._viewSupportsSSHc                 C   s   t jd}t j|st jdrtdr|  sbtd | j	t
dt
d td dS d	}| j	t
d
t
d| }|dkrtd tddd| dt|g}|dkrt
d}t
d| }t rd| }|t
d| 7 }| j	|| dS )z this will check for server mode and if we run over ssh.
            if this is the case, we will ask and spawn a additional
            daemon (to be sure we have a spare one around in case
            of trouble)
        z"/var/run/release-upgrader-sshd.pidz/procZsshdzupgrade over ssh not allowedz.Upgrading over remote connection not supportedzYou are running the upgrade over a remote ssh connection with a frontend that does not support this. Please try a text mode upgrade with 'do-release-upgrade'.

The upgrade will abort now. Please try without ssh.r   Fi  zContinue running under SSH?a  This session appears to be running under ssh. It is not recommended to perform a upgrade over ssh currently because in case of failure it is harder to recover.

If you continue, an additional ssh daemon will be started at port '%s'.
Do you want to continue?z/usr/sbin/sshdz-oz
PidFile=%sz-pr   zStarting additional sshdzTo make recovery in case of failure easier, an additional sshd will be started on port '%s'. If anything goes wrong with the running ssh you can still connect to the additional one.
z-iptables -I INPUT -p tcp --dport %s -j ACCEPTzIf you run a firewall, you may need to temporarily open this port. As this is potentially dangerous it's not done automatically. You can open the port with e.g.:
'%s'T)r@   rB   rC   existsisdirr   r   rH   rv   rK   rM   rx   ry   askYesNoQuestion
subprocessr   rU   r
   information)ri   ZpidfileZportressummaryZdescrr   r(   r(   r)   	_sshMagic  sT    





  zDistUpgradeController._sshMagicc                 C   s   ddl m} ddlm} d}| jdksD| jdksD| jdksD| jdkrHd	}|d|d
}| j  |jdsv| j  q^|jdkrt	
d dS | j }||j|| j| jd}|  dS )z this is a helper that is run if we are started from a CD
            and we have network - we will then try to fetch a update
            of ourself
        r   )MetaReleaseCore)DistUpgradeFetcherSelfFZdapperZhardyZlucidZpreciseT)ZuseDevelopmentReleaseforceLTSrn   NzNo new dist found)new_distprogressrE   Zview)ZMetaReleaser   r   releaserK   rs   Z
downloadedwaitr   rH   rv   getAcquireProgressrE   ro   )ri   r   r   r   mr   rO   r(   r(   r)   _tryUpdateSelfJ  s6    




z$DistUpgradeController._tryUpdateSelfc           
      C   s   t d dg}|D ]\}}d| }tj|rt }t|}|| W 5 Q R X z|dd}W n* t	k
r   t d||f  Y  dS X zt
d| }W n8 tk
r }	 zt d	|	  W Y  dS d
}	~	X Y nX ||tjd|fkrt d|||tjd|f   dS qdS )z sanity check that /usr/bin/python3 points to the default
            python version. Users tend to modify this symlink, which
            breaks stuff in obscure ways (Ubuntu #75557).
        z_pythonSymlinkCheck run)python3r   z/usr/share/%s/debian_defaultsZDEFAULTzdefault-versionz'no default version for %s found in '%s'Fz/usr/bin/%szos.readlink failed (%s)Nz/usr/binz8%s symlink points to: '%s', but expected is '%s' or '%s'T)rH   rI   r@   rB   r   SafeConfigParseropenZreadfpr   r   readlinkOSErrorrv   rC   )
ri   Zbinaries_and_dirnamesbinarydirnameZdebian_defaultsrR   fZexpected_defaultZfs_default_versionr|   r(   r(   r)   _pythonSymlinkChecki  s4    


z)DistUpgradeController._pythonSymlinkCheckc              
   C   s  t jdddgt jdd d   | _}td|  || jks|| j	kst
d|| jf  | j
td	td
|| j	f  td | jrF| jjrFt d }td|  tdt|d   tj|d rtjd|d  tj|d rtjd|d  |d }tj|rFtd|  ttj| |   |  st
d | j
td	td td z|   W n8 tk
r } zt
d|  W Y dS d}~X Y nX | j | jsdS | j! | _"| j"rdtj#d< n
dtj#d< | $ st
d dS ddl%m&} |D ]X}tj|r.t'|tj(s.t
d |  | j
td!| td"|  | )  q.| j*r| jr| jj+dkr| j,td#td$d%}|| _-| jd&d't.| j- td(|  |r| /  dS ))z: initial cache opening, sanity checking, network checking Zlsb_releasez-cz-sT)stdoutuniversal_newlinesr   zlsb-release: '%s'zBad upgrade: '%s' != '%s' zCan not upgradez=An upgrade from '%s' to '%s' is not supported with this tool.r   z
/backportszusing backports in '%s' zhave: %sz/*.udebr~   zDir::Bin::dpkgz/usr/lib/apt/methodszDir::Bin::methodsz/etc/apt/apt.conf.d/01ubuntuzadding config '%s'z%pythonSymlinkCheck() failed, abortingzMYour python3 install is corrupted. Please fix the '/usr/bin/python3' symlink.zopenCache() failed: '%s'FNZserverZRELEASE_UPGRADE_MODEZdesktopzcheckViewDepends() failed)SYSTEM_DIRSz%s not writablezCan not write to '%s'zIts not possible to write to the system directory '%s' on your system. The upgrade can not continue.
Please make sure that the system directory is writable.z)Include latest updates from the Internet?a  The upgrade system can use the internet to automatically download the latest updates and install them during the upgrade.  If you have a network connection this is highly recommended.

The upgrade will take longer, but when it is complete, your system will be fully up to date.  You can choose not to do this, but you should install the latest updates soon after upgrading.
If you answer 'no' here, the network is not used at all.ZYesr1   r2   z#useNetwork: '%s' (selected by user))0r   PopenPIPEcommunicatestripr   rH   rI   rW   rX   rv   rK   rM   rx   ry   rE   havePrerequistsr@   rA   infoglobrB   r   r\   rR   rT   Zread_config_filer   r   r}   SystemErrorrN   Zsanity_checkneed_server_mode
serverModer[   checkViewDependsZDistUpgradeMainr   accessW_OKrw   rQ   r2   r   rP   rU   r   )ri   r   backportsdirconfr|   r   Z	systemdirr   r(   r(   r)   prepare  s    







zDistUpgradeController.preparec                 C   s>   t d|  | js"t d dS d|j|jf }t|t jS )zf
        helper that checks if a sources.list entry points to 
        something downloadable
        zverifySourcesListEntry: %sz(skipping downloadable check (no network)Tz%s/dists/%s/Release)rH   rI   rP   uridistr   )ri   entryr   r(   r(   r)   _sourcesListEntryDownloadable  s    
z3DistUpgradeController._sourcesListEntryDownloadablec                    sv	  |rt d n
t d  jdd} jdddsBdtjkrPt d d}d}|d	kr jrt  jd
d  jd
dD ]X}t d|  | j	ks j	| j
rt j	| j
jdks j	| j
rt j	| j
jdkr j	| j
jd jdkrt d|  z.t }| j |d d	}t d W n tk
r } zvt d|  t d d}ddg}d}	 jd|	 j||  jd|	 jd ||  jdd jd || W 5 d }~X Y nX  qq jdd}
 jg fdd|
D  } jg fdd|
D  }d _g } jjd d  D ]D}d |jkrTq@|jd!rfq@|jd"rxq@|| q@| j_d}i  _i } jjd d  D ] }|j|krd#||j< |js|jrܐq|jd$r|j  jkrt d%|  d	|_qnX|jd$rjt!j"d&}tj#$d'|t!%|jd(|j d)f st d*|  d	|_qd+ j }|js|jd,r|j |kr j|_ d-g|_&t d.|  q|js|jd/ j rt d0 d	|_q|jsP j'rP j'j(d	krPd1 j |j krPt d2 d1 j |_ | j)t*d3 j 7  _)d	|_q|jsd4|jkrt d5 d6t+  dfD ]Z}	t,,|}|	|_ j|_  -|r|t .d7|	  |	|_|j|krd8||j<  qؐq|t d9t/|   0|j}| p 1|j}|s|rLd	}|jsj|j2d:ksjd;|jksjd< j |j ksjd= j |j ksjd>|jkrnd}|j |krt d?t/|  ||O }n|j |krL||j d#kr||O }t,,|} j|_  -|sd@||j< n
d8||j< ||j d@krd	|_d	 _t dAt/|  n,||O }||3|j  |_ t dBt/|  n6|j2d:kr^qn$|rd	|_d	 _t dCt/|  |j2dkrdD|jkrȈ j4dEks j4dFkrt dG j4  dH|_dI j dJ j d< j fD ]b} j5|t6  |js|j |kr|j2dkr|j&D ]$}||kr0q j| | qqq|j  jkrb j|_ dKt*dL j  }t7|j)t8r| j)|9dM7  _)n| j)|7  _)d	|_d	 _t dNt/|  |r|j |kr||3|j  |_ |j |krqdI j dJ j d< j fD ]Z} j5|t6  |j |kr|j2dkr|j&D ]$}||krXqF j| | qFqt dO|j2|j |j&f  d}	d} j|j2|	|j |j&| q jjd d  D ]}|js|js|j2d:ks|jd$s|j  jk	r q|j  jkr j j  j|j   }|rt .dPt/|  |j&:t;|t<dQ t .dRt/|   j|j = q|S )SNz&rewriteSourcesList() with mirror_checkzrewriteSourcesList()r5   Z
ComponentsZAllowThirdPartyFZ"RELEASE_UPGRADER_ALLOW_THIRD_PARTYz,mirror check skipped, *overriden* via configTr;   BaseMetaPkgszChecking pkg: %sr   r   Znowz)BaseMetaPkg '%s' has no candidate.originsr"   z/get_distro().enable_component("main") succeededz$NoDistroTemplateException raised: %sz]get_distro().enable_component("main") failed, overwriting sources.list instead as last resortz* auto generated by ubuntu-release-upgraderr#    http://archive.ubuntu.com/ubuntudeb-updatesz!http://security.ubuntu.com/ubuntu	-securityZPocketsc                    s   g | ]}d  j |f qS z%s-%s)rW   r=   xri   r(   r)   r>   5  s   z<DistUpgradeController.rewriteSourcesList.<locals>.<listcomp>c                    s   g | ]}d  j |f qS r   )rX   r   r   r(   r)   r>   7  s   z/extras.ubuntu.comzE## This software is not part of Ubuntu, but is offered by third-partyz5## developers who want to ship their latest software.unknownzcdrom:z,disabled '%s' cdrom entry (dist == fromDist)zDir::State::listsz%s/%s%s_%s_%sZdistsZReleasez:disabling cdrom source '%s' because it has no Release filez%s-commercialzhttp://archive.canonical.comZpartnerz transitioned commercial to '%s' z*http://landscape.canonical.com/packages/%sz&commenting landscape.canonical.com outz%s-proposedz2upgrade to development release, disabling proposedz5Not for humans during development stage of release %szold-releases.ubuntu.com/z-upgrade from old-releases.ubuntu.com detectedz"http://%sarchive.ubuntu.com/ubuntuz%transition from old-release.u.c to %sZpassedzexamining: '%s'zdeb-srcz/security.ubuntu.comz%s-securityz%s-backportsz/archive.canonical.comz%entry '%s' is already set to new distZfailedz)entry '%s' was disabled (no Release file)zentry '%s' updated to new distz&entry '%s' was disabled (unknown dist)zports.ubuntu.comZpowerpcZsparcz-moving %s source entry to 'ports.ubuntu.com' z%http://ports.ubuntu.com/ubuntu-ports/%s
%s-updates zdisabled on upgrade to %szUTF-8z(entry '%s' was disabled (unknown mirror)zAdding entry: %s %s %sz)fixing components inconsistency from '%s')keyzto new entry '%s')=rH   rI   rR   r`   getWithDefaultr@   r[   r   rP   rN   	candidater&   originsarchiver   Zget_sourcessourcesZenable_componentr   	exceptionaddrX   rW   sources_disabledlistr   line
startswithr   found_componentsinvaliddisabledr   r\   find_dirrB   r   Zuri_to_filenamecompsrE   rV   commentrM   r   copyr   r   r   isMirrorisThirdPartyMirrortyper$   rZ   
setdefaultrT   
isinstancebytesencodeextendsortedr*   )ri   mirror_checkZsync_componentsZmain_was_missingpkgnamedistror|   r   r   r   ZpocketsZ	fromDistsZtoDistsZnew_listr   ZfoundToDistZentry_uri_test_resultslistdirZcdistZ
test_entryZvalidMirrorZthirdPartyMirrorZvalidTodcompZdisable_commentZcomponent_diffr(   r   r)   rewriteSourcesList  s   





  




















z(DistUpgradeController.rewriteSourcesListc                    s&  t d t jd _ j j t fdd jD sf j	t
dt
d j }|sf    jdds`t d	  j	t
d	t
d
 j jf }|rXt jd _ jdds` jr`t
d}t
d j jf } j	||s   t d d}ddg} jd| j|  jd| jd |  jdd jd | n    j  zt }|  W nv tk
r   t d tjdrΈ jt
dt
d tddg n  jt
dt
d t d Y dS X  jr jt
dt
d t   j dS ) NzupdateSourcesList()ZmatcherPathc                 3   s$   | ]}|j d ko|j jkV  qdS )r   N)r   r   rW   )r=   r|   r   r(   r)   	<genexpr>%  s     z:DistUpgradeController.updateSourcesList.<locals>.<genexpr>z!No valid sources.list entry foundzWhile scanning your repository information no entry about %s could be found.

An upgrade might not succeed.

Do you want to continue anyway?T)r   zNo valid mirror foundaW  While scanning your repository information no mirror entry for the upgrade was found. This can happen if you run an internal mirror or if the mirror information is out of date.

Do you want to rewrite your 'sources.list' file anyway? If you choose 'Yes' here it will update all '%s' to '%s' entries.
If you select 'No' the upgrade will cancel.FzGenerate default sources?zAfter scanning your 'sources.list' no valid entry for '%s' was found.

Should default entries for '%s' be added? If you select 'No', the upgrade will cancel.z"Generated new default sources.listr   r"   r#   r   r   z"http://security.ubuntu.com/ubuntu/r   z<Repository information invalid after updating (we broke it!)/usr/bin/apport-bugzRepository information invalidzlUpgrading the repository information resulted in a invalid file so a bug reporting process is being started.
apport-bugubuntu-release-upgrader-corezUpgrading the repository information resulted in a invalid file. To report a bug install apport and then execute 'apport-bug ubuntu-release-upgrader'..Missing apport-bug, bug report not autocreatedzThird party sources disabledzSome third party entries in your sources.list were disabled. You can re-enable them after the upgrade with the 'software-properties' tool or your package manager.)!rH   rI   r   rD   r   ZbackuprS   anyrK   r   rM   rW   rw   r   rv   rX   rP   r   r   Zsaver\   Z
SourceListread_main_listr   r@   rB   r   r   r   r   r   get_telemetryZset_using_third_party_sources)ri   r   ZprimZseconr   r   r   r(   r   r)   updateSourcesList  s|    




 


z'DistUpgradeController.updateSourcesListc                 C   s   t d g }g }g }g }g }| jD ]r}|jr<||j q$|jrP||j q$|jrd||j q$|jr~|j	r~||j q$|jr$|j
r$||j q$t dd|  t dd|  t dd|  t dd|  t dd|  d S )Nz$About to apply the following changeszKeep at same version: %sr   zUpgradable, but held- back: %sz
Remove: %szInstall: %szUpgrade: %s)rH   rI   rN   Zmarked_installr   nameZmarked_upgradeZmarked_deleteis_installedZis_upgradableZmarked_keeprC   )ri   instZupZrmZheldZkeeppkgr(   r(   r)   _logChanges  s,    

     z!DistUpgradeController._logChangesc                 C   s  t d | jd | js dS t| jjdkrPt d | j| j	 | 
  t| jjdkr| jj}tddt|}tdd	t|d
| }| j	|| dS g }| jddD ]$}|| jkr| j| jr|| qt ddt|  | j | _| j| j| j| j| _t| jdkr:| jddd n| jddd | jr\| jj| _t ddt| j  t ddt| j  dS )NdoPostInitialUpdateZPostInitialUpdateFr   z-packages in reqReinstall state, trying to fixzPackage in inconsistent statezPackages in inconsistent statezThe package '%s' is in an inconsistent state and needs to be reinstalled, but no archive can be found for it. Please reinstall the package manually or remove it from the system.zThe packages '%s' are in an inconsistent state and need to be reinstalled, but no archive can be found for them. Please reinstall the packages manually or remove them from the system.z, r;   MetaPkgszMetaPkgs: %sr   r1   ZforeignPkgsr4   r3   Foreign: %sObsolete: %sT)rH   rI   r^   ro   rN   r&   Zreq_reinstall_pkgsr   Zfix_req_reinstrK   r}   r   rC   rv   rR   r`   r   r   r   _getObsoletesPkgsobsolete_pkgs_getForeignPkgsrY   rW   rX   foreign_pkgsrT   r   ZinstalledTaskstasks)ri   Z	reqreinstheaderr   Z	meta_pkgsr   r(   r(   r)   r     sJ    



z)DistUpgradeController.doPostInitialUpdatec              
   C   s   t d|  | js"t d dS | jj  | j }d}|d k	rJ|}n| j	dd}d}||k rz| j
| W nP ttfk
r } z.t|}t d||f  |d	7 }W Y q\W 5 d }~X Y nX dS t d
 |r| jtdtdd|  dS )Nz"running doUpdate() (showErrors=%s)zBdoUpdate() will not use the network because self.useNetwork==falseTr   r8   r9   r?   zHIOError/SystemError in cache.update(): '%s'. Retrying (currentRetry: %s)r   zdoUpdate() failed completelyzError during updatezA problem occurred during the update. This is usually some sort of network problem, please check your network connection and retry.r   F)rH   rI   rP   rN   Z_listr   rK   r   rR   r_   updater   IOErrorrU   rv   rM   )ri   
showErrorsforceRetriesr   currentRetryrk   Z	error_msgr|   r(   r(   r)   doUpdate  s6    


zDistUpgradeController.doUpdatec                    s>  t jdstd dS dddddd	g}t fd
d|D sNtd dS d}td}|D ]}| }z| \}}}}}	}
W n< t	k
r } ztd||f  W Y q`W 5 d}~X Y nX |dkrq`d}d|dkr`td|  W 5 Q R  dS q`W 5 Q R X |s$ j
tdtd n j
tdtd dS )z= check that /boot/efi is a mounted partition on an EFI systemz/sys/firmware/efizNot an UEFI systemTzshim-signedzgrub-efi-amd64zgrub-efi-ia32zgrub-efi-armzgrub-efi-arm64Z
sicherbootc                 3   s$   | ]}| j ko j | jV  qd S N)rN   r   )r=   Zblr   r(   r)   r     s     z6DistUpgradeController._checkBootEfi.<locals>.<genexpr>z'UEFI system, but no UEFI grub installedFz/proc/mountsz-line '%s' in /proc/mounts not understood (%s)Nz	/boot/efiZrw,zFound writable ESP %sz%EFI System Partition (ESP) not usablezxYour EFI System Partition (ESP) is not mounted at /boot/efi. Please ensure that it is properly configured and try again.zzThe EFI System Partition (ESP) mounted at /boot/efi is not writable. Please mount this partition read-write and try again.)r@   rB   r   rH   rI   r   r   r   splitr%   rK   rv   rM   )ri   ZbootloadersZmountedZmountsr   ZwhatwhereZfsrE   r'   br|   r(   r   r)   _checkBootEfi  s<    


z#DistUpgradeController._checkBootEfic                 C   s>  t d}t d}t d}t d}t d}t d}t d}| jdd	d
rVtd dS |  }z| j| W n tk
r8 }	 zt	j
d}
d}|
|d|d|d|d|d|i}|	jD ]^}|dkr|d7 }|j|kr||j|j|j|j||j d7 }q||j|j|j|jdd7 }q| j|| W Y d
S d}	~	X Y nX dS )z: this checks if we have enough free space on /var and /usrzNot enough free disk spacezThe upgrade has aborted. The upgrade needs a total of {str_total} free space on disk '{str_dir}'. Please free at least an additional {str_needed} of disk space on '{str_dir}'. {str_remedy}zIRemove temporary packages of former installations using 'sudo apt clean'.zYou can remove old kernels using 'sudo apt autoremove' and you could also set COMPRESS=xz in /etc/initramfs-tools/initramfs.conf to reduce the size of your initramfs.zbEmpty your trash and remove temporary packages of former installations using 'sudo apt-get clean'.z!Reboot to clean up files in /tmp.r?   Z	FreeSpaceZ	SkipCheckFz,free space check skipped via config overrideTDir::Cache::archivesz/varz/boot/z/tmpz/usrr   )Z	str_totalZstr_dirZ
str_neededZ
str_remedyN)rM   rR   r   rH   r    _is_apt_btrfs_snapshot_supportedrN   ZcheckFreeSpacer    r\   r   Zfree_space_required_listdirformatZ
size_totalZsize_neededrK   rv   )ri   Zerr_sumerr_msgZremedy_archivedirZremedy_bootZremedy_rootZ
remedy_tmpZ
remedy_usrZwith_snapshotsr|   
archivedirZerr_longZremedyZreqr(   r(   r)   _checkFreeSpace  sV    
     



z%DistUpgradeController._checkFreeSpacec                 C   s   | j td | j| j | j| js*dS | jrB| j| jsBdS | j	 }| j 
  |   | j 
  |  stdS |  sdS | j 
  | j | _t| jdkr| j  tdddd | jD   td| j  | j 
  |S )	NCalculating the changesFr   zdemoted: '%s'r   c                 S   s   g | ]
}|j qS r(   r   r   r(   r(   r)   r>   w  s     z9DistUpgradeController.calcDistUpgrade.<locals>.<listcomp>zfound components: %s)rK   rL   rM   rN   ZdistUpgrader   r]   ZinstallTasksr  get_changesrs   r   r  r  Zget_installed_demoted_packagesinstalled_demotionsr&   sortrH   rI   rC   r   )ri   changesr(   r(   r)   calcDistUpgradeX  s,    




 
z%DistUpgradeController.calcDistUpgradec                 C   s0   |   }|sdS | jtd|| j| jj}|S )NFz!Do you want to start the upgrade?)r!  rK   confirmChangesrM   r  rN   Zrequired_download)ri   r   r   r(   r(   r)   askDistUpgrade~  s    z$DistUpgradeController.askDistUpgradec                 C   s    t  }|| jotjdS )Nz2/var/snap/canonical-livepatch/common/machine-token)distro_infoUbuntuDistroInfois_ltsrW   r@   rB   isfile)ri   dir(   r(   r)   _isLivepatchEnabled  s    z)DistUpgradeController._isLivepatchEnabledc                    sZ   t  }  r| jr dS t fdd|dD  j} jd t	d| }|S )NTc                 3   s    | ]}|j  jkr|jV  qd S r  )ZseriesrX   version)r=   rr   r(   r)   r     s      z5DistUpgradeController.askLivepatch.<locals>.<genexpr>objectzdLivepatch security updates are not available for Ubuntu %s. If you upgrade, Livepatch will turn off.)
r$  r%  r)  r&  rX   nextZget_allrK   ZaskCancelContinueQuestionrM   )ri   r(  r*  r   r(   r   r)   askLivepatch  s     z"DistUpgradeController.askLivepatchc                 C   s0   t jdr,tdt| j  t dd d S )N/etc/cron.daily/aptzdisabling apt cron job (%s)i  )r@   rB   r   rH   rI   octrf   chmodr   r(   r(   r)   _disableAptCronJob  s    z(DistUpgradeController._disableAptCronJobc                 C   s(   t jdr$td t d| j d S )Nr/  zenabling apt cron job)r@   rB   r   rH   rI   r1  rf   r   r(   r(   r)   _enableAptCronJob  s    
z'DistUpgradeController._enableAptCronJobc           
   
   C   s  |    d}| j }| jddd }|rxzt d dd }W n   t	d d}Y nX | j
 d||f  | jdd	}d
}d }||k rXz.t| jj}t|| _| j| j| W n tjjk
r
 }	 z td d}|	}W Y qXW 5 d }	~	X Y nJ tk
rR }	 z*td|	|f  |d7 }|	}W Y qW 5 d }	~	X Y nX dS |rv| jtdtd n&td | jtdtdd|  |   |   d S )Nr   r;   ZSlideshowUrlrM   getdefaultlocaleZenz%s#locale=%sr8   r9   Fzuser canceledT:IOError in cache.commit(): '%s'. Retrying (currentTry: %s)r   zUpgrade canceledzwThe upgrade will cancel now and the original system state will be restored. You can resume the upgrade at a later time.+giving up on fetching after maximum retriesCould not download the upgradeszThe upgrade has aborted. Please check your Internet connection or installation media and try again. All files downloaded so far have been kept.r   )r2  rK   r   rR   r   localer4  r  rH   r   ZgetHtmlViewr   r_   r\   ZPackageManagerrN   	_depcacheZAcquirerO   Z_fetch_archivesaptZFetchCancelledExceptionr   r  rv   r   rM   r3  rw   )
ri   r  	fprogressZurlZlangrk   Zuser_canceledr   Zpmr|   r(   r(   r)   doDistUpgradeFetching  sR    





z+DistUpgradeController.doDistUpgradeFetchingc                 C   sd   zddl }W n tk
r"   Y dS X z| }| }W n   td Y dS X td|  |S )z' check if apt-btrfs-snapshot is usable r   Nzfailed to check btrfs supportFz!apt btrfs snapshots supported: %s)apt_btrfs_snapshotImportErrorAptBtrfsSnapshotZsnapshots_supportedrH   r   rI   )ri   r=  	apt_btrfsr   r(   r(   r)   r    s    
z6DistUpgradeController._is_apt_btrfs_snapshot_supportedc                 C   sF   |   sdS ddl}| }d| j }||}td||f  dS )z2 create btrfs snapshot (if btrfs layout is there) Nr   zrelease-upgrade-%s-z#creating snapshot '%s' (success=%s))r  r=  r?  rX   Zcreate_btrfs_root_snapshotrH   r   )ri   r=  r@  prefixr   r(   r(   r)    _maybe_create_apt_btrfs_snapshot   s    

z6DistUpgradeController._maybe_create_apt_btrfs_snapshotc                 C   s   i }t jd g|d< dt jd< dD ]$}t j|||d < t j| q"z|  W S |D ] }|| D ]}t j|| qfqZX d S )Nzdir::bin::dpkgz	/bin/true)zdpkg::pre-invokezdpkg::pre-install-pkgszdpkg::post-invokezdpkg::post-install-pkgsz::)r\   rR   Z
value_listclearrT   doDistUpgrade)ri   ZbackupsZlstitemr(   r(   r)   doDistUpgradeSimulation
  s    
z-DistUpgradeController.doDistUpgradeSimulationc                 C   s  d}| j  }| j | j}| jdd}| jsD| jd | 	  d}d }||k rz"| j
||}td|  W n( tk
r } z<td|  |}tjdrHtd}| }W 5 Q R X td	|  d
dg}	|	D ]b}
t||
rtd |   td}|dt| 7 }|td7 }| j td| |   qtd}| jspt sp|td| 7 }| j td| dddg}tjddkr|d | j  | |   W Y dS d }~X Y n tk
r } z*td||f  |d7 }|}W Y qLW 5 d }~X Y n t k
r } zdt!d |}|j"dkr|   td}|dt| 7 }|td7 }| j td| |   W 5 d }~X Y nX |   dS td | j tdtd d!|  |   d S )"Nr   r8   r9   ZStartUpgradeFzcache.commit() returned %sz#SystemError from cache.commit(): %sz./var/run/ubuntu-release-upgrader-apt-exceptionzfound exception: '%s'z;E:Internal Error, Could not perform immediate configurationz E:Couldn't configure pre-depend z-detected preconfigure error, restorting stateError during commitz
'%s'
Restoring original system statezCould not install the upgradeszrThe upgrade has aborted. Your system could be in an unusable state. A recovery will run now (dpkg --configure -a).z

Please report this bug in a browser at http://bugs.launchpad.net/ubuntu/+source/ubuntu-release-upgrader/+filebug and attach the files in /var/log/dist-upgrade/ to the bug report.
%sr~   r   r   r   r   r   r5  r   zcache.commit()   Tr6  r7  zdThe upgrade has aborted. Please check your Internet connection or installation media and try again. r   )#rK   r   getInstallProgressrN   rR   r_   r]   r^   ro   rB  commitrH   rI   r   rv   r@   rB   r   r   readrU   r   r3  rM   rw   r!   r[   r   r   r   r   r  r   r   errno)ri   r  r;  	iprogressrk   r   r   r|   r   Zpre_configure_errorsZpreconf_errormsgr   r(   r(   r)   rD    s    








z#DistUpgradeController.doDistUpgradec              
   C   s  t  d tjtjd}| jj	D ]D}tjtj
|j|kr(zt|j W q( tk
rj   Y q(X q(|   | jd | jtj | jtd | j }| j| j| j| j}tddt |  tddt |  | j!dd	D ]&}||krt"d
|  t# } q&q| j$dddsFtd dS | j!ddD ]6}|| jkrT| j| j%rT| j&'| j!|d qTtd| j&  | j(r|| j) }nt#dd | j*D }|t#| j&O }| j+ }tdd|  |t#|O }| j$dddstd t# }td|  td | j, }t-|D ]^\}	}
|.|	t/t0| d  |
| j1krH| j2  | j3|
|| j&| j1sHtd|
 qHtd |4  | j5 }tdddd |D   td}tdtdg}t0|dkr| j6||g d|d r| j7 }| j8| j}z| j9|| W nR t:t;fk
r } z.t"d!|  | j"td"td#d$|  W 5 d }~X Y nX | jd% | j<s| =  dS )&NZPOSTUPGRADEr  ZPostUpgradezSearching for obsolete softwarer   r   r   r;   r   zXthe BaseMetaPkg '%s' is in the obsolete list, something is wrong, ignoring the obsoletesZRemoveObsoletesTzSkipping obsolete Removalr   r<   zforced_obsoletes: %sc                 S   s   g | ]
}|j qS r(   r  )r=   pr(   r(   r)   r>     s     z7DistUpgradeController.doPostUpgrade.<locals>.<listcomp>zUnused dependencies: %sz0Skipping RemoveObsoletes as stated in the configzremove_candidates: '%s'z Start checking for obsolete pkgsg      Y@z:'%s' scheduled for remove but not safe to remove, skippingz!Finish checking for obsolete pkgsz1The following packages are marked for removal: %sc                 S   s   g | ]
}|j qS r(   r  )r=   r   r(   r(   r)   r>     s     zRemove obsolete packages?Z_KeepZ_Remover   Fz,cache.commit() in doPostUpgrade() failed: %srG  z[A problem occurred during the clean-up. Please see the below message for more information. r   ZPostCleanup)>r   Z	add_stager@   rB   r   r\   rR   r   rO   re   abspathZdestfileunlinkr   r}   r^   ro   rK   setStepr   ZCLEANUPrL   rM   rN   r  r  rY   rW   rX   rH   rI   rC   r   r`   rv   rT   r   r   ra   r   rP   r  r  Z_getUnusedDependenciesr   	enumerater  floatr&   r  rs   ZtryMarkObsoleteForRemovaldoner  r"  r   rJ  rK  r   r  r]   runPostInstallScripts)ri   r  rE  Znow_obsoleteZnow_foreignr   Zremove_candidatesZunused_dependenciesr   ir   r   r   Zactionsr;  rN  r|   r(   r(   r)   doPostUpgrade  s    










z#DistUpgradeController.doPostUpgradec                 C   s   | j r| j   | j   | jddD ]}tj|sHt	d|  q(t
d|  z"t| | j j|gdd W q( tk
r } ztd||f  W 5 d}~X Y q(X q(| j r| j   dS )	z 
        scripts that are run in any case after the distupgrade finished 
        whether or not it was successful

        Cache lock is released during script runs in the event that the
        PostInstallScripts require apt or dpkg changes.
        r;   ZPostInstallScriptsz!PostInstallScript: '%s' not foundzRunning PostInstallScript: '%s'T)Zhiddenz(got error from PostInstallScript %s (%s)N)rN   rp   rq   rR   r`   r@   rB   r   rH   r   rI   r   rK   r   r   	Exceptionrv   get_lock)ri   Zscriptr|   r(   r(   r)   rW    s    

&z+DistUpgradeController.runPostInstallScriptsc                 C   sj   t d t| dr"| j| j t| dr:| j| j | jt	d | j
  |   td dS )z2 abort the upgrade, cleanup (as much as possible) zabort calledr   rQ   rH  r   N)rH   rI   hasattrr   Zrestore_backuprS   rQ   rK   rL   rM   rw   r}   rx   ry   r   r(   r(   r)   rw     s    



zDistUpgradeController.abortc           	      C   s   t |D ]}td|  |D ]v}|d }|d }|d }|| jkr\td|    dS | j| }t|jdd}|dkr t |||d	kr   d	S q q
td
|  dS )z+ check if a given depends can be satisfied zchecking: '%s' r   r      z_checkDep: '%s' not in cacheFr*  NTzdepends '%s' is not satisfied)	r\   Zparse_dependsrH   rI   rN   rv   getattrZ	installedZ	check_dep)	ri   ZdepstrZor_groupdepZdepnameverZoperr   Zinstverr(   r(   r)   	_checkDep  s"    

zDistUpgradeController._checkDepc                 C   s   t d d}| jdd}|| j| jjjd |D ]F}t d| || |M }|s:| j	t
dt
d|  td q:|S )	z  check if depends are satisfied zcheckViewDepends()Tr   ZDependszdepends: '%s'z!Required depends is not installedz/The required dependency '%s' is not installed. r   )rH   rI   rR   r`   r   rK   rJ   r,   ra  rv   rM   rx   ry   )ri   r   Zdependsr_  r(   r(   r)   r   '  s"    
z&DistUpgradeController.checkViewDependsc                 C   s   | j dd}d}tj dd}| j dd}||k r| jdd	 |   |D ]$}|| jkrNt	d
|  t
|qN| |rqtd tj dd |d7 }q.||krt	d dS tj d| dS )NPreRequistsPackagesr   zAcquire::http::No-Cacher:   r8   r9   F)r	  zCan not find backport '%s'z>setting a cache control header to turn off caching temporarilytruer   z+pre-requists item is NOT trusted, giving upT)rR   r`   r\   findr_   r  r}   rN   rH   rv   r+   _allBackportsAuthenticatedrI   rT   )ri   backportslistrX  ZnoCacherk   r   r(   r(   r)   _verifyBackports;  s*    





z&DistUpgradeController._verifyBackportsc                 C   s   t jdddkr td dS z| jdd}|r:W dS W n tk
rP   Y nX |D ]8}| j| }|jsp dS |jj	D ]}|j
rx qVqx dS qVdS )NzAPT::Get::AllowUnauthenticatedFTzIskip authentication check because of APT::Get::AllowUnauthenticated==truer;   ZAllowUnauthenticated)r\   rR   Zfind_brH   r   Z
getbooleanr   rN   r   r   Ztrusted)ri   rg  r  r   r   Zcandr(   r(   r)   rf  V  s$    


z0DistUpgradeController._allBackportsAuthenticatedc           
      C   s   | d}t|\}}}}}d|kr2|dd }d|||f }| jD ]J}| d}t||rd dS |dd }	||	rFtd|   dS qFdS )	z  check if uri is a known mirror r  @r   z	%s://%s%sTz//z)found apt-cacher/apt-torrent style uri %sF)rstripr   r  rb   r   endswithrH   rI   )
ri   r   Zraw_uriZschemeZnetlocrB   ZqueryZfragmentmirrorZmirror_host_partr(   r(   r)   r   l  s    




zDistUpgradeController.isMirrorc                 C   s4   | d}| jD ]}| d}t||r dS qdS )z2 check if uri is a whitelisted third-party mirror r  TF)rj  rc   r   )ri   r   rl  r(   r(   r)   r     s    



z(DistUpgradeController.isThirdPartyMirrorc                 C   s   d}t dd}|jD ]}|js|jr&q|jdkr|jdkr| |jrd|jkrd| j |j	kr|j
ds|j
d	sd
|j| jf }||kr||7 }q|S )z7 get sources.list snippet lines for the current mirror r?   r0   r   r   Fr"   r   zhttp://security.ubuntu.comzhttp://archive.ubuntu.comzdeb %s %s-updates main
)r   r   r   r   r   r   r   r   rW   r   r   )ri   dumblinesr   r   Znew_liner(   r(   r)   _getPreReqMirrorLines  s,    






z+DistUpgradeController._getPreReqMirrorLinesc           	   
   C   s|   t d|  | |}t|dP}t|<}|D ]0}t|}|j|d}|| t d|  q2W 5 Q R X W 5 Q R X dS )z9 add prerequists based on template into the path outfile z*writing prerequists sources.list at: '%s' w)rl  zadding '%s' prerequistsT)rH   rI   ro  r   r   Zsafe_substitutewrite)	ri   templateoutrm  ZmirrorlinesoutfileZinfiler   Zoutliner(   r(   r)   _addPreRequistsSourcesList  s    

$z0DistUpgradeController._addPreRequistsSourcesListc                 C   s&  t d d}tjt d}tj|s6t| | j	dd}| j
r| jst d tj| j
jdtjd }t }t|d	 D ]2}t d
||f  |tj|dd  qt||st dt||f  dS | j  tjddgt|d	  tjtjdd}d}|dkrL| }| j  t d q"| jjdd | j!  t "d|  t d|# d   |dkrdS | $ S d}| j%d|d| j&  r|d| j&  }| j'd|}tj|st d|  dS tjtj(d|}	tjtj(d|}
| )||
 z| *  W nh t+k
r   | j)||
dd z| *  W n2 t+k
r } zt ,d|  W 5 d}~X Y nX Y dS X |D ]}| j| }| jj-.|j/}|st d|  t0|	  dS |j1dkrt d|  t0|	  dS t d|  |j2ddd  qd}z"| j3| j4 | j5| j}W nj t6k
r } zt d!|  d}W 5 d}~X Y n6 t7k
r } zt d"|  d}W 5 d}~X Y nX |dkrt ,d# zt0|
 W n2 t8k
r } zt d$|  W 5 d}~X Y nX | $ S )%z5 download the backports specified in DistUpgrade.cfg zgetRequiredBackports()TZ	backportsrb  rc  z#Searching for pre-requists on CDROMz*dists/stable/main/dist-upgrader/binary-%s/zAPT::Architecturez*_*.debzfound pre-req '%s' to '%s'rM   r   z%Expected backports: '%s' but got '%s'Fr~   z-i)r   stderrr   Ng{Gz?)Zfinishedz'installing backport debs exit code '%s'zdpkg output:
%sr   z-%szsourceslist not found '%s'zDir::Etc::sourceparts)rm  zno backport for '%s' foundzNo candidate for '%s'zNo ver.file_list for '%s'zmarking '%s' for install)Z	auto_instZauto_fixzfetch_archives returned '%s'zinstall_archives returned '%s'z,_fetch_archives for backports returned Falsez(failed to unlink pre-requists file: '%s')9rH   rI   r@   rB   rC   rA   r   mkdirrR   r`   rQ   rP   rj   r\   re  rT   r   r   basenamer  issubsetrv   rN   rp   r   r   r   ZSTDOUTZpollrK   ZpulseProgressrt   ru   r[  r   r   setupRequiredBackportsZ
has_optionrZ   r   r   ru  rh  r+   r   r9  Zget_candidate_verZ_pkgrR  Z	file_listmark_installrK  r   rJ  r  r   rZ  )ri   r   r   rg  rP  Z
found_pkgsr   Zconf_optionZprereq_templateZoutpathrt  r|   r   r   r`  r(   r(   r)   getRequiredBackports  s    



 



 




 z*DistUpgradeController.getRequiredBackportsc                 C   sx   dt jd< t  tdd tjdg }| jr<|	d n
|	d t
d ttjd	  t tjd	 |t j d S )
Nz$/usr/lib/release-upgrader-python-apt
PYTHONPATHz/var/log/dist-upgrade/main.logz&/var/log/dist-upgrade/main_pre_req.logz--have-prerequistsz--with-networkz--without-networkzrestarting upgraderr   )r@   r[   rH   Zshutdownshutilr   rx   argvrP   r   r   r   execve)ri   argsr(   r(   r)   rz  *  s    


z,DistUpgradeController.setupRequiredBackportsc              
   C   sb  | j td | j tj |  std t	j
drb| j tdtd tddg n | j tdtd td	 |   | jd
r| jr| jjdkrtd |  st	j
dr| j tdtd tddg n | j tdtd td	 |   |  s.|   | jddd |   |  sV|   z| j tj | j td |  s|   | jr| j| js| j tdtd |   |  s|   | j td | jdd | j | _ | j!"  | j#ddD ]}|| jks6| j$| j| std|  || jkr^td|  | j$| j| std|  | j tdtd|  t	j
drtddg n
td	 |   q| j td | % s|   | &  | j tj' | j td | ( s0| )  |   | j tj* | j td  | + sv| j td!td" |   | j,  d}z| jd# -  d}W n< t.k
r } zd$t/|krtd% W 5 d }~X Y nX W n t0k
r   |   Y nX |rH| j tj* | j td  | 1 sH| j 2td&td' t34d | jdd | j | _ | j!"  | j td | 5 s|   | j tj* | j td  | 1 s| 6  | j 2td(td) t34d | 7  | jr| j8  t	j
d*rt	9d* | j tj: | j td+ t; <  t= s^| j > r^td, t34d- dS ).NzChecking package managerzself.prepare() failedr   zPreparing the upgrade failedzXPreparing the system for the upgrade failed so a bug reporting process is being started.r   r   zPreparing the system for the upgrade failed. To report a bug install apport and then execute 'apport-bug ubuntu-release-upgrader'.r   rb  Fzneed backportsz$Getting upgrade prerequisites failedzThe system was unable to get the prerequisites for the upgrade. The upgrade will abort now and restore the original system state.

Additionally, a bug reporting process is being started.zThe system was unable to get the prerequisites for the upgrade. The upgrade will abort now and restore the original system state.

To report a bug install apport and then execute 'apport-bug ubuntu-release-upgrader'.r   )r	  r
  zUpdating repository informationzFailed to add the cdromz+Sorry, adding the cdrom was not successful.T)r{   r;   r   z@No '%s' available/downloadable after sources.list rewrite+updatez'%s' was not in the cachez'%s' was not downloadablezInvalid package informationa  After updating your package information, the essential package '%s' could not be located. This may be because you have no official mirrors listed in your software sources, or because of excessive load on the mirror you are using. See /etc/apt/sources.list for the current list of configured software sources.
In the case of an overloaded mirror, you may want to try the upgrade again later.r  Fetching	UpgradingzUpgrade infeasiblezQThe upgrade could not be completed, there were errors during the upgrade process.Zlibc6ZpkgProblemResolverz'Unable to mark libc6 alone for install.zUpgrade incompletezUThe upgrade has partially completed but there were errors during the upgrade process.Upgrade completeKThe upgrade has completed but there were errors during the upgrade process.z:/var/lib/ubuntu-release-upgrader/release-upgrade-availablezSystem upgrade is complete./sbin/rebootr   )?rK   rL   rM   rS  r   PREPAREr   rH   rv   r@   rB   r   r   r   rw   rR   rd   rE   r   rI   r|  r.  r  r}   r   MODIFY_SOURCESr   rQ   r   rS   rN   r   r   r^   Z+ensure_recommends_are_installed_on_desktopsr`   ZanyVersionDownloadabler#  _inhibitIdleFETCHr<  r3  INSTALLrF  rC  r{  r   rU   KeyboardInterruptrD  r   rx   ry   r!  rW  rY  Zcomment_out_cdrom_entryrR  REBOOTr   rV  r   confirmRestart)ri   r   Zlibc6_possibler|   r(   r(   r)   fullUpgrade?  s   




	






















z!DistUpgradeController.fullUpgradec                 C   s   | j   |  S r  )rK   rs   r  r   r(   r(   r)   ro   *  s    
zDistUpgradeController.runc                 C   s  | j tj | j tj | j tj d| _|   | 	 sDdS | 
 sPdS | j tj | j td |  szdS | j tj | j td |  s| j tdtd dS |  s| j tdtd dS tjtr| j  rtd n| j tdtd dS )	z, partial upgrade mode, useful for repairing TFr  r  r  r  r  z"The partial upgrade was completed.)rK   rS  r   r  ZhideStepr  r  r]   r   r   r#  r  rL   rM   r<  r  rD  r   rY  r@   rB   r   REBOOT_REQUIRED_FILEr  r   r   r   r(   r(   r)   doPartialUpgrade.  s@    
z&DistUpgradeController.doPartialUpgradec              	   C   s  t jdrt jdd| _| js4t jdd| _| jsRtd td d S | 	  | j
sdd S t t| j td zb| j
dd}|std	 td W d S tjd
dddg| j
tjd | jtdtd W n$ ttfk
r   td Y nX t t   d S )Nz/usr/bin/gnome-session-inhibitZSUDO_UIDr?   Z
PKEXEC_UIDz"failed to determine user upgradingz$failed to inhibit gnome-session idlezinhibit gnome-session idleZXDG_CURRENT_DESKTOPz"failed to find XDG_CURRENT_DESKTOPzgnome-session-inhibitz	--inhibitZidlez--inhibit-only)envr   zLock screen disabledzMYour lock screen has been disabled and will remain disabled until you reboot.)r@   rB   r   r[   r   rg   rH   rI   rv   _getUserEnvrh   seteuidintr   r   ZDEVNULLrK   r   rM   r   r%   r   getuid)ri   Zxdg_desktopr(   r(   r)   r  U  s@    




 z"DistUpgradeController._inhibitIdlec              
   C   s   zt dd| jdg}| dd }td| d d}| d	}W 5 Q R X |D ]*}t|rX|d
d}|d | j|d < qXW nD t j	k
r } z$|j
dkrtd n
td W 5 d }~X Y nX d S )NZpgrepz-uzgnome-session
r   z/proc/z/environr+   =r   z"gnome-session not running for userzfailed to read user env)r   Zcheck_outputrg   decoder  r   rL  r&   rh   ZCalledProcessError
returncoderH   rI   r   )ri   pidr   datar   r  r|   r(   r(   r)   r  w  s    
z!DistUpgradeController._getUserEnv)NN)TF)T)TN)F)F)1r,   r-   r.   __doc__rl   r}   rr   r   r   r   r   r   r   r   r   r   r   r  r  r  r!  r#  r)  r.  r2  r3  r<  r  rB  rF  rD  rY  rW  rw   ra  r   rh  rf  r   r   ro  ru  r|  rz  r  ro   r  r  r  r(   r(   r(   r)   r/   ^   sb   
h
#;`
  +e3
$/:&J
fm	

z l'"r/   __main__)DistUpgradeViewText)level)Kr:  r\   r$  rx   r@   r   r8  rH   r~  r   rt   r   configparserr   r   r   Z	telemetryr   r   Zutilsr   r   r   r	   r
   r   r   r   r   stringr   Zurllib.parser   ZDistUpgradeViewr   ZDistUpgradeCacher   ZDistUpgradeConfigParserr   r   ZDistUpgradeAptCdromr   Z
aptsourcesr   r   ZDistInfoZaptsources.sourceslistr   r   r   r   r   ZDistUpgradeGettextr   rM   r   r   r   r    ZDistUpgradeApportr!   r  r*   rZ  r+   r,  r/   r,   r  ZbasicConfigDEBUGvZdcr2  r3  r(   r(   r(   r)   <module>   sv   ,	                <
