3 _@sdZdZd%Zd d lZd d lZd d lZd d lZd d lZd d lZd d l Z d d l Z d d l Z d d l Z d d l Z d d lZd d lZd d lZd d lmZd d lmZd dlmZd dlmZd dlmZddZejdZejdZd&Zej ej!ej"ej#ej$ej%ej&dZ'e(ddZ)Gdddej*j+j,Z-Gdddej*j+j.Z/Gddde/Z0Gddde1Z2Gddde2Z3Gd dde2Z4Gd!dde4Z5Gd"dde5Z6Gd#d$d$e5Z7d S)'z$Progress handlers for APT operationsz%Sebastian Heinlein DaemonAcquireProgressDaemonOpenProgressDaemonInstallProgressDaemonDpkgInstallProgressDaemonForkProgressDaemonDpkgRecoverProgressDaemonLintianProgressN)GLib)enums)lock)mainloop)IsoCodescCs|S)N)srr4/usr/lib/python3/dist-packages/aptdaemon/progress.py+srzAptDaemon.WorkerzAptDaemon.Worker.Terminal <)installZ configureremoveZtrigprocpurgeZ disappearZupgradez\[[;?0-9]*[A-Za-z]c@s,eZdZdZd ddZd dd Zd d ZdS)rz*Handles the progress of the cache opening.rdFcsTtjjjj|||_fddd D|_t|_|jj d|_ d|_||_ dS) aRInitialize a new DaemonOpenProgress instance. Keyword arguments: transaction -- corresponding transaction D-Bus object begin -- begin of the progress range (defaults to 0) end -- end of the progress range (defaults to 100) quiet -- do not emit any progress information for the transaction csg|]}|qSrr).0Zmodifier)beginendrr Lsz/DaemonOpenProgress.__init__..????rN)rrr r!r!) aptprogressbase OpProgress__init__ _transactionstepsfloatprogress_beginpop progress_endquiet)self transactionrrr-r)rrrr&As    zDaemonOpenProgress.__init__NcCs\|p|j}|dkr@t|j|d|j|j}|j|krDdSnd}||_|jsX||j_dS)zpCallback for progress updates. Keyword argument: percent - current progress in percent erN)percentintr*r,r#r-r')r.r1r#rrrupdateUs   zDaemonOpenProgress.updatec Cs>|j|_|j|_y|jjd|_WntjdYnXdS)zXCallback after completing a step. Sets the progress range to the next interval.rz0An additional step to open the cache is requiredN)r,r#r*r(r+logwarning)r.rrrdonehs zDaemonOpenProgress.done)rrF)N)__name__ __module__ __qualname____doc__r&r3r6rrrrr=s  c@sleZdZdZdddZdddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZdS)rz- Handle the package download process rrcCs,tjjjj|||_||_||_d|_dS)Nr)r"r#r$AcquireProgressr&r/r,r*)r.r/rrrrrr&zs zDaemonAcquireProgress.__init__cCs|jjtjjkr tj}|jj}nf|jjtjjkr8tj }nN|jjtjj krPtj }n6|jjtjj krhtj }n|jjtjjkrtj}ntj}|jjtjjkr|jjr|jj}n|jjr|jj}nd}|j||j||jjB||jjB|f|j_dS)N)ownerstatusapt_pkg AcquireItem STAT_DONEr DOWNLOAD_DONEfilesizeSTAT_AUTH_ERRORDOWNLOAD_AUTH_ERROR STAT_FETCHINGDOWNLOAD_FETCHING STAT_ERRORDOWNLOAD_ERROR STAT_IDLE DOWNLOAD_IDLEDOWNLOAD_NETWORK_ERROR error_textmodeuri shortdesc partialsizer/progress_download)r.item total_size current_sizer>msgrrr_emit_acquire_items,      z(DaemonAcquireProgress._emit_acquire_itemcCsjt}x0|D](}|jjr(|j|jjq |j|jq W|rf|jjddt|ddj|i}||j_ dS)z$Emit the transaction status details.zDownloading %(files)sfiles N) setr=idaddrPr/ngettextlenjoinstatus_details)r.itemsnamesrSrVrrr_emit_status_detailss  z*DaemonAcquireProgress._emit_status_detailscCs|j|dS)z cancellable)r.rrrstarts zDaemonAcquireProgress.startcCsd|j_|j|j_d|j_dS)z$Callback at the end of the operationrFN)rrrrrqr)r/rhr,r#ro)r.rrrstops zDaemonAcquireProgress.stopcCsR||f|j_d|j_tj|j_x|jjr6tjjq Wtj |j_|jj rNdSdS)zCallback for media changesTF) r/Zrequired_mediumpausedr ZSTATUS_WAITING_MEDIUMr>r rjrlrnrg)r.ZmediumZdriverrr media_changes    z"DaemonAcquireProgress.media_changeN)rr)rr)r7r8r9r:r&rWrcr6rdrerfrmrprrrtrrrrrvs   #c@s4eZdZdZd ddZddZdd Zdd d Zd S)DaemonAcquireRepoProgressz*Handle the repository information downloadrrcCs6tj||||tdddd|_tdd|_d|_dS)NZiso_639Ziso_639_1_codeZiso_639_2T_code)tagZ fallback_tagZiso_3166Z alpha_2_coder0)rr&r languagesregionsr#)r.r/rrrrrr&s   z"DaemonAcquireRepoProgress.__init__cCstj|j_d|j_dS)z*Callback at the beginning of the operationTN)r ZSTATUS_DOWNLOADING_REPOr/r>ro)r.rrrrp s zDaemonAcquireRepoProgress.startc Cst}xV|D]N}y"|j|jjdjddWq tk rX|j|jjdYq Xq W|r|jjddt|dj |}||j_ dS) z$Emit the transaction status details.rz://r zlocal repositoryzDownloading from %srYN) rZr\ descriptionsplit IndexErrorr/gettextr]r^r_r`)r.raZreposrSrVrrrrcs "z.DaemonAcquireRepoProgress._emit_status_detailscCs|jjtjjkr tj}|jj}nf|jjtjjkr8tj }nN|jjtjj krPtj }n6|jjtjj krhtj }n|jjtjjkrtj}ntj}|jjtjjkr|jjr|jj}n|jjr|jj}nd}|jjdd\}}y|jdd}Wn"tk r|jjd}YnXd||f} |jdkr6|jjd | }nn|jd krV|jjd | }nN|jd krv|jjd | }n.|jjdr|jjd| }n |jjdr|jjd| }n|jdkr|jjd| }n|jjdr|jjddd} y| jd\} } Wntk r"d} YnX|jj| |jj} |jj| |jj} | rl| rl|jjd| | | f}n0| r|jjd| | f}n|jjd| | f}n|j}|j||||jjB||jj B|f|j_!dS)Nr<rz://r zlocal repositoryz%s %sZ InReleasezStructure of %sZReleasezDescription of %sz Release.gpgzDescription signature of %sZPackageszAvailable packages from %sZSourceszAvailable sources from %sZTranslationIndexzAvailable translations from %sz Translation--_z Translations for %s (%s) from %szTranslations for %s from %szTranslations (%s) from %sry)"r=r>r?r@rAr rBrCrDrErFrGrHrIrJrKrLrMrNrzr{r|r/r}rP startswith ValueErrorrwZget_localised_namelocalerxrOrQrR)r.rSrTrUr>rVhostdistdescZrepoZ lang_codeZ region_codeZlangZregionrrrrW!sz                    z,DaemonAcquireRepoProgress._emit_acquire_itemN)rr)rr)r7r8r9r:r&rprcrWrrrrrus  ruc@szeZdZdZdddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZdS) ra Forks and executes a given method in the child process while monitoring the output and return state. During the run() call the mainloop will be iterated. Furthermore a status file descriptor is available to communicate with the child process. 2rcCsv||_d|_d|_||_||_d|_d|_d|_tj \|_ |_ t tdrftj |j dtj |j dd|_d|_dS)Nr<rr set_inheritableTry)r/r>r#r*r, _child_exit last_activity child_pidospipestatus_parent_fdstatus_child_fdhasattrroutput _line_buffer)r.r/rrrrrr&s zDaemonForkProgress.__init__cCs |j|S)N) start_update)r.rrr __enter__szDaemonForkProgress.__enter__cCs |jdS)N) finish_update)r.etypeZevalueZetbrrr__exit__szDaemonForkProgress.__exit__cCs4tjdtj|j_d|j_tj|_tj|_ dS)Nz Start updateT) r4debugr STATUS_COMMITTINGr/r> term_attachedtimerZ start_time)r.rrrrs    zDaemonForkProgress.start_updatecCs d|j_dS)z$Callback at the end of the operationFN)r/r)r.rrrrsz DaemonForkProgress.finish_updatecGs ||tjdtjddS)zCall the given method or function with the corrsponding arguments in the child process. This method should be replace in subclasses. g?rN)rsleepr_exit)r.methodargsrrr_childs zDaemonForkProgress._childc +OsRtjdd}|jjrty:tj|jjtjtjBtjB}t j |}t j |t j Wn"tt jfk rrd|j_YnX|j}|dkrtj|jz:y|j|j||Wntk rtjYnXWdtjdtjtjjXn||_tj|jtjd|g}t j!t j"Bt j#B}|jjr@|j$t j%|t j&||j'|j$t j%|j(t j&||j)||j$t j*t j&||j+|j$t j%|jt j&t j!|j,x|j-dkrt j.j/qWx|D]}t j0|qW|ryt j1|t j2|Wnt jk rYnXxB|j(|j|gD]0} ytj| Wntt3fk r>YnXqWtj4|j-S)zxSetup monitoring, fork and call the self._child() method in the child process with the given arguments. ZRunNr<rr z Child pid: %sry)5r4rr/terminalropenO_RDWRO_NOCTTY O_NONBLOCKtermiosZ tcgetattrttyZsetrawZTCSANOWOSErrorerror_forkcloser _setup_childr Exception traceback print_excrrrr?PackageManager RESULT_FAILEDrrr IO_INZIO_ERRZIO_HUPriZ io_add_watchZPRIORITY_HIGH_IDLE_copy_io master_fd_copy_io_masterZchild_watch_add_on_child_exit_on_status_updaterrjrlZ source_removeZ tcsetattrZ TCSADRAIN TypeError WEXITSTATUS) r.rkwargsZ terminal_fdZ terminal_attrpidZwatchersflagsr[fdrrrrunsp                   zDaemonForkProgress.runcCstjd|||_dS)NzChild exited: %sF)r4rr)r.r conditionrrrrs z!DaemonForkProgress._on_child_exitcCsdS)zCallback for changes on the status file descriptor. The method has to return True to keep the monitoring alive. If it returns False the monitoring will stop. Replace this method in your subclass if you use the status fd. Fr)r.sourcerrrrrsz$DaemonForkProgress._on_status_updatecCstj\}|_|S)zgFork and create a master/slave pty pair by which the forked process can be controlled. )rforkptyr)r.rrrrr szDaemonForkProgress._forkcCsdd}tjtj|tjt_tjdtjd<t j t j d|j j rVtj d|j j |j jsxtj ddtj d dn tj d d |j jrtj d |j jtj d dtjtjkrtj ddn|j jstj d d|j jrtjjd|j jd|j j|j jf}tjjd|dS)z+Setup the environment of the child process.cSstjtjjdS)N)rrr?rr)Zsignumframerrrinterrupt_handlersz:DaemonForkProgress._setup_child..interrupt_handler1ZDPKG_UNTRANSLATED_MESSAGESCZLANGZAPT_LISTCHANGES_FRONTENDZnoneZAPT_LISTBUGS_FRONTENDZTERMlinuxZ DEBCONF_PIPEZDEBIAN_FRONTENDZ passthroughZ DEBCONF_DEBUG.ZnoninteractivezAcquire::http::Proxyzaptdaemon role='%s' sender='%s'zCommandLine::AsStringN)signalSIGINTsys__excepthook__ excepthookr quitrenvironr setlocaleLC_ALLr/putenvrZdebconfr4levelloggingDEBUGZ http_proxyr?configrZZroleZsender)r.rcmdrrrrs2        zDaemonForkProgress._setup_childcCs|tjkrtj|_ytj|d}Wntk rBtjddSX|j dd}|dkrt j t d|j }|rtj||j|d7_d|_ n|j |7_ |rytj||Wntk rtjdYnXdSytj|Wntk rYnXd S) Nr zFaild to read from masterTzUTF-8ignore r<z'Failed to write to controlling terminalF)r rrrrreadrr4rdecoderesubREGEX_ANSI_ESCAPE_CODEr log_terminalrwriter)r.rrtargetZ char_bytecharlinerrrr?s4     z"DaemonForkProgress._copy_io_masterc Csp|tjkrbytj|d}tj|j|Wntk r<Yn&X|jjr^|jj r^d|j_ d|j_dStj |dS)Nr FT) r rrrrrrr/rsconfig_file_conflictconfig_file_conflict_resolutionr)r.rrrrrrr_s   zDaemonForkProgress._copy_ioN)rr)r7r8r9r:r&rrrrrrrrrrrrrrrrrys   K . c@seZdZdZddZdS)rzPerforms a lintian call.cCsytj|jjgWntk r(YnXtj|jjtj|jjtj ddkr\d}nd}tj dt j j dd}tj||dd d ||tjddS) Nr Zdebianzdebian/aptdaemonzubuntu/aptdaemonHOMEzDir::Bin::Lintianz/usr/bin/lintianz--no-cfgz--fail-on-warningsz --profile)r setgroupsr/gidrsetgidsetuidZuidplatformrunsetenvr?rZ find_fileexeclpr)r.pathZprofileZ lintian_pathrrrrus   zDaemonLintianProgress._childN)r7r8r9r:rrrrrrqsc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)rz>Progress to execute APT package operations in a child process.cCstj|tjjdS)N)rrr status_lockrelease)r.rrrrs z"DaemonInstallProgress.start_updatecCstj|tj|jtjdS)z$Callback at the end of the operationN)rrr Z wait_for_lockr/r)r.rrrrs z#DaemonInstallProgress.finish_updatec Cs:y|j|j}WntjtjjYn Xtj|dS)N)Z do_installrrrr?rr)r.Zpmresrrrrs zDaemonInstallProgress._childc CsRtjdd}y6x0|jds@tj|_|tj|djdd7}qWWn dSy|jdd \}}}}Wnt k r|d SX|j }|d kr|j ||nz|d krt j d |} | r| jd| jd} } |j| | n<|dkr|jdrtj} n |jdrtj} n|jdr$tj} n|jdr8tj} n|jdrLtj} n|jdr`tj} n|jdrttj} n|jdrtj} n~|jdrtj} nj|jdrtj} nV|jdrtj} nB|jdrtj} n.|jdrtj} n|jdrtj} ntj } |j!|t"|| |jt#tjkrN|j$rNtj%dt#tj&|j$dd S) z)Parse messages from APT on the status fd.UpdateInterfacer<rr zUTF-8rF:TZpmerrorZ pmconffilez\s*'(.*)'\s*'(.*)'.*r~ZpmstatusZ InstallingZ InstalledZ ConfiguringzPreparing to configurezPreparing for removal ofZRemovingZRemovedzPreparing to completely removezCompletely removingzCompletely removedZ UnpackingZ PreparingzNoting disappearance ofZRunningz#Killing child since timeout of %s s)'r4rendswithrrrrrr{rstrip_errorrmatchgroup _conffilerr PKG_INSTALLINGZ PKG_INSTALLEDPKG_CONFIGURINGZPKG_PREPARING_CONFIGUREZPKG_PREPARING_REMOVE PKG_REMOVINGZ PKG_REMOVEDZPKG_PREPARING_PURGE PKG_PURGINGZ PKG_PURGEDZ PKG_UNPACKINGZPKG_PREPARING_INSTALLPKG_DISAPPEARINGPKG_RUNNING_TRIGGERZ PKG_UNKNOWN_status_changedr)INSTALL_TIMEOUTrcriticalkill) r.rrZ status_msgr>pkgr1Z message_rawmessagernewold status_enumrrrrsr                     z'DaemonInstallProgress._on_status_updatecCsxtjd||||j|d|j|j}|j|krFt||j_||_tj|}|jj ||}||j_ ||f|j_ dS)z%Callback to update status informationzAPT status: %s, %s, %srN) r4rr*r,r#r2r/r Zget_package_status_from_enumr}r`Zprogress_package)r.rr1r r#rrVrrrrs    z%DaemonInstallProgress._status_changedcCstjd||f||f|j_d|j_tj|j_x|jjrHtj j q2Wtj d|jj |jj dkrvt j|jdn|jj dkrt j|jdd|j_ d|j_tj|j_dS) z#Callback for a config file conflictzConfig file prompt: '%s' (%s)TzSending config file answer: %sreplacesy Zkeepsn N)r4r5r/rrsr Z!STATUS_WAITING_CONFIG_FILE_PROMPTr>r rjrlrrrrrr)r.Zcurrentrrrrrs        zDaemonInstallProgress._conffilecCstjd||fdS)zCallback for an errorz%s: %sN)r4r)r.rrVrrrrszDaemonInstallProgress._errorN) r7r8r9r:rrrrrrrrrrrrsBc@s*eZdZdZd ddZddZddZd S) rz9Progress handler for a local Debian package installation.r0cCstj||||dS)N)rr&)r.r/rrrrrr&sz"DaemonDpkgInstallProgress.__init__cCsftjddt|jg}|jtjjd|jjs>|jddg|jd|gtj tjdf|dS)NzDir::Bin::DPkgz --status-fdz DPkg::Optionsz--force-confdefz--force-confoldz-i) r?rstrrextend value_listr/rrr)r.Zdebfilerrrrrs  z DaemonDpkgInstallProgress._childc Cs:tjdd}y,x&|jds6|tj|djdd7}qWWn dSydd |jd d D}Wntk rtd SX|d dkr|ddkr|j|d|d nB|ddkrt j d|d }|r|j d|j d}}|j ||nN|d dkr6yt |d}Wntk r"tj}YnX|j|dd|d S)Nrr<rr zUTF-8rFcSsg|] }|jqSr)r)rrrrrr$sz?DaemonDpkgInstallProgress._on_status_update..rrTrr>r~rconffilez\s*'(.*)'\s*'(.*)'.*Z processingr0)r4rrrrrr{rrrrrrMAP_DPKG_STAGEKeyErrorr Z PKG_UNKONWNr) r.rrZ status_rawr>rrrr rrrrs4       z+DaemonDpkgInstallProgress._on_status_updateN)r0r0)r7r8r9r:r&rrrrrrr s  c@seZdZdZddZdS)rz-Progress handler for dpkg --confiure -a call.cCs\tjddt|jddg}|jtjjd|jjsB|jddgtj tjdf|dS) NzDir::Bin::Dpkgz --status-fdz --configurez-az Dpkg::Optionsz--force-confdefz--force-confoldzDir::Bin::DPkg) r?rr rr r r/rrr)r.rrrrrDs  z DaemonDpkgRecoverProgress._childN)r7r8r9r:rrrrrr@sc@seZdZdZddZdS)DaemonDpkgReconfigureProgressz+Progress handler for dpkg-reconfigure call.cCs8dg}|dkr|jd|g|j|tjd|dS)N/usr/sbin/dpkg-reconfiguredefaultz --priority)r)r rr)r.ZpackagesZpriorityrrrrrQs  z$DaemonDpkgReconfigureProgress._childN)r7r8r9r:rrrrrrMsr)rrrrrrriX)8r: __author____all__rrrrrrrrrrrr?Zapt.progress.baser"Z apt.debfileZ gi.repositoryr r<r r Zloopr ZutilsrrZ getLoggerr4rrrrrrrrZ PKG_UPGRADINGrchrrr#r$r%rr;rruobjectrrrrrrrrrrs^          9 zy5