Y]4@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Zddl m Z mZddl mZmZddlmZmZmZddlmZGdddeZGd d d eZGd d d eZed kreZeZeZejddejZx)ej ddD]Z!ee!j"qWej#eeej$dej%ddS)N)NoSectionError NoOptionError)PIPEPopen)DistUpgradeViewInstallProgressAcquireProgress)DistUpgradeConfigc@seZdZddZdS)NonInteractiveAcquireProgresscCsttj||||||tjkrptd||jtjt|jft j j rpt j j dS)Nzfetched %s (%.2f/100) at %sb/s) r update_statusapt_pkgZ STAT_DONEprintpercentZ size_to_strintZ current_cpssysstdoutisattyflush)selfZuriZdescrZ shortDescrstatusrK/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeViewNonInteractive.pyr +s &z+NonInteractiveAcquireProgress.update_statusN)__name__ __module__ __qualname__r rrrrr *s r c@sveZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)NonInteractiveInstallProgressz Non-interactive version of the install progress class This ensures that conffile prompts are handled and that hanging scripts are killed after a (long) timeout via ctrl-c cCstj|tjddtjkr6dtjd print error and reboot rNzgot exception '%s': %s znot handled exception: %s) tracebackr!Z exceptionformat_exceptionr3rSconfirmRestart)rtypevaluetbrtlinesrrrrqs  z(DistUpgradeViewNonInteractive.excepthookcCs|jS)z5 return a OpProgress() subclass for the given graphic)rp)rrrrgetOpCacheProgress sz0DistUpgradeViewNonInteractive.getOpCacheProgresscCs|jS)z# return an acquire progress object )rl)rrrrgetAcquireProgresssz0DistUpgradeViewNonInteractive.getAcquireProgresscCs|jS)z" return a install progress object )rm)rcacherrrgetInstallProgresssz0DistUpgradeViewNonInteractive.getInstallProgresscCsdS)z\ update the current status of the distUpgrade based on the current view Nr)rmsgrrr updateStatussz*DistUpgradeViewNonInteractive.updateStatuscCsdS)z we have 5 steps current for a upgrade: 1. Analyzing the system 2. Updating repository information 3. Performing the upgrade 4. Post upgrade stuff 5. Complete Nr)rsteprrrsetStepsz%DistUpgradeViewNonInteractive.setStepTcCstj||||||tjddd|jDtjddd|jDtjddd|jDdS) Nztoinstall: '%s'cSsg|]}|jqSr)rE).0prrr &s z@DistUpgradeViewNonInteractive.confirmChanges..ztoupgrade: '%s'cSsg|]}|jqSr)rE)rrrrrr's ztoremove: '%s'cSsg|]}|jqSr)rE)rrrrrr(s T)rconfirmChangesr!r"Z toInstallZ toUpgradeZtoRemove)rsummaryZchangesZ demotionsZ downloadSizeZactionsZ removal_boldrrrr"s  !!!z,DistUpgradeViewNonInteractive.confirmChangesZNocCsdS)z0 ask a Yes/No question and return True on 'Yes' Tr)rrrdefaultrrraskYesNoQuestion*sz.DistUpgradeViewNonInteractive.askYesNoQuestioncCs#tjd|jjdddS)z2 generic ask about the restart, can be overridden zconfirmRestart() calledrZ RealRebootF)r!r"r%r()rrrrrv1s z,DistUpgradeViewNonInteractive.confirmRestartcCstjd|||fdS)z display a error z %s %s (%s)N)r!r3)rrrZ extended_msgrrrr36sz#DistUpgradeViewNonInteractive.errorcCstjddS)Nzview.abort called)r!r3)rrrrabort9sz#DistUpgradeViewNonInteractive.abort)rrrrjr rqr{r|r~rrrrrvr3rrrrrrks       rk__main__z xserver-xorgzpre-installation script failedr.)&rnr r!rarIrr#rgr^r@r4Z apt.progress configparserrrrrrrr ZDistUpgradeConfigParserr r rrkrZviewZapZipr3ZCacher}argvrBZ mark_installZcommitrJexitrrrrs:             C