U ٪zcA@sddlmZddlmZddlmZddlZddlm Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlmZze e jde \ZZWned d ZYnXesd Ze d d ze e jd Wne jk rYnXed e ddZGdddejj j!Z!Gdddejj j"Z"Gddde#Z$Gddde#Z%Gddde Z&edededededfGdd d e#Z'e(d!kre!Z)e)*dS)")gettext)ngettext)getN)Enum)apport_pkgfailurezgetting the encoding failedzutf-8LC_CTYPEzC.UTF-8zlocale: '%s' '%s'cCs0t|}|d}|dd}|dd}|d}|dkr 60 secs iQi<rrr)str_days str_hours str_minutes str_secondsz%li dayz%li daysr z%li hourz %li hoursr z %li minutez %li minutesrz %li secondz %li secondsrz%(str_days)s %(str_hours)sz%(str_hours)s %(str_minutes)s)intr_)ZsecZdaysZhoursZminutesZsecondsmapr=/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeView.pyFuzzyTimeToStr;s@       rcsZeZdZfddZfddZfddZddZdfd d Zd d ZddZ Z S)AcquireProgresscstt|d|_dS)N)superr__init__ est_speedself __class__rrrszAcquireProgress.__init__cs*tt|d|_d|_d|_d|_dS)NrF)rrstartretapercentrelease_file_download_errorrrrrr s zAcquireProgress.startcsdtt||||||tjkr`td||dsB|dr`dD]}||rFd|_ q`qFdS)Nz update_status: dlFailed on '%s' z Release.gpgZRelease)ZhttpZftpZmirrorT) rr update_statusapt_pkgZ STAT_FAILEDloggingZwarningendswith startswithr#)rZuriZdescrZ shortDescrZstatusZnetrrrr$s  zAcquireProgress.update_statuscCsdS)NTr)rargrrr pulse_itemsszAcquireProgress.pulse_itemsNcsvtt|||j|jdt|j|j|_|j |j krP|j |j d|_ |j dkrr|j|jt|j |_ dS)NgY@g@rT) rrpulseZ current_bytesZ current_itemsfloatZ total_bytesZ total_itemsr"Z current_cpsrr!)rownerrrrr+s   zAcquireProgress.pulsecCs |jdkS)Nr)rrrrrisDownloadSpeedEstimatedsz(AcquireProgress.isDownloadSpeedEstimatedcCsP|jdkr6|d}|d}tdt|t|f}|Stdt||j}|S)z! get the estimated download time rg@gAz[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. )rrr)rZrequired_downloadZ timeModemZtimeDSLsrrrestimatedDownloadTimes z%AcquireProgress.estimatedDownloadTime)N) __name__ __module__ __qualname__rr r$r*r+r.r0 __classcell__rrrrrs    rc@s0eZdZdZddZddZddZdd Zd S) InstallProgressz_ Base class for InstallProgress that supports some fancy stuff like apport integration cCstjjj|d|_dSN)aptprogressbaser5rZ master_fdrrrrrszInstallProgress.__init__c Cszt|jggg|jWn6tjk rP}z|jdtjkr@W5d}~XYnX|z&t |j tj \}}||j kr~WqWqt k r}z&|jtjkr|jtj krWYqW5d}~XYqXq|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_childc Cs|}|dkrttjtjz||j}Wnftk r}zHtd|t dt dd}| t |W5QRXt |jW5d}~XYnXt |||_t |}|S)Nrz!Exception during pm.DoInstall(): zException during pm.DoInstall()z./var/run/ubuntu-release-upgrader-apt-exceptionw)forksignalSIGPIPESIG_IGNZ do_installZwritefd Exceptionprintr& exceptionopenwritestrr>_exitZ RESULT_FAILEDr@ WEXITSTATUSrF)rZpmrDrErCfrrrruns    zInstallProgress.runcCstjjj|||td||fd|kr:tj|}| dd rdd | ddd}d|krz| dd}t ||dS)z install error from a package z*got an error from dpkg for pkg: '%s': '%s'/-rrNr) r7r8r9r5r;r&r>pathbasenamesplitisdigitjoinr)rpkgZerrormsgrrrr;s zInstallProgress.errorN)r1r2r3__doc__rrFrUr;rrrrr5s r5c@seZdZdddZdS) DumbTerminalFcCsddl}||dS)z7 expects a command in the subprocess style (as a list) rN) subprocesscall)rcmdZhiddenr`rrrraszDumbTerminal.callN)F)r1r2r3rarrrrr_sr_c@s$eZdZddZddZddZdS) DummyHtmlViewcCsdSr6r)rZurlrrrrOszDummyHtmlView.opencCsdSr6rrrrrshowszDummyHtmlView.showcCsdSr6rrrrrhideszDummyHtmlView.hideN)r1r2r3rOrdrerrrrrcsrcc@s(eZdZdZdZdZdZdZdZdZ dS) SteprrN) r1r2r3ZPREPAREZMODIFY_SOURCESZFETCHZINSTALLZCLEANUPZREBOOTNrrrrrfsrfzPreparing to upgradezGetting new software channelszGetting new packageszInstalling the upgradesz Cleaning upc@seZdZdZddZddZddZd/d d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZd0ddZd1ddZd2d d!Zd"d#Zd3d$d%Zd4d&d'Zd(d)Zd5d+d,Zd-d.ZdS)6DistUpgradeViewz" abstraction for the upgrade view cCs d|_dS)NF)Z needs_screenrrrrrszDistUpgradeView.__init__cCs tjjS)z5 return a OpProgress() subclass for the given graphic)r7r8r9Z OpProgressrrrrgetOpCacheProgresssz"DistUpgradeView.getOpCacheProgresscCstS)z# return an acquire progress object )rrrrrgetAcquireProgresssz"DistUpgradeView.getAcquireProgressNcCstS)z" return a install progress object )r5)rcacherrrgetInstallProgress sz"DistUpgradeView.getInstallProgresscCstSr6)r_rrrr getTerminal#szDistUpgradeView.getTerminalcCstSr6)rcrrrr getHtmlView%szDistUpgradeView.getHtmlViewcCsdS)z\ update the current status of the distUpgrade based on the current view Nr)rmsgrrr updateStatus'szDistUpgradeView.updateStatuscCsdS)z8 provide a visual feedback that the upgrade was aborted Nrrrrrabort,szDistUpgradeView.abortcCst|jdS)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 N) get_telemetryZ add_stagenamersteprrrsetStep/s zDistUpgradeView.setStepcCsdS)z" hide a certain step from the GUI NrryrrrhideStep:szDistUpgradeView.hideStepcCsdS)z" show a certain step from the GUI NrryrrrshowStep=szDistUpgradeView.showStepTc Csd|_||_g|_g|_g|_g|_g|_g|_|D]}|jrL|j |q4|j r`|j |q4|j rt|j |q4|j r|j j|jr|j |q|j |q4|jr4|j |q4|j|j|_|j|j|j|j|jd}t|jt|j} t|jt|j} t|j} t|jdkrl|tddt|jdt|ji7}|d7}| dkr|tdd | | 7}|d 7}| dkr|td d | | 7}|d 7}| dkr|td d| | 7}|d 7}|dkr*t|} t| tr| t} |td| 7}|||7}| | dkr|| | | dkr||rj|dtd7}n|dtd7}n| dkr|dtd7}| | | dkrtd}td}|||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_downgradesortlenrr%Z size_to_str isinstancebytesdecodeENCODINGrror0r.r;) rsummaryZchangesrZ downloadSizeZactionsZ removal_boldr]rtZ pkgs_removeZ pkgs_instZ pkgs_upgradeZdownloadSizeStrrrrconfirmChanges@s                 zDistUpgradeView.confirmChangesNocCsdS)z0 ask a Yes/No question and return True on 'Yes' NrrrrtdefaultrrraskYesNoQuestionsz DistUpgradeView.askYesNoQuestionCancelcCsdS)z= ask a Cancel/Continue question and return True on 'Continue'NrrrrraskCancelContinueQuestionsz)DistUpgradeView.askCancelContinueQuestioncCstd}td}|||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?)rr)rrrtrrrconfirmRestartszDistUpgradeView.confirmRestartcCsdS)z display a error NrrrrtZ extended_msgrrrr;szDistUpgradeView.errorcCsdS)z display a information msgNrrrrr informationszDistUpgradeView.informationcCsdS)zQ process gui events (to keep the gui alive during a long computation Nrrrrr processEventsszDistUpgradeView.processEventsFcCsdS)zX do a progress pulse (e.g. bounce a bar back and forth, show a spinner) Nr)rZfinishedrrr pulseProgressszDistUpgradeView.pulseProgresscCs|||d|dS)zr show demoted packages to the user, default implementation is to just show a information dialog r~N)rr\)rrrtrrrr showDemotionsszDistUpgradeView.showDemotions)N)NT)r)r)N)N)F)r1r2r3r^rrnrorqrrrsrurvr{r|r}rrrrr;rrrrrrrrrms.   c     rm__main__)+ZDistUpgradeGettextrrrZ telemetryrrwr7enumrr=r>r%Zlocaler&rIr:ZDistUpgradeApportr setlocaleLC_ALLZgetdefaultlocalecoderrNputenvr r;infoZ getlocalerr8r9rr5objectr_rcrfrmr1fpr+rrrrsV        D6@ 3