Y]C@sSddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z mZddlmZy)ejejdej\ZZWne jdd ZYnXes8d Zejd d yejejd Wnejk r7YnXe jd ejd dZGdddejjjZGdddejjj Z Gddde!Z"Gddde!Z#e$dd\Z%Z&Z'Z(Z)Z*Z+edededededfGddde!Z,e-dkrOeZ.e.j/dS) )gettext)ngettextN) doAufsChrootdoAufsChrootRsync)apport_pkgfailurezgetting the encoding failedzutf-8LC_CTYPEzC.UTF-8zlocale: '%s' '%s'cCst|}|d}|dd}|dd}|d}|dkrNd}ddddddd di}|dkrtd d |||d<|dkrtd d |||d<|dkrtdd|||d|dS|dkrp|ddkrb|dStd|S|dkr|dS|d S)z; return the time a bit fuzzy (no seconds if time > 60 secs <rrZstr_daysrZ str_hoursZ str_minutesZ str_secondsz%li dayz%li daysz%li hourz %li hoursz %li minutez %li minutesz %li secondz %li secondsz%(str_days)s %(str_hours)sz%(str_hours)s %(str_minutes)siiQi)intr_)ZsecZdaysZhoursZminutesZsecondsmapr=/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeView.pyFuzzyTimeToStr:s>              rcseZdZfddZfddZfddZddZd fd d Zd d ZddZ S)AcquireProgresscs tt|jd|_dS)Ng)superr__init__ est_speed)self) __class__rrrszAcquireProgress.__init__cs;tt|jd|_d|_d|_d|_dS)NgF)rrstartretapercentrelease_file_download_error)r)rrrrs    zAcquireProgress.startcstt|j|||||tjkrtjd||jds]|jdrx0dddgD]}|j|rmd|_ PqmWdS)Nz update_status: dlFailed on '%s' z Release.gpgZReleaseZhttpZftpZmirrorT) rr update_statusapt_pkgZ STAT_FAILEDloggingZwarningendswith startswithr)rZuriZdescrZ shortDescrZstatusZnet)rrrrs zAcquireProgress.update_statuscCsdS)NTr)rargrrr pulse_itemsszAcquireProgress.pulse_itemsNcstt|j||j|jdt|j|j|_|j |j krj|j |j d|_ |j dkr|j|jt|j |_ dS)NgY@g@rT) rrpulseZ current_bytesZ current_itemsfloatZ total_bytesZ total_itemsrZ current_cpsrr)rowner)rrrr$s zAcquireProgress.pulsecCs |jdkS)Nr)r)rrrrisDownloadSpeedEstimatedsz(AcquireProgress.isDownloadSpeedEstimatedcCsj|jdkrI|d}|d }tdt|t|f}|Stdt||j}|S) z! get the estimated download time r8iz[This download will take about %s with a 1Mbit DSL connection and about %s with a 56k modem.z7This download will take about %s with your connection. ig@igA)rrr)rZrequired_downloadZ timeModemZtimeDSLsrrrestimatedDownloadTimes   z%AcquireProgress.estimatedDownloadTime) __name__ __module__ __qualname__rrrr#r$r'r+rr)rrr~s   rc@sFeZdZdZddZddZddZdd Zd S) InstallProgressz_ Base class for InstallProgress that supports some fancy stuff like apport integration cCs#tjjjj|d|_dS)N)aptprogressbaser/rZ master_fd)rrrrrszInstallProgress.__init__cCsxy#tj|jggg|jWnAtjk ri}z|jdtjkrWWYdd}~XnX|jy2tj |j tj \}}||j krPWqt k r}z-|jtjkr|jtj krPWYdd}~XqXqW|S)zWait for child progress to exit. The return values is the full status returned from os.waitpid() (not only the return code). rN)selectZstatusfdZselect_timeouterrorargserrnoZEINTRZupdate_interfaceoswaitpid child_pidWNOHANGOSErrorZECHILD)repidresrrr wait_childs"# zInstallProgress.wait_childcCs|j}|dkrdtjkr[ttjdtjds[tdtjdtjtjtjy|j |j }Wn{t k r}z[td|t j dtdd }|jt|WdQRXtj|jWYdd}~XnXtj|||_tj|j}|dkrd tjkrt jd ttjdst jd |jS|S) NrZRELEASE_UPGRADE_USE_AUFS_CHROOTZRELEASE_UPGRADE_AUFS_RWDIRz*ERROR: failed to setup aufs chroot overlayrz!Exception during pm.DoInstall(): zException during pm.DoInstall()z./var/run/ubuntu-release-upgrader-apt-exceptionwZ!RELEASE_UPGRADE_RSYNC_AUFS_CHROOTz doing rsync commit of the updatez/FATAL ERROR: doAufsChrootRsync() returned FALSE)forkr7environrprint_exitsignalSIGPIPESIG_IGNZ do_installZwritefd Exceptionr exceptionopenwritestrZ RESULT_FAILEDr9 WEXITSTATUSr?inforr4)rZpmr=r>r<frrrruns4       #     zInstallProgress.runcCstjjjj|||tjd||fd|krQtjj|}d|krp|j dd}t ||dS)z install error from a package z*got an error from dpkg for pkg: '%s': '%s'/rrN) r0r1r2r/r4rr7pathbasenamesplitr)rpkgZerrormsgrrrr4s  zInstallProgress.errorN)r,r-r.__doc__rr?rPr4rrrrr/s    !r/c@seZdZdddZdS) DumbTerminalFcCsddl}|j|dS)z7 expects a command in the subprocess style (as a list) rN) subprocesscall)rcmdZhiddenrXrrrrYs zDumbTerminal.callN)r,r-r.rYrrrrrWs rWc@s4eZdZddZddZddZdS) DummyHtmlViewcCsdS)Nr)rZurlrrrrJszDummyHtmlView.opencCsdS)Nr)rrrrshow szDummyHtmlView.showcCsdS)Nr)rrrrhide szDummyHtmlView.hideN)r,r-r.rJr\r]rrrrr[s   r[r)zPreparing to upgradezGetting new software channelszGetting new packageszInstalling the upgradesz Cleaning upc@seZdZdZddZddZddZdd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddddZdddZdd Zdd!d"Zdd#d$Zd%d&Zd'd(d)Zd*d+ZdS),DistUpgradeViewz" abstraction for the upgrade view cCs d|_dS)NF)Z needs_screen)rrrrr!s zDistUpgradeView.__init__cCstjjjS)z5 return a OpProgress() subclass for the given graphic)r0r1r2Z OpProgress)rrrrgetOpCacheProgress$sz"DistUpgradeView.getOpCacheProgresscCstS)z# return an acquire progress object )r)rrrrgetAcquireProgress'sz"DistUpgradeView.getAcquireProgressNcCstS)z" return a install progress object )r/)rcacherrrgetInstallProgress*sz"DistUpgradeView.getInstallProgresscCstS)N)rW)rrrr getTerminal-szDistUpgradeView.getTerminalcCstS)N)r[)rrrr getHtmlView/szDistUpgradeView.getHtmlViewcCsdS)z\ update the current status of the distUpgrade based on the current view Nr)rmsgrrr updateStatus1szDistUpgradeView.updateStatuscCsdS)z8 provide a visual feedback that the upgrade was aborted Nr)rrrrabort6szDistUpgradeView.abortcCsdS)z we have 6 steps current for a upgrade: 1. Analyzing the system 2. Updating repository information 3. fetch packages 3. Performing the upgrade 4. Post upgrade stuff 5. Complete Nr)rsteprrrsetStep9s zDistUpgradeView.setStepcCsdS)z" hide a certain step from the GUI Nr)rrhrrrhideStepCszDistUpgradeView.hideStepcCsdS)z" show a certain step from the GUI Nr)rrhrrrshowStepFszDistUpgradeView.showStepTc Csd|_||_g|_g|_g|_g|_g|_g|_x|D]}|jrq|jj |qO|j r|jj |qO|j r|jj |qO|j r|j jj|jr|jj |q |jj |qO|jrO|jj |qOW|j|j|_|jj|jj|jj|jj|jjd}t|jt|j} t|jt|j} t|j} t|jdkr|tddt|jdt|ji7}|d7}| dkr/|tdd | | 7}|d 7}| dkr_|td d | | 7}|d 7}| dkr|td d| | 7}|d 7}|dkrtj|} t| tr| jt} |td| 7}||jj|7}| | dkrY| | | dkrY|jjrB|dtd7}qy|dtd7}n | dkry|dtd7}| | | dkrtd}td}|j||dS||_dS)zm display the list of changed packages (apt.Package) and return if the user confirms them r rzo%(amount)d installed package is no longer supported by Canonical. You can still get support from the community.zq%(amount)d installed packages are no longer supported by Canonical. You can still get support from the community.Zamountz z"%d package is going to be removed.z$%d packages are going to be removed. z(%d new package is going to be installed.z*%d new packages are going to be installed.z#%d package is going to be upgraded.z%%d packages are going to be upgraded.z& You have to download a total of %s. dz %sznInstalling the upgrade can take several hours. Once the download has finished, the process cannot be canceled.z{Fetching and installing the upgrade can take several hours. Once the download has finished, the process cannot be canceled.z.Removing the packages can take several hours. rz,The software on this computer is up to date.zRThere are no upgrades available for your system. The upgrade will now be canceled.FT) ZconfirmChangesMessage demotionsZ toInstallZ toReinstallZ toUpgradeZtoRemoveZ toRemoveAutoZ toDowngradeZmarked_installappendZmarked_upgradeZmarked_reinstallZ marked_deleteZ_pcacheZ _depcacheZis_auto_installedZ_pkgZmarked_downgradesortlenrrZ size_to_str isinstancebytesdecodeENCODINGrr`r+r'r4) rsummaryZchangesroZ downloadSizeZactionsZ removal_boldrUreZ pkgs_removeZ pkgs_instZ pkgs_upgradeZdownloadSizeStrrrrconfirmChangesIs                   &            $    zDistUpgradeView.confirmChangesZNocCsdS)z0 ask a Yes/No question and return True on 'Yes' Nr)rrwredefaultrrraskYesNoQuestionsz DistUpgradeView.askYesNoQuestioncCs(td}td}|j||S)z2 generic ask about the restart, can be overridden zReboot requiredzMThe upgrade is finished and a reboot is required. Do you want to do this now?)rrz)rrwrerrrconfirmRestarts  zDistUpgradeView.confirmRestartcCsdS)z display a error Nr)rrwre extended_msgrrrr4szDistUpgradeView.errorcCsdS)z display a information msgNr)rrwrer|rrr informationszDistUpgradeView.informationcCsdS)zQ process gui events (to keep the gui alive during a long computation Nr)rrrr processEventsszDistUpgradeView.processEventsFcCsdS)zX do a progress pulse (e.g. bounce a bar back and forth, show a spinner) Nr)rZfinishedrrr pulseProgressszDistUpgradeView.pulseProgresscCs |j||dj|dS)zr show demoted packages to the user, default implementation is to just show a information dialog rlN)r}join)rrwrerorrr showDemotionsszDistUpgradeView.showDemotions)r,r-r.rVrr_r`rbrcrdrfrgrirjrkrxrzr{r4r}r~rrrrrrr^s(          b  r^__main__)0ZDistUpgradeGettextrrrr0r6r7rZlocalerrEr3ZDistUpgradeAufsrrZDistUpgradeApportr setlocaleLC_ALLZgetdefaultlocalecodervrIputenvr r4rNZ getlocalerr1r2rr/objectrWr[rangeZ STEP_PREPAREZSTEP_MODIFY_SOURCESZ STEP_FETCHZ STEP_INSTALLZ STEP_CLEANUPZ STEP_REBOOTZSTEP_Nr^r,fpr$rrrrsP           D6L$