File: //lib/python3/dist-packages/uaclient/__pycache__/snap.cpython-310.pyc
o
nHJe_ � @ s� d dl Zd dlZd dlZd dlZd dlZd dlmZmZm Z d dl
m Z m Z m
Z
mZmZmZ dZg d�ZdZdZdZd Ze �� Ze�e�e��Z ed
d e fd e fd
e fde fde fg�Z de fdd�Z de fdd�Z! d-de e de e de ee" ddfdd�Z# d.de de ee" ddfdd �Z$d e de e fd d �Z%dee fd d!�Z&d"d#� Z'd$d%� Z( &d/d'e de e d(e fd)d*�Z)d'e de fd+d,�Z*dS )0� N)�List�
NamedTuple�Optional)�apt� event_logger�
exceptions�messages�system�utilz
/usr/bin/snap)g �?g �?g @z
proxy.httpz proxy.httpsz/run/snapd.socketz /v2/snaps/{}� SnapPackage�name�version�revision�channel� publisher�returnc C s dt �� v S )z(Returns whether or not snap is installed�snapd)r � 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_sleepsc C sx t � s
t�d� dS | s|rt�tjjdd�� | r(tj dddd�| �g|d� |r:tj dddd �|�g|d� dS 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 �SETTING_SERVICE_PROXY�formatr �subp)r r r r r r �configure_snap_proxy+ s
�
��r+ �
protocol_typec C s4 t � s
t�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 �unsetr zproxy.{}r# )r r$ r% r r* r) )r, r r r r �unconfigure_snap_proxyP s
�r. �keyc C s8 zt �ddd| g�\}}|�� W S tjy Y dS w )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_valuee s
�r5 c C sR t �g d��\} }| �� }|dd � }g }|D ]}|�� d }|�t|�� q|S )N)r �listz--color�neverz --unicoder7 � r )r r* �
splitlines�split�append�
get_snap_info)r3 r4 �apps�snaps�liner r r r r r s � r c
C s� t �tjjdd�� zt�d� W n tjy, } zt �
dt | �� W Y d } ~ nd } ~ ww z
t j
g d�tjd� W d S tjyF t�� �w )Nzstandard Ubuntur z/etc/apt/sources.listz<Trying to install snapd. Ignoring apt-get update failure: %s)zapt-get�installz --assume-yesr r# )r& r' r �APT_UPDATING_LISTr) r �update_sources_listr �UbuntuProErrorr$ r% �strr r* � APT_RETRIESr2 �CannotInstallSnapdError��er r r �
install_snapd� s$ ��� ��rI c
C st zt jtdddgdd� W d S tjy9 } z t�dt| ��� �r-t �
d� t � t
j� n� W Y d } ~ d S d } ~ ww )N�waitr z seed.loadedT)�capturezunknown command .*waitz9Detected version of snapd that does not have wait command)r r* �SNAP_CMDr r2 �re�searchrD �lowerr$ �warningr& r' r � SNAPD_DOES_NOT_HAVE_WAIT_CMDrG r r r �run_snapd_wait_cmd� s ����rR Fr �classic_confinement_supportc C s@ t d| g}|r |dg7 }|r|d�|�g7 }tj|dtd� d S )Nr@ z --classicz --channel={}T)rK r )rL r) r r* �SNAP_INSTALL_RETRIES)r r rS �cmdr r r � install_snap� s
�rV c
C s� t � t jt j�}|�t� tj�d�}||_t �
| �}z~zh|� dt �
| �� |� � }|�
� �d�}zt�|�}W n tjyX } ztjd|||d� tjd�
|�|d��d }~ww |jdkr�|jd krt|�d
i ��d �d krttj| d
��|�d
i ��d�}tj|d��W n ty� t�� �w W |� � |� � n |� � |� � w |�d
i �} t | �dd�| �dd�| �dd�| �dd�| �di ��dd�d�S )N� localhost�GETzutf-8z]JSONDecodeError while parsing result of snap api call to %s, returning None. output was: "%s")�exc_infoz SNAPD API {})�sourcer3 �� i� �result�kindzsnap-not-found)r �message)�errorr � r
r r r �username)r r
r r r ) �socket�AF_UNIX� SOCK_STREAM�connect�SNAPD_SOCKET_PATH�http�client�HTTPConnection�sock�SNAPD_SNAPS_APIr) �request� getresponse�read�decode�json�loads�JSONDecodeErrorr$ rP r � InvalidJson�statusr0 �SnapNotInstalledError�UnexpectedSnapdAPIError�ConnectionRefusedError�SnapdAPIConnectionRefused�closer )
r � snap_sock�conn�url�responser3 �datarH � error_msg� snap_infor r r r< � s\
�
���
�
��
�
�r< )NNN)N)NF)+� http.clientrg rp �loggingrM rb �typingr r r �uaclientr r r r r r
rL rT �HTTP_PROXY_OPTION�HTTPS_PROXY_OPTIONrf rk �get_event_loggerr& � getLogger� replace_top_level_logger_name�__name__r$ rD r �boolr r �floatr+ r. r5 r rI rR rV r< r r r r �<module> sx �� ���
�
�&��
�
�
���
�