o
    =SbL                     @   sL  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rd 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 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v r| d | j  | jr| j  d S d S |tkrE| j  | jr4| j  | d d S | jrC| 	d | d d S d S | jre|t
krg| j  | jrY| j  | d | 	d d S d S 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J  J t |}| j| | j|_|  j|7  _| jdkr+d| _| j|_|d| j |d| j d S )Nr      znotify::fractionznotify::text)	r   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 )Nd   )r7   r6   Zget_fractionZset_fractionr3   set_progress)r   inner_progressdataZdeltaZposition_in_deltaZnew_progressr#   r#   r$   r9   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;   r9   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 r| j  rz| j   W dS  ty   Y dS w dS r%   )rE   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 dS  tjy.   | j| jddddd Y dS  ty@   | j| jddddd  w )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_donerK   )r   r(   r#   r#   r$   update   s$   
zInstallBackendAptdaemon.updatec                 C   sP   t |jj}t|jj}| jrd}|d | j }nd }| j|dd|||d d S )NT
FrP   rQ   rR   rS   trans_failed)r   errorcoder   rF   rY   )r   r(   rJ   rR   rS   r]   r#   r#   r$   _show_transaction_error   s   
z/InstallBackendAptdaemon._show_transaction_errorc                 C   s$   |t kr| || d S |   d S r   )r   r`   _apt_update_oem)r   r(   r)   rJ   r#   r#   r$   _update_next_package   s   z,InstallBackendAptdaemon._update_next_packagec              
   c   sN   | j sJ | j  }d| d}z0tj|r?| jj|dV }| j r'| j}n| j}| j	|| j
tdd|d| j dV  W d S W d S  tjyW   | j| j
ddd d d	 Y d S  tjyo } zt|| _W Y d }~d S d }~w tjy } z| d
kr~ | j| j
ddd d d	 W Y d }~d S d }~w ty   | j| j
ddd d d	  w )Nz/etc/apt/sources.list.d/z.list)Zsources_list   Installing updates…T皙?on_finished_handlerprogress_bar_maxFrO   "org.freedesktop.DBus.Error.NoReply)_oem_packages_to_updatepoprA   rB   existsr   rT   rb   _on_finishedrU   ACTION_PRE_INSTALLrW   _len_oem_updatesr   rX   rY   TransactionFailedstrrF   dbusDBusExceptionget_dbus_namerK   )r   elemZsources_list_filer(   Zfinished_handlerer#   r#   r$   ra      sV   

z'InstallBackendAptdaemon._apt_update_oemc                 C   sJ   |t kr| || d S |j\}}}}}}t|| _t|| _|   d S r   )r   r`   Zpackagessetri   lenrn   ra   )r   r(   r)   rJ   installrW   r#   r#   r$   _update_oem   s   

z#InstallBackendAptdaemon._update_oemc              
   c   sH   t |t |B | _z4|s|s| j| jddd d d d W d S |r<| jj|ddV }| j|| jtdd| jddV  W d S W d S  t	j
yT   | j| jddd d d Y d S  t	jyl } zt|| _W Y d }~d S d }~w tjy } z| d	kr{ | j| jddd d d W Y d }~d S d }~w ty   | j| jddd d d  w )
NTr\   rM   rc   rd   re   FrO   rh   )rv   Zall_oem_packagesrY   rm   r   Zinstall_packagesrU   rW   ry   r   rX   ro   rp   rF   rq   rr   rs   rK   )r   Zpkgs_install_oemZpkgs_upgrade_oemr(   ru   r#   r#   r$   
commit_oem   sZ   
z"InstallBackendAptdaemon.commit_oemc           	   
   c   s   z#g  } }}| j j||||||ddV }| || jtddV  W dS  tjy:   | j| jddddd Y dS  tjyR } zt	|| _
W Y d}~dS d}~w tjyw } z| dkra | j| jddddd W Y d}~dS d}~w ty   | j| jddddd  w )z)Commit a list of package adds and removesTrM   rc   FNrO   rh   )r   Zcommit_packagesrU   ZACTION_INSTALLrW   r   rX   rY   ro   rp   rF   rq   rr   rs   rK   )	r   Zpkgs_installZpkgs_upgradeZpkgs_removeZ	reinstallpurgeZ	downgrader(   ru   r#   r#   r$   commit  sB   

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   sD   | t| |tkr| jr| jj r | | j d S d S d S d S r   )r}   r   r   rH   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 u r| j }| j| | js#t| j| _| j  | j| j | j	|| | j
r4| j
| nt|| _
| j
  | j| j
 || jkrQ| j
tj |r| jst|| _| jd | jd | j  | jd| j | j| j | j  n| j| || jk| j_n| jr| 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)$rl   Zlabel_headerr}   rG   r0   r3   r&   Zprogressbar_slotaddr;   rE   Zset_transactionr   Zbutton_cancel_slotrV   r   Z
STOCK_STOPrH   r   Zset_vexpandZset_hexpandZshow_allr8   _on_expandedZexpander_slotrm   r   Z_expander_slotr'   r   r~   r*   _on_medium_required_on_config_file_conflictZset_debconf_frontendrun)r   r(   rJ   headerZshow_detailsrg   rf   r#   r#   r$   rU   >  sT   








z)InstallBackendAptdaemon._show_transactionc                 C   s|   |  s|j | j f| _| j  d S | jr7| j\}\}}||j kr.| | d S | j|| d S | | d S r   )	get_expandedr    r   rI   get_sizerD   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   |  rF| j \}}| j}| j}|j r5|j d }|j	 d }|| | }|| | }	n|d }|d }	| j
||	 dS 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   rI   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   sD   t ||| j}| }|  |tjjkr|| d S |  d S r   )	r
   rI   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   sJ   t ||| j}| }|  |tjjkr||d d S ||d d S )NreplaceZkeep)r   rI   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r"t|jj}t|jj}| jr"d}|d | j }| jd |tk}z| j	|d||||d W d S  t
yN   | j	|d|||d Y d S w )NFTr[   r\   rO   )r   r   r^   r_   r   rF   r3   r=   r   rY   	TypeError)r   r(   r)   rJ   rR   rS   r]   Z
is_successr#   r#   r$   rl     s.   z$InstallBackendAptdaemon._on_finished)r4   N)r+   r,   r-   r.   r   rL   r   rZ   r`   rb   ra   ry   rz   r|   r   r*   rU   r   r   r   r   rl   r#   r#   r#   r$   r@   q   s4    

(
'
5		r@   __main__z/usr/share/update-manager)UpdateManagerZ2vcard)2Z
__future__r   Zgi.repositoryr   Z	aptdaemonr   r   rN   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   rW   rq   rA   r   ZProgressBarr0   r@   r+   ZmockZMockoptionsZdata_dirZUpdateManager.UpdateManagerr   Zappbr|   mainr#   r#   r#   r$   <module>   s6   ,
"*  ^

