File: //lib/python3/dist-packages/UpdateManager/backend/__pycache__/__init__.cpython-38.pyc
U
ρ�f�> � @ s` d Z ddlmZ ddlZe�dd� ddlmZmZmZ ddl m
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlmZ dd lmZ dd
lmZ dd lmZmZ dd lmZ G d
d� de�Ze
j � d��rde
j k�rzddl m Z W n e k
�r e �!d� Y nX e
j � d��rTde
j k�rTzddl"m"Z" W n e k
�rR e �!d� Y nX dd� Z#dS )z2Integration of package managers into UpdateManager� )�absolute_importN�Snapd�1)�GLib�Gtkr )�Cache)�gettext)�Thread)�MyCache)�
inhibit_sleep�get_dist_version)�Dialogc @ sr e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd dd�ZdS ) �InstallBackendr � � c C s t �| |� || _d | _d S �N)r
�__init__�action� sleep_cookie)�self� window_mainr � r �@/usr/lib/python3/dist-packages/UpdateManager/backend/__init__.pyr s zInstallBackend.__init__c C s� dt jd< t� | _| j| jkr�| jj}t| jjj d�}g }g }| jj
D ]B}|| }|| }|j rr|j sr|�
|� qF|jrF|jrF|�
|� qF| �||� n�| j| jk�rxg }g } g }
t| jjj d�}| jjD ]�}z�|j �r||j j �s|j} |jr�| d7 } |�
| � nF|j�r*||j j�r*| �
|j� n"|j�rL||j j �rL|
�
|j� W q� tk
�rd Y q�X q�| �|| |
� n| �� d S )NZnoneZAPT_LISTCHANGES_FRONTEND)�rootdirz#auto)�os�environr r r �ACTION_PRE_INSTALLr �cacher r Zoem_metapackagesZmarked_install� is_installed�appendZmarked_upgradeZ
is_upgradable�
commit_oem�ACTION_INSTALL�name�is_auto_installed�
marked_delete�KeyError�commit�update) r Z
unfresh_cacheZ fresh_cache�pkgs_install_oem�pkgs_upgrade_oem�pkgZ unfresh_pkgZ fresh_pkg� pkgs_install� pkgs_upgrade� pkgs_removeZpkgnamer r r �start% sZ
� �
�
�
�zInstallBackend.startc C s t �dS )z(Run a update to refresh the package listN��NotImplementedError�r r r r r' Y s zInstallBackend.updatec C s | j | jdddddd� dS )z Install these OEM packages TN)�
authorized�success� error_string�
error_desc� trans_failed)� _action_doner )r r( r) r r r r ] s �zInstallBackend.commit_oemc C s t �dS )zCommit the cache changes Nr/ )r r+ r, r- r r r r& d s zInstallBackend.commitc
C s� i }i }dt � }z�tdd�}t�|�}|�� |d D ]4}|d | }|�dd �}|�d|�} || f||< q8|d D ]$}|d | }
|
�d|�} | ||< qvW n0 tk
r� } zt�d | � W 5 d } ~ X Y nX ||fS )
Nzstable/ubuntu-z0/usr/share/ubuntu-release-upgrader/deb2snap.json�rZseeded�deb�
to_channelZunseeded� from_channelz%error reading deb2snap.json file (%s)) r �open�json�load�close�get� Exception�logging�debug)
r � seeded_snaps�unseeded_snapsZ curr_channelZd2s_fileZd2s�snapZseedr9 r: Zunseedr; �er r r �get_snap_seedsh s,
�
z InstallBackend.get_snap_seedsc
C s z@| j jd kr td �| j _n| j j�d � | j j�� | j j}W n8 tk
rx } zt�d| � g W Y �S d }~X Y nX g }| �� \}}|� � D ]l\}\}}||kr�|| j
r�d}|| � � |� � D ] } | j
r�| j
r�| js�d} q�q�|�� |r�|�|� q�|S )Nzerror reading cache (%s)TF)r r r
r< Z
_initDepCacherA rB rC rH �itemsr Z mark_deleteZ get_changesr$ r# �clearr )
r r rG �
duplicatesrD �_rF r9 Z deb_is_autor* r r r �get_deb2snap_dups� s2 � z InstallBackend.get_deb2snap_dupsc C s� i }| � � \}}|�� D ]v\}\}}i }tjdd|gdtjd��� }t�d|d tj�rjt �
d| � qq|| j j krd|d < ||d
< |||< q|�� D �]>\}} i }tjdd|gdtjd��� }t�d|d tj�r�t �
d| � d }
t�d |
|d tj��st �
d
| � q�d|d < tjdd|gdtjd��� } | d �
d�D ]�} | �
� }
t|
�dk�rb�qD|
d }|
d }|�|d ��rD|�
d�d }|dk�rD||k�rDt �
d||f � d|d < d|d
< �qАqD|||< q�|S )NrF �infoT)Zuniversal_newlines�stdoutz ^installed: r zSnap %s is installed�install�command�channelz#stable/ubuntu-[0-9][0-9].[0-9][0-9]z^tracking:.*%sz+Snap %s is not tracking the release channel�removeZ connections�
� r r �:�-z9Snap %s is being used by %s. Switching it to stable track�refreshZstable)rH rI �
subprocess�Popen�PIPEZ communicate�re�search� MULTILINErB rC r �duplicate_packages�split�len�
startswith)r � snap_listrD rE rF r9 r: � snap_objectZ snap_infor; Z
re_channelZconnsZconnZ conn_colsZplugZslotZ plug_snapr r r �get_snap_transitions� sp �
�
�� �
��
z#InstallBackend.get_snap_transitionsc
C sv |\}}}|sd S d}d} |� � D ] }
||
�� 7 }| |
�� 7 } q"| | } | | || } | |�� krrt�|j| � d S )Nr )Z get_tasksZget_progress_totalZget_progress_doneZ get_fractionr �idle_add� set_fraction)
r �clientZchangerL Z user_data�index�count� progress_barZ
task_totalZ task_doneZtaskZ
task_fractionZtotal_fractionr r r �update_snap_cb� s
z InstallBackend.update_snap_cbc
s� � fdd�}dd� }|t d�� d }d }� j�� }|r\t|d tj�r\|d }t�d||�}� �� }|r|t� |� |�
d� z�t � � }|�
� d}t|�} |�� D ]�\}
} | d } | d kr�|t d
|
�� |j|
| d � j|| |fd � nb| d
k�r |t d|
�� |j|
� j|| |fd � n.|t d|
�� |j|
| d � j|| |fd � |d7 }q�W nV tjk
�r� }
z4t�d|
� t�� jjdt d�t d�� W Y �d S d }
~
X Y nX t�� jj� d S )Nc s t �� jj| � d S r )r rf Z
label_detailsZ set_label)Zstatusr1 r r �
update_status s z2InstallBackend.update_snaps.<locals>.update_statusc S s | � � dS )NT)Zpulse)rk r r r �update_progress s z4InstallBackend.update_snaps.<locals>.update_progresszUpdating snapsr �d rQ rX zRefreshing %s snaprR )Zprogress_callback_datarS zRemoving %s snapzInstalling %s snapr zerror updating snaps (%s)Fz!Upgrade only partially completed.zMAn error occurred while updating snaps. Please check your network connection.)rL Zprogressbar_slotZ get_children�
isinstancer Z ProgressBarr � timeout_addre Z
source_removerg r ZClientZ connect_syncra rI Z refresh_syncrl Z remove_syncZ install_sync�ErrorrB rC rf r � start_error�start_available)r rm rn rk Zprogress_timerZ
progress_barsrc rh ri rj rF rd rQ rG r r1 r � update_snaps sj
�
��
�� �� �zInstallBackend.update_snapsFc C s� || j kr |r | j| _| �� n�|| jkr�|rVtj�d�rVt| d�rVt| j d��� q�|rf| j
� � q�|r|| j
� |||� q�t
�d| j
j� nZ|r�| j
� d||� nD|r�tj�d�r�t| d�r�| �� | j
_| j
� � n| }| j
� |� d S )Nz
/usr/bin/snapZpane_update_progress)�targetr T)r r! r r. r �path�exists�hasattrr ru r rt rs r rq �exitrM r_ )r r r2 r3 r4 r5 r6 Zis_cancelled_updater r r r7 @ s0
�
�� zInstallBackend._action_doneN)F)�__name__�
__module__� __qualname__Z
ACTION_UPDATEr r! r r. r' r r& rH rM re rl ru r7 r r r r r s 4&AA �r �/usr/sbin/aptd�%UPDATE_MANAGER_FORCE_BACKEND_SYNAPTICr )�InstallBackendAptdaemonzimporting aptdaemon�/usr/sbin/synaptic�&UPDATE_MANAGER_FORCE_BACKEND_APTDAEMON)�InstallBackendSynapticzimporting synapticc O s� t j�d�rBdt jkrBz t| |�W S tk
r@ t�d� Y nX t j�d�rzdt jkrzz t| |�W S tk
rx Y nX t d��dS )z,Select and return a package manager backend.r~ r zusing aptdaemon failedr� r� zENo working backend found, please try installing aptdaemon or synapticN)
r rw rx r r� � NameErrorrB � exceptionr� rA )�args�kwargsr r r � get_backendx s � � r� )$�__doc__Z
__future__r ZgiZrequire_versionZ
gi.repositoryr r r Zaptr r= rB r r\ rY r rL Z threadingr ZUpdateManager.Core.MyCacher
ZUpdateManager.Core.utilsr r ZUpdateManager.Dialogsr
r rw rx r r� � ImportErrorr� r� r� r r r r �<module> sB O��