OwlCyberSecurity - MANAGER
Edit File: attach.cpython-38.pyc
U Uٵg� � @ sh d dl Z d dlZd dlmZmZmZmZmZmZ d dl m Z d dl m Z m Z d dlmZmZ d dlmZ d dlmZmZmZ d dlmZ d d lmZmZ d d l m Z m Z d d l m Z d d l!m"Z" e�#� Z$d d� Z%ej&ej'e�(d�dd� ���Z)edej*ej+e)dej,deedej-dd�edej.ddd�ed ej/e �0d �d �ed ej1j2d d!�d"d d#gd d$�gd%�gd&�Z3dS )'� N)�actions�contract� event_logger� exceptions�messages�secret_manager)� _initiate)�MagicAttachRevokeOptions�_revoke)�MagicAttachWaitOptions�_wait)�cli_util)� ProArgument�ProArgumentGroup� ProCommand)� HelpCategory)�AttachActionsConfigFile�IncorrectTypeError)�*create_enable_entitlements_not_found_error�get_valid_entitlement_names)�CanEnableFailure)� safe_loadc K s� | j dkrtjd| j d��t�tj� t|d�}t�dtjj |j d� � t |j d�}zt ||d�}W nL tj k r� } z,t�tj� t|j d�}t||d� |�W 5 d }~X Y nX t�dtj � |jS ) N�json�--format)Zparam�value��cfg� )� user_code)Z magic_token)Zoptionsr )�formatr ZMagicAttachInvalidParam�event�infor ZCLI_MAGIC_ATTACH_INITr ZCLI_MAGIC_ATTACH_SIGN_INr r �tokenr ZMagicAttachTokenErrorZCLI_MAGIC_ATTACH_FAILEDr r ZCLI_MAGIC_ATTACH_PROCESSINGZcontract_token)�argsr �kwargsZ initiate_respZ wait_optionsZ wait_resp�eZrevoke_options� r& �5/usr/lib/python3/dist-packages/uaclient/cli/attach.py� _magic_attach# s4 � ��� � r( z pro attachc K s� | j r| jrt�� �n�| j s4| js4t| |d�}d }nv| j rR| j }tj�|� d }nXzt� t | j��}W n6 t k r� } ztj | jj |jd��W 5 d }~X Y nX |j }|j}| jo�|d k}ztj|||d� W n tjk r� t�� �Y n�X d}|d k �r�| j�r�t||�\} } | D ]t} t�|| �\} } | �s|d}| d k �r�t| t��r�| jd k �r�t�| jj� tj | jj| jj | d� n t� | � �q| �r�t | |d�}tj|jt j d� tj |j|j!d� d}t"�#|�}|�$� t%�&|� |S d S ) Nr )Z config_name�error)r" � allow_enabler � )� error_msg� error_codeZservice)Z file_type)r, r- )'r" � attach_configr ZCLIAttachTokenArgXORConfigr( r ZsecretsZ add_secretr Z from_dictr r Z AttachInvalidConfigFileError�name�msgZenable_services� auto_enabler Zattach_with_tokenZConnectivityErrorZ AttachErrorr Zenable_entitlement_by_name� isinstancer �messager r! r) Zservice_processedr �sys�stderrZmsg_coder ZUAContractClientZupdate_activity_tokenr Zpost_cli_attach)r# r r$ r" Zenable_services_overrider. r% r* Zret�foundZ not_foundr/ Zent_ret�reasonr) Zcontract_clientr& r& r' � action_attachD sx � � ����� � r8 ZattachT� r" �?)�help�nargsz--no-auto-enableZ store_falser1 )r; �action�destz--attach-config�r)r; �typer Zcli)�defaultZstorer )r; r= �choicesrA )Z arguments)r; Z descriptionr= Zpreserve_descriptionZ help_categoryZ help_positionZargument_groups)4�argparser4 Zuaclientr r r r r r Z+uaclient.api.u.pro.attach.magic.initiate.v1r Z)uaclient.api.u.pro.attach.magic.revoke.v1r r Z'uaclient.api.u.pro.attach.magic.wait.v1r r Z uaclient.clir Zuaclient.cli.commandsr r r Zuaclient.cli.parserr Zuaclient.data_typesr r Zuaclient.entitlementsr r Z(uaclient.entitlements.entitlement_statusr Z uaclient.yamlr Zget_event_loggerr r( Zassert_not_attachedZ assert_rootZassert_lock_filer8 ZCLI_ROOT_ATTACHZCLI_ATTACH_DESCZ QUICKSTARTZCLI_ATTACH_TOKENZCLI_ATTACH_NO_AUTO_ENABLEZCLI_ATTACH_ATTACH_CONFIGZFileTypeZCLI_FORMAT_DESCr Zattach_commandr& r& r& r' �<module> sl !D ��� �����