OwlCyberSecurity - MANAGER
Edit File: __init__.cpython-38.pyc
U ��g % � @ 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 >