File: //proc/thread-self/root/lib/python3/dist-packages/uaclient/cli/__pycache__/attach.cpython-38.pyc
U
��Jh� � @ 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_code�service)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_foundr0 Zent_ret�reasonr) Zcontract_clientr& r& r' �
action_attachD sx
� � ����� �
r9 ZattachT� r" �?)�help�nargsz--no-auto-enableZ store_falser2 )r< �action�destz--attach-config�r)r< �typer Zcli)�defaultZstorer )r< r> �choicesrB )Z arguments)r< Z descriptionr> Zpreserve_descriptionZ
help_categoryZ
help_positionZargument_groups)4�argparser5 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_filer9 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 ��� �����