CasperSecurity
o
��Jh�* � @ s� d dl mZmZmZmZmZmZmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZm Z d d l m Z m Z e� � Z!d d
� Z"dd� Z#e
j$dd� �Z%e
j$dd� �Z&edej'ej'e#eedej(dd�gd�gd�Z)edej*ej*e%eedej+j,d �-ej.�d �d �gd�gd�Z/ed ej0ej0e&eedej1j,d �-ej.�d �dd �gd�gd�Z2ed!ej3ej4e"ej5e)e/e2gd"�Z6d#S )$� )�apt_news�config� entitlements� event_logger�
exceptions�http�messages)� _is_attached)�
AptProxyScope)�cli_util)� ProArgument�ProArgumentGroup�
ProCommand)� HelpCategory)�ApplicationStatus)� state_files)�LXDGuestAttachEnum)�configure_livepatch_proxy�unconfigure_livepatch_proxy)�configure_snap_proxy�unconfigure_snap_proxyc K s ddl m} |� �d� dS )Nr ��
get_parserr )� uaclient.clir �print_help_for_command)�args�cfg�kwargsr � r �5/usr/lib/python3/dist-packages/uaclient/cli/config.py�
action_config s r c K s� | j r)| j tjvrtjd�| j �d�tj�d��tdj| j t|| j d�d�� dS t t
dd � tjD ��d
�}d | d }tjD ]}t|j|t||d�d�� q@|j sV|j rc|j
s\|jrettj� dS dS dS )
z�Perform the 'config show' action optionally limit output to a single key
:return: 0 on success
:raise UbuntuProError: on invalid keys
z'{}'�, ��arg�choicesz
{key} {value}N��key�valuer c S s g | ]}t |��qS r )�len)�.0�xr r r �
<listcomp>6 s z&action_config_show.<locals>.<listcomp>� z{key: <z } {value})r& r �UA_CONFIGURABLE_KEYSr �InvalidArgChoice�format�join�print�getattr�str�max�global_apt_http_proxy�global_apt_https_proxy�ua_apt_http_proxy�ua_apt_https_proxyr � CLI_CONFIG_GLOBAL_XOR_UA_PROXY)r r r � col_width�row_tmplr& r r r �action_config_show# s.
���
���r< c
K sv ddl m} |� }z
| j�d�\}}W n ty' |�d� tjd| jd��w |tj vr=|�d� tj
dd� tj �d ��|� � sL|�d� tj
|d
��tt||d ��tkri|�� d vrctj
d
dd ��|�� dk}|dv r�|�d�d }|dkr|tj}ntj}t�|||� ||i}td#i |�� tj�|�} | �� \}
} |
tj kr�t d#i |�� �n||j v r�|�d�d }|dkr�tj }ntj }t�|||� t|j!p�|j"�} | r�t#t$j%j&ddd�� t'�(|t)j*||� d |_!d |_"n�||j+|j, v �rQd|v r�dnd}|dk�rtj }ntj }||j+v �r$t#t$j-j&d�&|�d�&|�d�� d| }t�|||� t|j.�p2|j/�} | �rAt#t$j%j&ddd�� t'�(|t)j0||� d |_.d |_/nb|d v �r�zt1|�}|dk �rgtd �&|���W nJ t�y� |�d� t#d � tj2||d ��w |d k�r�|�r�t3�4|� n$t5j6�7� n |d!k�r�t8|�j9�s�t�:� �t5j;�<t5j=t>�?|�� �d"�� t@|||� d S )$zMPerform the 'config set' action.
@return: 0 on success, 1 otherwise
r r �=z
config setz
<key>=<value>)�expected�actual�<key>r! r" )r# N)�true�falsez<value>z true, falserA ��
http_proxy� https_proxy�_r � zpro scoped aptz
global apt)�
current_proxy�previous_proxy�https� apt_{}_proxy�global_apt_{}_proxy��old�new�global_)�update_messaging_timer�metering_timerzInvalid interval for {}� r% r �lxd_guest_attach�� guest_attachr )Ar r �key_value_pair�split�
ValueErrorr r �GenericInvalidFormatr r- r. r0 �strip�EmptyConfigValue�typer2 �bool�lowerr � PROXY_VALIDATION_SNAP_HTTP_URL� PROXY_VALIDATION_SNAP_HTTPS_URL�validate_proxyr r � livepatch�LivepatchEntitlement�application_statusr �ENABLEDr �ua_scoped_proxy_options� PROXY_VALIDATION_APT_HTTP_URL� PROXY_VALIDATION_APT_HTTPS_URLr5 r6 r1 r �WARNING_APT_PROXY_OVERWRITEr/ r �configure_apt_proxyr
�UACLIENT�&deprecated_global_scoped_proxy_options�global_scoped_proxy_options�WARNING_CONFIG_FIELD_RENAMEr7 r8 �GLOBAL�int�InvalidPosIntConfigValuer �update_apt_newsr �apt_news_contents_file�deleter � is_attached�UnattachedError�lxd_pro_config_file�write� LXDProConfigr �
from_value�setattr)
r r r r �parser�set_key� set_value�
protocol_type� validate_url� entitlement�livepatch_statusrF �
unset_currentr r r �action_config_setB s�
��
�
�
�
���
���
����
�
�
��
��r� c K sX ddl m} | jtjvr |� }|�d� tjdd�tj�d��| jdv rH| j� d�d }t
|d � t j �
|�}|�� \}}|tjkrGt|d � n[| j|jv rYt�|tj| jd
� nJ| j|j|j v r�| j|jv r�d | jv rod nd }t�tj j d
� |�d� |�d�� d| j | _t�|tj | jd
� n| jdkr�t j �!t j"t#j$d�� t%|| jd
� dS )zOPerform the 'config unset' action.
@return: 0 on success, 1 otherwise
r r z config unsetr@ r! r" rC rF )r� NrJ r rK rL rM rP rT rU )&r r r& r r- r r r. r0 rX r r rc rd re r rf r rg r rk r
rl rm rn �event�infor ro r/ rp r rx ry rz r �OFFr| ) r r r r r} r� r� r� rF r r r �action_config_unset� sP
�
� ��� ��
�r� �showr& �?)�help�nargs)� arguments)r� � description�action�argument_groups�setrW r! )�options)r� �unset)r� �metavarr )r� r� r� �
help_category� subcommandsN)7�uaclientr r r r r r r �(uaclient.api.u.pro.status.is_attached.v1r � uaclient.aptr
r r �uaclient.cli.commandsr r
r �uaclient.cli.parserr �(uaclient.entitlements.entitlement_statusr �uaclient.filesr � uaclient.files.user_config_filer �uaclient.livepatchr r �
uaclient.snapr r �get_event_loggerr� r r< � assert_rootr� r� �CLI_CONFIG_SHOW_DESC�CLI_CONFIG_SHOW_KEY�show_subcommand�CLI_CONFIG_SET_DESC�CLI_CONFIG_SET_KEY_VALUEr/ r0 r- �set_subcommand�CLI_CONFIG_UNSET_DESC�CLI_CONFIG_UNSET_KEY�unset_subcommand�CLI_ROOT_CONFIG�CLI_CONFIG_DESC�OTHER�config_commandr r r r �<module> s� $
0�����
������
������
�