OwlCyberSecurity - MANAGER
Edit File: InstallBackendAptdaemon.cpython-38.pyc
U ρ�f�L � @ sJ d dl mZ d dlmZ d dlmZmZ d dlmZ d dl m Z m Z m Z m Z mZ d dlmZmZmZmZmZmZmZmZmZ d dlmZ d dlm 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 dd� dej$�Z%G dd� dee �Z&e'dk�rFd d l(Z(e(�)� Z*dZ+d dl,m-Z- e-e+e*�Z.e&e.d �Z/e/�0dgg 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 Zd ZdZd� fdd� Zdd� Z� ZS ) �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 tfkr2| �d� | j�� | jr�| j�� n�|tkr|| j�� | jr`| j�� | �d� q�| jr�| � d� | �d� n<| jr�|t kr�| j�� | jr�| j�� | �d� | � d� d S �NTF) r r Z set_sensitiveZdownload_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 Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS ) �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| krdksn t �t|�}| j�|� | j|_| j|7 _| jdkrTd| _| j|_|�d| j� |�d| j � d S )Nr � znotify::fractionz notify::text) �AssertionErrorr r2 �appendr1 �min�max�connect�_update_progress� _update_text)r r( Z max_progressZprogressr# 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�dataZdeltaZposition_in_deltaZ new_progressr# r# r$ r: f s z&AptStackedProgressBar._update_progressc C s | � |�� � d S r )Zset_textZget_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 Zd ZdZdd� Zdd� Zedd� �Zdd � Zd d � Z ed d � �Z dd� Z edd� �Z edd� �Z dd� Zdd� Zed&dd��Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� ZdS )'�InstallBackendAptdaemonzCMakes use of aptdaemon to refresh the cache and to install updates.c C sh t �| ||� tj�|jd�}t�| ||d� t�� | _t � | _ d | _ d | _ d | _ d | _d | _d | _d S )Nz gtkbuilder/UpdateProgress.uiZpane_update_progress)r r �os�path�joinZdatadirr r Z AptClientr r3 �_expanded_size� button_cancel�trans_failed_msg� progressbarZ_active_transaction� _expander)r � window_main�actionZui_pathr# r# r$ r t s � � z InstallBackendAptdaemon.__init__c C s@ | j r8| j �� r8z| j �� W n tk r2 Y nX dS dS d S r% )rF Z get_sensitiveZclicked� Exception)r r# r# r$ �close� s z InstallBackendAptdaemon.closec c s� z,| j jdd�V }| �|| jtd�d�V W nV tjk rX | j| jddddd� Y n, tk r� | j| jddddd� � Y nX dS )zRefresh 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 |jj�}t|jj�}| jr2d}|d | j }nd }| j|dd|||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 kr| �||� 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�}zXtj�|�rv| jj|d�V }| j rL| j}n| j }| j || j t d�d|d| j d�V W n� tjk r� | j| j ddd d d � Y n� tjk r� } zt|�| _W 5 d }~X Y nv tjk �r } z(|�� d kr� | j| j ddd d d � W 5 d }~X Y n. tk �rD | j| j ddd d d � � Y nX 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 �elemZsources_list_filer( Zfinished_handler�er# r# r$ rb � sZ � � � � �z'InstallBackendAptdaemon._apt_update_oemc C sJ |t kr| �||� d S |j\}}}}}}t|�| _t|�| _| �� d S r )r ra Zpackages�setrj �lenro rb )r r( r) rK ZinstallrX r# r# r$ � _update_oem� s z#InstallBackendAptdaemon._update_oemc c sD t |�t |�B | _z`|s:|s:| j| jddd d d d� W d S |rp| jj|dd�V }| j|| jtd�d| jdd�V W n� t j k r� | j| jddd d d� Y n� t j k r� } zt |�| _ W 5 d }~X Y nv tjk �r } z(|�� d kr� | j| jddd d d� W 5 d }~X Y n. tk �r> | j| jddd d d� � Y nX d S ) NTr] rN rd re rf FrP ri )rw Zall_oem_packagesrZ rn r Zinstall_packagesrV rX ry r rY rp rq rG rr rs rt rL )r Zpkgs_install_oemZpkgs_upgrade_oemr( rv r# r# r$ � commit_oem� sb �� � � � �z"InstallBackendAptdaemon.commit_oemc c s zDg } }}| j j||||||dd�V }| �|| jtd�d�V W n� tjk rp | j| jddddd� Y n� tjk r� } zt |�| _ W 5 d}~X Y nt t j k r� } z(|� � dkr�� | j| jddddd� W 5 d}~X Y n. tk �r | j| jddddd� � Y nX dS )z)Commit a list of package adds and removesTrN rd FNrP ri )r Zcommit_packagesrV ZACTION_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( Zdetails� label_detailsr# r# r$ �_on_details_changed3 s z+InstallBackendAptdaemon._on_details_changedc C s8 |� t|�� |tkr4| jr4| jj�� 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 kr| j }| j�|� | jsDt| j�| _| j�� | j�| j� | j� ||� | j rf| j � |� n"t |�| _ | j �� | j �| j � || jkr�| j �tj� |�r | j�st|�| _| j�d� | j�d� | j�� | j�d| j� | j�| j� | j�� n | j� |� || jk| j_n| j�r2| j� � |�d| j | j � |�d| j | j � |�d||� |�d| j � |�d| j!� |�"d�V |�#� V d S ) NTznotify::expandedzstatus-details-changedzstatus-changedZfinishedzmedium-requiredzconfig-file-conflictZgnome)$rm Z label_headerr} rH r0 r3 r& Zprogressbar_slot�addr< rF Zset_transactionr Zbutton_cancel_slotrW r Z STOCK_STOPrI r Z set_vexpandZ set_hexpandZshow_allr9 � _on_expandedZ expander_slotrn r Z_expander_slotr' r r~ r* �_on_medium_required�_on_config_file_conflictZset_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| jrh| j\}\}}||j�� krX| �|� qr| j�||� n | �|� d S r ) � get_expandedr r� rJ �get_sizerE Zend_user_resizabler� �begin_user_resizable)r �expanderZparamZ 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� || � dS )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� Zget_allocation�widthZheightr r� Zget_char_widthZget_char_heightr� ) r r� Z win_widthZ win_heightZ exp_widthZ exp_heightZterminal_widthZterminal_heightZ new_widthZ new_heightr# r# r$ r� � s z/InstallBackendAptdaemon._resize_to_show_detailsc C sB t ||| j�}|�� }|�� |tjjkr6|�|� n|�� d S r ) r rJ r� r' r � ResponseTypeZOKZprovide_mediumZcancel)r r ZmediumZdrive�dialog�resr# r# r$ r� � s z+InstallBackendAptdaemon._on_medium_requiredc C sH t ||| j�}|�� }|�� |tjjkr8|�|d� n |�|d� d S )N�replaceZkeep)r rJ r� r' r r� ZYESZ 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 krDt|jj�}t|jj�}| jrDd}|d | j }| j�d� |tk}z| j |d||||d� W n( t k r� | j |d|||d� Y nX 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 ry rz r| r r* rV r� r� r� r� rm r# r# r# r$ rA q s2 ( ' � 5 rA �__main__z/usr/share/update-manager)� UpdateManagerZ2vcard)2Z __future__r Z gi.repositoryr Z aptdaemonr r rO r Zaptdaemon.gtk3widgetsr r r r r Zaptdaemon.enumsr r r r r r r r r ZUpdateManager.backendr ZUpdateManager.UnitySupportr ZUpdateManager.Dialogsr r rX rr rB r Z ProgressBarr0 rA r+ ZmockZMockZoptionsZdata_dirZUpdateManager.UpdateManagerr� Zapp�br| �mainr# r# r# r$ �<module> s2 , "* ^