OwlCyberSecurity - MANAGER
Edit File: cli_util.cpython-38.pyc
U ��gQ � @ 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� Zservice)r? � service_namesr3 r3 )r � is_attached�getattrr ZUnattachedError)r, r3 r- r? rA )r5 �raise_custom_error_functionr r r6 w s �z/assert_attached.<locals>.wrapper.<locals>.new_fr r7 �rD r9 r r: v s z assert_attached.<locals>.wrapperr )rD r: r rE r �assert_attachedp s rF 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 rB 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 rK 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 )rC �re�matchr r Z InvalidSecurityIssueIdFormat)r, r3 r- rL )rM r5 r r r6 � s �z@assert_vulnerability_issue_valid.<locals>.wrapper.<locals>.new_fr r7 �rM r9 r r: � s z1assert_vulnerability_issue_valid.<locals>.wrapperr )rM r: r rP r � assert_vulnerability_issue_valid� s rQ 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)rS rT )rT rU rV N)r Zget_valid_entitlement_namesr Z UnattachedMixedServicesError�joinZ UnattachedValidServicesErrorZ UnattachedInvalidServicesError)r? rA r3 Zentitlements_foundZentitlements_not_foundr r r �&_raise_enable_disable_unattached_error� s* �� ��rX )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 rH rZ r" r# r ZATTACH_SUCCESS_TMPLr= Z ATTACH_SUCCESS_NO_CONTRACT_NAMEr �stopZcleanupr r Zformat_tabularr Zhandle_unicode_charactersZprocess_events)r3 rJ rZ Z status_dictZ_ret�outputr r r �post_cli_attach� s � r] )r3 �scope�set_key� set_valuerY 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_ r` ra rb r r r �configure_apt_proxy� s �rc 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 � ru r@ �rt 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, � ) rk rl �isatty� threadingZEventZThreadru �start�setrW rm ro rp )r, r3 r- rs Zspinner_threadr4 )r5 rt r r r6 � s � z,with_spinner.<locals>.wrapper.<locals>.new_fr r7 rv r9 r r: � s z with_spinner.<locals>.wrapperr )rt r: r rv 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@ )4ri rN rk rz rq � 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" ro ZAbstractProgressr r. r( r0 r; r< r> rF rK rQ rX r] r* rc ru r} r~ r r r r �<module> sD 0 �