File: //proc/self/root/lib/python3/dist-packages/uaclient/cli/__pycache__/__init__.cpython-38.pyc
U
��Jh % � @ s� d Z ddlZddlZddlmZmZmZmZmZm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd l m Z dd l m Z dd
l m!Z! ddl"m#Z# ddl$m%Z% ddl&m'Z'm(Z( ddl)m*Z* ddl+m,Z, ddl-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 e�5� Z6e�7e �8e9��Z:dZ;eeeeeeee e e!e#e%e*e,e.e0gZ<dd� Z=d&dd�dd �Z>dd�d d �Z?d d � Z@d!d"� ZAeAd'd#d$��ZBe9d%k�r�e�CeB� � dS )(z2Client to manage Ubuntu Pro services on a machine.� N)
�apt�defaults� event_logger�
exceptions�http�lock�log�messages�util�version)� api_command)�attach_command)�auto_attach_command)�collect_logs_command)�config_command)� cve_command)� cves_command)�detach_command)�disable_command)�enable_command)� fix_command)� help_command)� HelpCategory�ProArgumentParser)�refresh_command)�security_status_command)�status_command)�system_command)�UAConfig)� get_user_or_root_log_file_pathZproc C s� t tdtjjtdd�d�} | �tjdtjjtd�� | j ddtj
d � | �tjdtj
� | j d
d t � � tj
jtd�d � | �tjd
tj
jtd�� | jtjd
dd�}d|_tD ]}|�|� q�| S )NFz <command>)�name�command)�progZ
use_main_helpZepilogz
-h, --help)r z--debug�
store_true)�action�help� --versionr )r$ r r% r! )�title�dest�metavarT)r �NAMEr ZCLI_HELP_EPILOG�formatZadd_help_entryr �FLAGSZCLI_HELP_FLAG_DESC� add_argumentZCLI_ROOT_DEBUGr Z get_versionZCLI_ROOT_VERSIONZadd_subparsersZCLI_AVAILABLE_COMMANDSZrequired�COMMANDS�register)�parserZ
subparsersr! � r1 �7/usr/lib/python3/dist-packages/uaclient/cli/__init__.py�
get_parser? sP � � � � � �� r3 )�returnc C sX | r| j dks t| dd�dkr"d S t�� }|rTt�d|� tjtj j
|d�t j d� d S )NZapir+ � ��json�yamlzNew version available: %s)r �� file_type)
r! �getattrr Zcheck_for_new_version�LOG�warning�event�infor Z WARN_NEW_VERSION_AVAILABLE_CLIr+ �sys�stderr)�cmd_argsZ new_versionr1 r1 r2 �_warn_about_new_versionk s ��� ��rC c C sV | j dkrRtj�� sRt| d�r,| jdkr,dS t�d� tj t
j j| j d�tj d� dS )z4Warn users that the user readable output may change.)�statuszsecurity-statusr+ r6 Nz.Not in a tty and human-readable command called)r! r9 )
r! r@ �stdout�isatty�hasattrr+ r< r= r> r? r Z WARNING_HUMAN_READABLE_OUTPUTrA �rB r1 r1 r2 � _warn_about_output_redirection� s ��
��rI c C sT | j dkrPt�| j � t| d�rP| jdkr8t�tjj� | jdkrPt�tjj � dS )z3Set the right event mode based on the args provided)Zattach�detach�enable�disablerD r+ r7 r8 N)
r! r> Z set_commandrG r+ �set_event_moder ZEventLoggerModeZJSONZYAMLrH r1 r1 r2 rM � s
rM c s � fdd�}|S )Nc
s� z � | |�W S t k
rN t�d� ttjtjd� t� � t�
d� Y �n� t j k
�r4 } z�dt
|�kr�tj}t�d�r�tj}|j|jd�}tj|j|jd� tj|jtjd� nJtjd |j|d
� tjj|j|jd �}tj|j|jd� tj|jtjd� t� � t�� t � t�
d� W 5 d }~X Y �n� t j k
�r� } zntj}t�d��r`tj}|j|jd�}tj|j|jd� tj|jtjd� t� � t�� t � t�
d� W 5 d }~X Y �n t j k
�rR } znt�|j� tj|j|j |j d � tjd
�|j�tjd� t!|t j"��s(t� � t�� t � t�
|j#� W 5 d }~X Y n� t$k
�r� } zlt�d� t� � tjtj%jt
|�t&� d�jtjd� tjt'|dt
|��dd� t�� t � t�
d� W 5 d }~X Y nX d S )N�KeyboardInterrupt)�file� ZCERTIFICATE_VERIFY_FAILEDzca-certificates)�url)� error_msg�
error_code)Zinfo_msgr: zFailed to access URL: %s)�exc_info)rQ � cause_error)rR rS �additional_infoz{}z&Unhandled exception, please file a bug)rR Zlog_path�msg� exception)rR Z
error_type)(rN r< �error�printr ZCLI_INTERRUPT_RECEIVEDr@ rA r Zclear_lock_file_if_present�exitr ZConnectivityError�strZ&SSL_VERIFICATION_ERROR_CA_CERTIFICATESr Z is_installedZ%SSL_VERIFICATION_ERROR_OPENSSL_CONFIGr+ rQ r> rW r r? rX ZE_CONNECTIVITY_ERRORrU Zprocess_eventsrC ZPycurlCACertificatesErrorZUbuntuProErrorZmsg_coderV �
isinstanceZ
LockHeldErrorZ exit_code� ExceptionZUNEXPECTED_ERRORr r; )�args�kwargs�excZtmplrW �e��funcr1 r2 �wrapper� s�
�� �
�� �z#main_error_handler.<locals>.wrapperr1 )rd re r1 rc r2 �main_error_handler� s Srf c
C sp t �tjd tjd � t� }t �|j|j� | s8tj} t � }| dd � }|s`|�
� t� d� |d dkrtd|d< d|kr�|� d�}|d |� }||d d � }n|}g }|j
|d�}|jr�t�tj�}|�tj� t�d ��|� t|� |j} |j}
tj| |
d
� t�d | � |� � d d
� t t � � � � �D �} | �rLt�d| � t!|� |j"|||d�} t#|� | S )N� log_level�log_filerP r r r& z--)r_ Z ubuntupro)�
http_proxy� https_proxyzExecuted with sys.argv: %rc S s g | ]\}}d � ||��qS )z{}={})r+ )�.0�k�vr1 r1 r2 �
<listcomp># s �zmain.<locals>.<listcomp>z'Executed with environment variables: %r)�cfg�
extra_args)$r Zsetup_cli_loggingr ZCONFIG_DEFAULTSr rg rh r@ �argvr3 Z
print_helpr[ �index�
parse_args�debug�loggingZ
StreamHandlerrA ZsetLevel�DEBUG� getLoggerZ
addHandlerrM ri rj r Zconfigure_web_proxyr< Zwarn_about_invalid_keys�sortedr
Zget_pro_environment�itemsrI r$ rC )
Zsys_argvro r0 Z
cli_argumentsZdouble_dash_indexZ pro_cli_argsrp r_ Zconsole_handlerri rj Zpro_environmentZ return_valuer1 r1 r2 �main� sR �
�rz �__main__)N)N)D�__doc__ru r@ Zuaclientr r r r r r r r r
r Zuaclient.cli.apir Zuaclient.cli.attachr
Zuaclient.cli.auto_attachr Zuaclient.cli.collect_logsr Zuaclient.cli.configr Zuaclient.cli.cver Zuaclient.cli.cvesr Zuaclient.cli.detachr Zuaclient.cli.disabler Zuaclient.cli.enabler Zuaclient.cli.fixr Zuaclient.cli.helpr Zuaclient.cli.parserr r Zuaclient.cli.refreshr Z uaclient.cli.security_statusr Zuaclient.cli.statusr Zuaclient.cli.systemr Zuaclient.configr Z uaclient.logr Zget_event_loggerr> rw Z replace_top_level_logger_name�__name__r< r* r. r3 rC rI rM rf rz r[ r1 r1 r1 r2 �<module> sf 0 �, W >