File: //lib/python3/dist-packages/uaclient/cli/__pycache__/cli_util.cpython-38.pyc
U
��JhQ � @ sr d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
m Z m Z m
Z
mZmZmZmZmZmZ d dlmZ d dlmZ d dlmZmZ d dlm Z d d l m Z e� � Z d
Z!G d d � d e j"�Z#d
d� Z$e%d�dd�Z&d.dd�Z'dd� Z(dd� Z)d/dd�Z*dd� Z+d d � Z,d d � Z-e dd �d!d"�Z.e ee/ee/ dd#�d$d%�Z0d&d'� Z1d0ee/ d)�d*d+�Z2d,d-� Z3dS )1� N��wraps)�Optional)
�actions�api�daemon� entitlements� event_logger�
exceptions�lock�messages�status�util)�CVE_OR_USN_REGEX)� _is_attached)�
AptProxyScope�setup_apt_proxy)�UAConfig)�
machine_tokenz[Kc @ sD e Zd Zed�dd�Zeeee ee d�dd�Zed�dd �Z d
S ) �CLIEnableDisableProgress)�
assume_yesc C s | | _ || _d S �N)Zis_interactiver )�selfr � r �7/usr/lib/python3/dist-packages/uaclient/cli/cli_util.py�__init__" s z!CLIEnableDisableProgress.__init__)� total_steps�
done_steps�previous_step_message�current_step_messagec C s |d k rt |� d S r )�print)r r r r r r r r �progress& s z!CLIEnableDisableProgress.progress)�eventc C s4 |dkrt |� n |dkr0t�|| j�s0t�� �d S )N�infoZmessage_operation)r r Zhandle_message_operationsr r
ZPromptDeniedError)r r" Zpayloadr r r � _on_event1 s
z"CLIEnableDisableProgress._on_eventN)
�__name__�
__module__� __qualname__�boolr �intr �strr! r$ r r r r r ! s � r c O s d S r r ��args�kwargsr r r � _null_print9 s r. �Z json_outputc C s | rt S tS d S r )r. r r/ r r r �&create_interactive_only_print_function= s r0 c s � fdd�}|S )z1Decorator asserting exclusive access to lock filec s t � �� �fdd��}|S )Nc s0 t j�dd�� � |d| i|��}W 5 Q R X |S )N� )� lock_holderZ
sleep_time�cfg)r Z RetryLock)r3 r, r- �retval)�fr2 r r �new_fH s z0assert_lock_file.<locals>.wrapper.<locals>.new_fr �r5 r6 �r2 �r5 r �wrapperG s z!assert_lock_file.<locals>.wrapperr )r2 r: r r8 r �assert_lock_fileD s r; c s t � �� fdd��}|S )z Decorator asserting root userc s t �� st�� �n
� | |�S d S r )r Zwe_are_currently_rootr
ZNonRootUserErrorr+ r9 r r r6 V s
zassert_root.<locals>.new_fr r7 r r9 r � assert_rootS s r< c s t � �� fdd��}|S )z>Decorator to verify if correct params are used for json formatc sB | s� | f|�|�S | j dkr.| js.t�� �n� | f|�|�S d S )NZjson)�formatr r
Z CLIJSONFormatRequireAssumeYes)Zcmd_argsr, r- r9 r r r6 c s
z&verify_json_format_args.<locals>.new_fr r7 r r9 r �verify_json_format_args` s r> c s � fdd�}|S )z�Decorator asserting attached config.
:param msg_function: Optional function to generate a custom message
if raising an UnattachedError
c s t � �� �fdd��}|S )Nc sR t |�js>�r6t| dd�}t| dd�}�|||d� nt�� �� | fd|i|��S )N�command� �service)r? �
service_namesr3 r3 )r � is_attached�getattrr
ZUnattachedError)r, r3 r- r? rB )r5 �raise_custom_error_functionr r r6 w s
�z/assert_attached.<locals>.wrapper.<locals>.new_fr r7 �rE r9 r r: v s z assert_attached.<locals>.wrapperr )rE r: r rF r �assert_attachedp s rG c s t � �� fdd��}|S )z&Decorator asserting unattached config.c s< t |�jr(t�� }tj|j�dd�d��� | fd|i|��S )N�namer@ )Z account_namer3 )r rC r �get_machine_token_filer
ZAlreadyAttachedErrorZaccount�get)r, r3 r- �machine_token_filer9 r r r6 � s
�z"assert_not_attached.<locals>.new_fr r7 r r9 r �assert_not_attached� s rL c s � fdd�}|S )Nc s t � ��� fdd��}|S )Nc s: t | dd�}t�t|�s&tj|� d���| fd|i|��S )N�security_issuer@ )Zissue�cmdr3 )rD �re�matchr r
Z InvalidSecurityIssueIdFormat)r, r3 r- rM )rN r5 r r r6 � s �z@assert_vulnerability_issue_valid.<locals>.wrapper.<locals>.new_fr r7 �rN r9 r r: � s
z1assert_vulnerability_issue_valid.<locals>.wrapperr )rN r: r rQ r � assert_vulnerability_issue_valid� s rR c C sn t j||d�\}}|r:|r:tjd�|�| d�|�dd��n0|rTtjd�|�| d��ntj| d�|�dd��dS )z�Raises a custom error for enable/disable commands when unattached.
Takes into consideration if the services exist or not, and notify the user
accordingly.)�namesr3 z, r@ )�
valid_service� operation�invalid_service� service_msg)rT rU )rU rV rW N)r Zget_valid_entitlement_namesr
Z UnattachedMixedServicesError�joinZ UnattachedValidServicesErrorZ UnattachedInvalidServicesError)r? rB r3 Zentitlements_foundZentitlements_not_foundr r r �&_raise_enable_disable_unattached_error� s* �� ��rY )r3 �returnc C s| t �| �}|j}|r*t�tjj|d�� n t�tj� t �
� t � | � t �
| �\}}t
�|�}t�t�|�� t�� d S )N)�
contract_name)r rI r[ r" r# r ZATTACH_SUCCESS_TMPLr= Z ATTACH_SUCCESS_NO_CONTRACT_NAMEr �stopZcleanupr r
Zformat_tabularr Zhandle_unicode_charactersZprocess_events)r3 rK r[ Z status_dictZ_ret�outputr r r �post_cli_attach� s
�
r^ )r3 �scope�set_key� set_valuerZ c C sR |t jkr| j}| j}n|t jkr.| j}| j}d|kr<|}n|}t|||d� dS )zS
Handles setting part the apt proxies - global and uaclient scoped proxies
Zhttps)�
http_proxy� https_proxyZ proxy_scopeN)r ZGLOBALZglobal_apt_http_proxyZglobal_apt_https_proxyZUACLIENTZua_apt_http_proxyZua_apt_https_proxyr )r3 r_ r` ra rb rc r r r �configure_apt_proxy� s
�rd c C sZ t �ddddg�}| �� sVtj�d�|t|�t�� tj� � t
� d� tj�d� qd S )N�|�/�-�\z
{}{}{}g�������?�) � itertools�cycleZis_set�sys�stdout�writer= �next�CLEAR_LINE_ANSI_CODE�flush�time�sleep)� stop_spinner�msgZspinnerr r r � run_spinner� s �
rv r@ �ru c s � fdd�}|S )Nc s t � �� �fdd��}|S )Nc s� t j�� s � | fd|i|��S t�� }tjt|�fd�}|�� � | fd|i|��}|�� |� � t j�
dt � t j� � |S )Nr3 )�targetr, �
)
rl rm �isatty� threadingZEventZThreadrv �start�setrX rn rp rq )r, r3 r- rt Zspinner_threadr4 )r5 ru r r r6 � s
�
z,with_spinner.<locals>.wrapper.<locals>.new_fr r7 rw r9 r r: � s z with_spinner.<locals>.wrapperr )ru r: r rw r � with_spinner� s r~ c C sx | dkr t jj| t jj S | dkr8t jj| t jj S | dkrTt jj| t jj S | dkrpt jj| t jj S | S d S )NZlowZmediumZhighZcritical)r ZTxtColorZINFOBLUEZENDCZ
WARNINGYELLOWZORANGEZFAIL)Zpriorityr r r �colorize_priority s �r )N)N)r@ )4rj rO rl r{ rr � functoolsr �typingr Zuaclientr r r r r r
r r r
r Z'uaclient.api.u.pro.security.fix._commonr Z(uaclient.api.u.pro.status.is_attached.v1r Z uaclient.aptr r Zuaclient.configr Zuaclient.filesr Zget_event_loggerr" rp ZAbstractProgressr r. r( r0 r; r< r> rG rL rR rY r^ r* rd rv r~ r r r r r �<module> sD 0
�