U
    óûˆc¥L  ã                   @   sJ  d dl mZ d dlmZ d dlmZmZ d dlmZ d dl	m
Z
mZmZmZ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!Z!d dl"Z"G dd„ deƒZ#G dd„ dej$ƒZ%G dd„ deeƒZ&e'dkrFd dl(Z(e( )¡ Z*dZ+d dl,m-Z- e-e+e*ƒZ.e&e.dƒZ/e/ 0dgg g ¡ e 1¡  dS )é    )Úprint_function)ÚGtk)ÚclientÚerrors)Úinline_callbacks)ÚAptCancelButtonÚAptConfigFileConflictDialogÚAptDetailsExpanderÚAptMediumRequiredDialogÚAptProgressBar)	ÚEXIT_SUCCESSÚEXIT_FAILEDÚSTATUS_COMMITTINGÚSTATUS_DOWNLOADINGÚSTATUS_DOWNLOADING_REPOÚSTATUS_FINISHEDÚget_error_description_from_enumÚget_error_string_from_enumÚget_status_string_from_enum)ÚInstallBackend)ÚUnitySupport)ÚBuilderDialog)ÚgettextNc                       s*   e Zd ZdZd‡ fdd„	Zdd„ Z‡  ZS )	ÚUpdateManagerExpanderaF  An AptDetailsExpander which can be used with multiple terminals.

       The default AptDetailsExpander will shrink/hide when its transaction
       finishes. But here we want to support "chaining" transactions. So we
       override the status-changed handler to only do that when we are
       running the final transaction.TFc                    s   t ƒ  ||¡ || _d S ©N)ÚsuperÚ__init__Úfinal)ÚselfÚtransactionÚterminalr   ©Ú	__class__© úO/usr/lib/python3/dist-packages/UpdateManager/backend/InstallBackendAptdaemon.pyr   -   s    zUpdateManagerExpander.__init__c                 C   s¼   |t tfkr2|  d¡ | j ¡  | jr¸| j ¡  n†|tkr|| j ¡  | jr`| j ¡  |  d¡ q¸| jr¸|  	d¡ |  d¡ n<| jr¸|t
kr¸| j ¡  | jr¤| j ¡  |  d¡ |  	d¡ d S ©NTF)r   r   Zset_sensitiveZdownload_scrolledÚshowr    Úhider   r   Zset_expandedr   )r   ÚtransÚstatusr#   r#   r$   Ú_on_status_changed1   s&    







z(UpdateManagerExpander._on_status_changed)TF)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r*   Ú__classcell__r#   r#   r!   r$   r   %   s   r   c                       s8   e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd	„ Z‡  ZS )
ÚAptStackedProgressBaraq   A GtkProgressBar which represents the state of many aptdaemon
    transactions.

    aptdaemon provides AptProgressBar for the state of *one* transaction to
    be represented in a progress bar. This widget creates one of those per
    containing transaction, and scales its progress to the given ratio, so
    one progress bar can show the state of many transactions.c                    s    d| _ g | _|| _tƒ  ¡  d S )Nr   )Úcurrent_max_progressÚprogress_barsÚunityr   r   )r   r3   r!   r#   r$   r   P   s    zAptStackedProgressBar.__init__c                 C   s|   d|  krdksn t ‚t|ƒ}| j |¡ | j|_|  j|7  _| jdkrTd| _| j|_| d| j¡ | d| j	¡ d S )Nr   é   znotify::fractionznotify::text)
ÚAssertionErrorr   r2   Úappendr1   ÚminÚmaxÚconnectÚ_update_progressÚ_update_text)r   r(   Zmax_progressZprogressr#   r#   r$   Úadd_transactionW   s    
z%AptStackedProgressBar.add_transactionc                 C   s@   |j |j }|| ¡  }|j| }|  |¡ | j |d ¡ d S )Néd   )r8   r7   Zget_fractionZset_fractionr3   Úset_progress)r   Úinner_progressÚdataZdeltaZposition_in_deltaZnew_progressr#   r#   r$   r:   f   s
    

z&AptStackedProgressBar._update_progressc                 C   s   |   | ¡ ¡ d S r   )Zset_textZget_text)r   r?   r@   r#   r#   r$   r;   m   s    z"AptStackedProgressBar._update_text)	r+   r,   r-   r.   r   r<   r:   r;   r/   r#   r#   r!   r$   r0   G   s
   r0   c                   @   s®   e Zd ZdZdd„ Zdd„ Zedd„ ƒZdd	„ Zd
d„ Z	edd„ ƒZ
dd„ Zedd„ ƒZedd„ ƒZdd„ Zdd„ Zed&dd„ƒZdd„ Zdd„ Zd d!„ Zd"d#„ Zd$d%„ ZdS )'ÚInstallBackendAptdaemonzCMakes use of aptdaemon to refresh the cache and to install updates.c                 C   sh   t  | ||¡ tj |jd¡}t | ||d¡ t ¡ | _t	ƒ | _
d | _d | _d | _d | _d | _d | _d S )Nzgtkbuilder/UpdateProgress.uiZpane_update_progress)r   r   ÚosÚpathÚjoinZdatadirr   r   Z	AptClientr   r3   Ú_expanded_sizeÚbutton_cancelÚtrans_failed_msgÚprogressbarZ_active_transactionÚ	_expander)r   Úwindow_mainÚactionZui_pathr#   r#   r$   r   t   s    
ÿ
ÿ
z InstallBackendAptdaemon.__init__c                 C   s@   | j r8| j  ¡ r8z| j  ¡  W n tk
r2   Y nX dS dS d S r%   )rF   Zget_sensitiveZclickedÚ	Exception)r   r#   r#   r$   Úclose„   s    zInstallBackendAptdaemon.closec              
   c   sˆ   z,| j jddV }|  || jtdƒd¡V  W nV tjk
rX   | j| jddddd Y n, tk
r‚   | j| jddddd ‚ Y nX dS )zRefresh the package listT©Údeferu   Checking for updatesâ€¦FN©Ú
authorizedÚsuccessÚerror_stringÚ
error_desc)	r   Úupdate_cacheÚ_show_transactionÚACTION_UPDATEÚ_r   ÚNotAuthorizedErrorÚ_action_donerL   )r   r(   r#   r#   r$   Úupdate   s*    
 ÿ
  þ
  þzInstallBackendAptdaemon.updatec                 C   sP   t |jjƒ}t|jjƒ}| jr2d}|d | j }nd }| j|dd|||d d S )NTÚ
F©rQ   rR   rS   rT   Útrans_failed)r   ÚerrorÚcoder   rG   rZ   )r   r(   rK   rS   rT   r^   r#   r#   r$   Ú_show_transaction_error¡   s     üz/InstallBackendAptdaemon._show_transaction_errorc                 C   s$   |t kr|  ||¡ d S |  ¡  d S r   )r   ra   Ú_apt_update_oem)r   r(   r)   rK   r#   r#   r$   Ú_update_next_package¯   s    z,InstallBackendAptdaemon._update_next_packagec              
   c   sJ  | j s
t‚| j  ¡ }d|› d}zXtj |¡rv| jj|dV }| j rL| j}n| j	}| j
|| jtdƒd|d| j dV  W nÌ tjk
r¤   | j| jddd d d	 Y n¢ tjk
rÐ } zt|ƒ| _W 5 d }~X Y nv tjk
r } z(| ¡ d
krò‚ | j| jddd d d	 W 5 d }~X Y n. tk
rD   | j| jddd d d	 ‚ Y nX d S )Nz/etc/apt/sources.list.d/z.list)Zsources_listõ   Installing updatesâ€¦Tçš™™™™™¹?©Úon_finished_handlerÚprogress_bar_maxFrP   ú"org.freedesktop.DBus.Error.NoReply)Ú_oem_packages_to_updater5   ÚpoprB   rC   Úexistsr   rU   rc   Ú_on_finishedrV   ÚACTION_PRE_INSTALLrX   Ú_len_oem_updatesr   rY   rZ   ÚTransactionFailedÚstrrG   ÚdbusÚDBusExceptionÚget_dbus_namerL   )r   ÚelemZsources_list_filer(   Zfinished_handlerÚer#   r#   r$   rb   µ   sZ    

ÿ û  þ
  þ  þz'InstallBackendAptdaemon._apt_update_oemc                 C   sJ   |t kr|  ||¡ d S |j\}}}}}}t|ƒ| _t|ƒ| _|  ¡  d S r   )r   ra   ZpackagesÚsetrj   Úlenro   rb   )r   r(   r)   rK   ÚinstallrX   r#   r#   r$   Ú_update_oemÞ   s    

z#InstallBackendAptdaemon._update_oemc              
   c   sD  t |ƒt |ƒB | _z`|s:|s:| j| jddd d d d W d S |rp| jj|ddV }| j|| jtdƒd| jddV  W nÌ t	j
k
rž   | j| jddd d d Y n¢ t	jk
rÊ } zt|ƒ| _W 5 d }~X Y nv tjk
r } z(| ¡ d	krì‚ | j| jddd d d W 5 d }~X Y n. tk
r>   | j| jddd d d ‚ Y nX d S )
NTr]   rN   rd   re   rf   FrP   ri   )rw   Zall_oem_packagesrZ   rn   r   Zinstall_packagesrV   rX   rz   r   rY   rp   rq   rG   rr   rs   rt   rL   )r   Zpkgs_install_oemZpkgs_upgrade_oemr(   rv   r#   r#   r$   Ú
commit_oemî   sb      ýÿ û  þ
  þ  þz"InstallBackendAptdaemon.commit_oemc           	   
   c   s  zDg  } }}| j j||||||ddV }|  || jtdƒd¡V  W nÊ tjk
rp   | j| jddddd Y n  tjk
rœ } zt	|ƒ| _
W 5 d}~X Y nt tjk
râ } z(| ¡ dkr¼‚ | j| jddddd W 5 d}~X Y n. tk
r   | j| jddddd ‚ Y nX dS )z)Commit a list of package adds and removesTrN   rd   FNrP   ri   )r   Zcommit_packagesrV   ZACTION_INSTALLrX   r   rY   rZ   rp   rq   rG   rr   rs   rt   rL   )	r   Zpkgs_installZpkgs_upgradeZpkgs_removeZ	reinstallÚpurgeZ	downgrader(   rv   r#   r#   r$   Úcommit  sR         þ
 ÿ
  þ
  þ  þzInstallBackendAptdaemon.commitc                 C   s   |  |¡ d S r   )Ú	set_label)r   r(   ZdetailsÚlabel_detailsr#   r#   r$   Ú_on_details_changed3  s    z+InstallBackendAptdaemon._on_details_changedc                 C   s8   |  t|ƒ¡ |tkr4| jr4| jj ¡ r4|  | j¡ d S r   )r~   r   r   rI   r    Úget_visibleÚ_resize_to_show_details)r   r(   r)   r   r#   r#   r$   r*   6  s
    
ÿz*InstallBackendAptdaemon._on_status_changedr4   Nc                 c   sš  |d kr| j }| j |¡ | jsDt| jƒ| _| j ¡  | j | j¡ | j 	||¡ | j
rf| j
 |¡ n"t|ƒ| _
| j
 ¡  | j | j
¡ || jkr | j
 tj¡ |r | jst|ƒ| _| j d¡ | j d¡ | j ¡  | j d| j¡ | j | j¡ | j ¡  n| j |¡ || jk| j_n| jr2| j ¡  | d| j| j¡ | d| j| j¡ | d||¡ | d| j ¡ | d| j!¡ | "d¡V  | #¡ V  d S )	NTznotify::expandedzstatus-details-changedzstatus-changedZfinishedzmedium-requiredzconfig-file-conflictZgnome)$rm   Zlabel_headerr~   rH   r0   r3   r&   Zprogressbar_slotÚaddr<   rF   Zset_transactionr   Zbutton_cancel_slotrW   r   Z
STOCK_STOPrI   r   Zset_vexpandZset_hexpandZshow_allr9   Ú_on_expandedZexpander_slotrn   r   Z_expander_slotr'   r€   r   r*   Ú_on_medium_requiredÚ_on_config_file_conflictZset_debconf_frontendÚrun)r   r(   rK   ÚheaderZshow_detailsrh   rg   r#   r#   r$   rV   >  sR    





ÿ

ÿ
ÿz)InstallBackendAptdaemon._show_transactionc                 C   sv   |  ¡ s*|j ¡ | j ¡ f| _| j ¡  nH| jrh| j\}\}}||j ¡ krX|  |¡ qr| j ||¡ n
|  |¡ d S r   )	Úget_expandedr    r   rJ   Úget_sizerE   Zend_user_resizabler‚   Úbegin_user_resizable)r   ÚexpanderZparamZterm_visibleZstored_widthZstored_heightr#   r#   r$   r„   t  s    ÿÿz$InstallBackendAptdaemon._on_expandedc           
      C   sŒ   |  ¡ rˆ| j ¡ \}}| ¡ j}| ¡ j}|j ¡ rj|j ¡ d }|j 	¡ d }|| | }|| | }	n|d }|d }	| j 
||	¡ dS )a¯  Resize the window to show the expanded details.

        Unfortunately the expander only expands to the preferred size of the
        child widget (e.g showing all 80x24 chars of the Vte terminal) if
        the window is rendered the first time and the terminal is also visible.
        If the expander is expanded afterwards the window won't change its
        size anymore. So we have to do this manually. See LP#840942
        éP   é   r=   éÈ   N)r‰   rJ   rŠ   Zget_allocationÚwidthZheightr    r   Zget_char_widthZget_char_heightr‹   )
r   rŒ   Z	win_widthZ
win_heightZ	exp_widthZ
exp_heightZterminal_widthZterminal_heightZ	new_widthZ
new_heightr#   r#   r$   r‚   ‰  s    	


z/InstallBackendAptdaemon._resize_to_show_detailsc                 C   sB   t ||| jƒ}| ¡ }| ¡  |tjjkr6| |¡ n| ¡  d S r   )	r
   rJ   r‡   r'   r   ÚResponseTypeZOKZprovide_mediumZcancel)r   r   ZmediumZdriveÚdialogÚresr#   r#   r$   r…      s    z+InstallBackendAptdaemon._on_medium_requiredc                 C   sH   t ||| jƒ}| ¡ }| ¡  |tjjkr8| |d¡ n| |d¡ d S )NÚreplaceZkeep)r   rJ   r‡   r'   r   r‘   ZYESZresolve_config_file_conflict)r   r   ÚoldÚnewr’   r“   r#   r#   r$   r†   ©  s    z0InstallBackendAptdaemon._on_config_file_conflictc              
   C   s    d }d }d}|t krDt|jjƒ}t|jjƒ}| jrDd}|d | j }| j d¡ |tk}z| j	|d||||d W n( t
k
rš   | j	|d|||d Y nX d S )NFTr\   éÿÿÿÿr]   rP   )r   r   r_   r`   r   rG   r3   r>   r   rZ   Ú	TypeError)r   r(   r)   rK   rS   rT   r^   Z
is_successr#   r#   r$   rm   ²  s4      ý
  þz$InstallBackendAptdaemon._on_finished)r4   N)r+   r,   r-   r.   r   rM   r   r[   ra   rc   rb   rz   r{   r}   r€   r*   rV   r„   r‚   r…   r†   rm   r#   r#   r#   r$   rA   q   s2   

(
'
  þ5		rA   Ú__main__z/usr/share/update-manager)ÚUpdateManagerZ2vcard)2Z
__future__r   Zgi.repositoryr   Z	aptdaemonr   r   rO   r   Zaptdaemon.gtk3widgetsr   r   r	   r
   r   Zaptdaemon.enumsr   r   r   r   r   r   r   r   r   ZUpdateManager.backendr   ZUpdateManager.UnitySupportr   ZUpdateManager.Dialogsr   r   rX   rr   rB   r   ZProgressBarr0   rA   r+   ZmockZMockZoptionsZdata_dirZUpdateManager.UpdateManagerrš   ZappÚbr}   Úmainr#   r#   r#   r$   Ú<module>   s2   ,
"*  ^


