File: //lib/python3/dist-packages/uaclient/__pycache__/snap.cpython-38.pyc
U
��Jh � @ 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 )�service�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
r4 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)r2 r3 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_RETRIESr1 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 r1 �re�searchr> �lowerr% Zwarning�emitr
Z SNAPD_DOES_NOT_HAVE_WAIT_CMD)rA r? r r r �run_snapd_wait_cmd� s �rJ 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)rD r! )rE r) r r* �SNAP_INSTALL_RETRIES)r r rK �cmdr r r � install_snap� s
�rN �r c C s t jtd| gdd� d S )NZrefreshTrC )r r* rE rO r r r � refresh_snap� s rP )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-foundrO �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)+ZloggingrF �typingr r r Zuaclientr r r r r r
r r rE rL ZHTTP_PROXY_OPTIONZHTTPS_PROXY_OPTIONrW rV Zget_event_loggerr' Z getLoggerZ replace_top_level_logger_name�__name__r% r> r
�boolr r �floatr+ r- r4 r r@ ZProgressWrapperrJ rN rP r; r r r r �<module> sd (
�� �
� & �
�
��