OwlCyberSecurity - MANAGER
Edit File: snap.cpython-38.pyc
U ��g � @ s� d dl Z d dlZd dlmZmZmZ d dlmZmZm Z m Z m Z m Z m Z mZ dZdddgZdZd Zd Zd Ze �� Ze �e�e��Zed d efdefdefdefdefg�Z e d�dd�Z e d�dd�Z d1ee ee eee dd�dd�Z!d2eeee dd�dd �Z"eee d �d d �Z#ee d�d d!�Z$d"d#� Z%ej&d$�d%d&�Z'd3eee e d(�d)d*�Z(ed+�d,d-�Z)ee d.�d/d0�Z*dS )4� N)�List� NamedTuple�Optional)�api�apt� event_logger� exceptions�http�messages�system�utilz /usr/bin/snapg �?g �?g @z proxy.httpz proxy.httpsz/run/snapd.socketz /v2/snaps/{}� SnapPackage�name�version�revision�channel� publisher)�returnc C s dt �� kS )z(Returns whether or not snap is installed�snapd)r Z get_installed_packages_names� r r �//usr/lib/python3/dist-packages/uaclient/snap.py�is_snapd_installed' s r c C s t dd� t� D ��S )z3Returns whether or not snapd is installed as a snapc s s | ]}|j d kV qdS )r N�r )�.0�snapr r r � <genexpr>. s z/is_snapd_installed_as_a_snap.<locals>.<genexpr>)�any�get_installed_snapsr r r r � is_snapd_installed_as_a_snap, s r )� http_proxy� https_proxy� retry_sleepsr c C st t � st�d� dS | s |r0t�tjjdd�� | rPtj dddd�| �g|d� |rptj dddd �|�g|d� dS ) a Configure snap to use http and https proxies. :param http_proxy: http proxy to be used by snap. If None, it will not be configured :param https_proxy: https proxy to be used by snap. If None, it will not be configured :param retry_sleeps: Optional list of sleep lengths to apply between retries. Specifying a list of [0.5, 1] tells subp to retry twice on failure; sleeping half a second before the first retry and 1 second before the second retry. z6Skipping configure snap proxy. snapd is not installed.Nr )Zservice�setr z proxy.http={}�r! zproxy.https={}) r �LOG�debug�event�infor ZSETTING_SERVICE_PROXY�formatr �subp)r r r! r r r �configure_snap_proxy1 s ��r* )� protocol_typer! r c C s4 t � st�d� dS tjdddd�| �g|d� dS )a� Unset snap configuration settings for http and https proxies. :param protocol_type: String either http or https :param retry_sleeps: Optional list of sleep lengths to apply between retries. Specifying a list of [0.5, 1] tells subp to retry twice on failure; sleeping half a second before the first retry and 1 second before the second retry. z8Skipping unconfigure snap proxy. snapd is not installed.Nr Zunsetr zproxy.{}r# )r r$ r% r r) r( )r+ r! r r r �unconfigure_snap_proxyV s �r, )�keyr c C s> z t �ddd| g�\}}|�� W S tjk r8 Y dS X dS )z� Gets the config value from snap. :param protocol: can be any valid snap config option :return: the value of the snap config option, or None if not set r �getr N)r r) �stripr �ProcessExecutionError)r- �out�_r r r �get_config_option_valuek s r3 c C sZ t �ddddddg�\} }| �� }|dd � }g }|D ] }|�� d }|�t|�� q6|S )Nr �listz--color�neverz --unicode� r )r r) � splitlines�split�append� get_snap_info)r1 r2 ZappsZsnaps�liner r r r r x s � r c C s� t �tjjdd�� zt�t�� � W n4 tj k rZ } zt � dt | �� W 5 d } ~ X Y nX z t jddddgtjd� W n tjk r� t�� �Y nX d S ) Nzstandard Ubuntur z<Trying to install snapd. Ignoring apt-get update failure: %szapt-get�installz --assume-yesr r# )r& r' r ZAPT_UPDATING_LISTr( r Zupdate_sources_listZget_system_sources_filer ZUbuntuProErrorr$ r% �strr r) Z APT_RETRIESr0 ZCannotInstallSnapdError)�er r r � install_snapd� s � � r? )�progressc C st zt jtdddgdd� W nT tjk rn } z4t�dt|��� �r\t � d� | � dt j � n� W 5 d }~X Y nX d S ) N�waitr z seed.loadedT��capturezunknown command .*waitz9Detected version of snapd that does not have wait commandr' )r r) �SNAP_CMDr r0 �re�searchr= �lowerr$ Zwarning�emitr Z SNAPD_DOES_NOT_HAVE_WAIT_CMD)r@ r> r r r �run_snapd_wait_cmd� s �rI F)r r �classic_confinement_supportc C s@ t d| g}|r|dg7 }|r,|d�|�g7 }tj|dtd� d S )Nr< z --classicz --channel={}T)rC r! )rD r( r r) �SNAP_INSTALL_RETRIES)r r rJ �cmdr r r � install_snap� s �rM �r c C s t jtd| gdd� d S )NZrefreshTrB )r r) rD rN r r r � refresh_snap� s rO )r r c C s� t �| �}zlt�td|�}|jdkrt|jdkrT|j�di ��d�dkrTtj | d��n |j�di ��d�}tj |d ��W n t k r� t� � �Y nX |j�di �}t |�d d �|�d d �|�d d �|�dd �|�di ��dd �d�S )NZGET�� i� �resultZkindzsnap-not-foundrN �message)�errorr � r r r r Zusername)r r r r r )�SNAPD_SNAPS_APIr( r Zunix_socket_request�SNAPD_SOCKET_PATH�codeZ json_dictr. r ZSnapNotInstalledErrorZUnexpectedSnapdAPIError�ConnectionRefusedErrorZSnapdAPIConnectionRefusedr )r ZurlZrespZ error_msgZ snap_infor r r r: � s. ��� �r: )NNN)N)NF)+ZloggingrE �typingr r r Zuaclientr r r r r r r r rD rK ZHTTP_PROXY_OPTIONZHTTPS_PROXY_OPTIONrV rU Zget_event_loggerr& Z getLoggerZ replace_top_level_logger_name�__name__r$ r= r �boolr r �floatr* r, r3 r r? ZProgressWrapperrI rM rO r: r r r r �<module> sd ( �� � � & � � ��