File: //lib/python3/dist-packages/uaclient/cli/__pycache__/enable.cpython-38.pyc
U
��JhD � @ s� d dl Z d dlZd dlmZmZmZmZmZ d dlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZ d dlmZmZ d dlmZmZmZ d dlmZm Z d dl m Z d dl m Z d d l!m"Z"m#Z#m$Z$ d d
l%m&Z& e�'e�(e)��Z*ed d e+fd
e+fdeee,ef fg�Z-d5e
j.e+ee j/ d�dd�Z0e
j.e,ee ee e,e,e,d�dd�Z1e+ee,ef ee, ee, eee,ef eee,ef e+d�dd�Z2e
j.e,e,e+e+e+eee, ee ee e-d�
dd�Z3e j4e j5e �6e j7�e8d�d d ����Z9e$d ej:ej;e9e&j<d e#e"d ej=j>d!�?e j@e
�.� d"��d#�d$d%d&�e"d'ejAj>d d(�d)d*�e"d+ejBd)d*�e"d,ejCd)d*�e"d-ejDj>d.d/�d$d.d0gd.d1�e"d2ejEd$d*�gd3�gd4�ZFdS )6� N)�Any�Dict�List�
NamedTuple�Optional)
�api�config�contract� entitlements� event_logger�
exceptions�lock�messages�status�util)�ServiceWithDependencies�
_dependencies)�
EnableOptions� EnableResult�_enable)�EnabledService�_enabled_services)� _is_attached)�cli_util)� ProArgument�ProArgumentGroup�
ProCommand)� HelpCategory�_EnableOneServiceResult�success� needs_reboot�error)�cfg� access_only�progress_objectc
C s� t �|�}tj| d||d�}d}d}z,tjdd�� |j|d�\}}W 5 Q R X W n. tk
r~ } zt�� |�W 5 d}~X Y nX |s�|dk r�|j dk r�|j } nt
j } t j
d| d��tdgg dg d �S )
a
Landscape gets special treatment because it currently not supported by our
enable API. This function is a temporary workaround until we have a proper
API for enabling landscape, which will happen after Landscape is fully
integrated with the contracts backend.
� landscape)� called_namer# �
extra_argsFNz cli.enable._enable_landscape)Z lock_holder)�progress)�service�reason)�enabledZdisabled�reboot_requiredr )r ZProgressWrapperr
ZLandscapeEntitlementr
Z RetryLock�enable� ExceptionZclear_lock_file_if_present�messager ZGENERIC_UNKNOWN_ISSUEr �EntitlementNotEnabledErrorr )
r" r# r' r$ r( r% r Z fail_reason�er* � r2 �5/usr/lib/python3/dist-packages/uaclient/cli/enable.py�_enable_landscape/ s>
�� � �r4 )r" r) �all_dependencies�enabled_servicesr& �variant�
service_titlec s� g }g }dd� |D �� t �fdd�|D �d �} | d k r^� fdd�| jD �}� fdd�| jD �}|D ]N}
tj| jdd�} t�| |
�} tj j
|| d �}
| s�tj |
d
�sbt j
|| d ��qb|D ]:}t�| |�}tjj
||d �}
tj |
d
�s�t j||d ��q�t ��fd d�|D �d �}|d k �r��d k �r�t�| ���}t�| �|j�}tj| jdd�} tj j
||d �}
| �srtj |
d
��s�t j
||d ��d S )
Nc S s g | ]
}|j �qS r2 ��name��.0�sr2 r2 r3 �
<listcomp>f s z2prompt_for_dependency_handling.<locals>.<listcomp>c 3 s | ]}|j � kr|V qd S �Nr9 r; )r) r2 r3 � <genexpr>i s
z1prompt_for_dependency_handling.<locals>.<genexpr>c s g | ]}|j � kr|j �qS r2 r9 r; ��enabled_service_namesr2 r3 r> l s
�c s g | ]}|j � kr|j �qS r2 r9 r; rA r2 r3 r> q s
�z features.block_disable_on_enable)r Z
path_to_value)�service_being_enabled�incompatible_service)�msg)rC �required_servicec 3 s, | ]$}|j � kr|jr|j�kr|V qd S r? �r: �variant_enabled� variant_namer; )r) r7 r2 r3 r@ � s
�)�nextZincompatible_withZ
depends_onr Zis_config_value_truer" r
Z get_titler ZINCOMPATIBLE_SERVICE�formatZprompt_for_confirmationr Z IncompatibleServiceStopsEnableZREQUIRED_SERVICEZRequiredServiceStopsEnablerI )r" r) r5 r6 r& r7 r8 Zincompatible_servicesZrequired_servicesZ dependenciesrD Zcfg_block_disable_on_enableZincompatible_service_titleZuser_msgrF Zrequired_service_titlerH Zto_be_enabled_titleZenabled_variant_titler2 )rB r) r7 r3 � prompt_for_dependency_handling[ s� �
�
�� ���� � � ��
���
��rL �� json_output�
json_response�processed_services�failed_services�errors�warningsr c C s^ | rZ|� � |� � |r dnd|d< ||d< ||d< ||d< ||d< ttj|tjdd �� d S )
Nr Zfailure�resultrP rQ rR rS T)�clsZ sort_keys)�sort�print�json�dumpsr ZDatetimeAwareJSONEncoderrM r2 r2 r3 �_print_json_output� s ��rZ )
r" �ent_namer7 r# �
assume_yesrN r' r6 r5 �returnc s� t �|�} tj| |�||d�}
|
j� |
j} t� �fdd�|D �d �} | d k r�tjj | d�}
| |
j
� | tj j | d�� t ddd||
j
|
jd�d�S |�szt
| � |||�| d � W n^ tjk
�r } z<t�|� | |j
� | tj j | d�� t ddd d� W Y �S d }~X Y nX z�|�r d }n t j|d
�}� d k�rHt| |||d �}ntt|�|d
�| |d�}tj| d� |�r�| tjj | d�� n| tjj | d�� |j}|�r�| tjj dd�� |jD ]}| |� �q�t d|d d�W S tjk
�rZ } z^|j d }|d }|d }|d }| |� | tj j | d�� t ddd||||d�d� W Y �S d }~X Y nf tjk
�r� } zD| |j
� | tj j | d�� t ddd||j
|j |j d�d� W Y �S d }~X Y nX d S )N)r7 r# r' c 3 s0 | ](}|j � kr�r$|jr|j�kr|V qd S r? rG r; �Z real_namer7 r2 r3 r@ � s
�z&_enable_one_service.<locals>.<genexpr>)�titleFr) )�typer) r/ � message_code)r r r! )r& r7 r8 )r\ r% )r' r$ )r) r7 r# )r$ �r" �install)Z operationTr* �coder_ �additional_info�r` r) r/ ra re ) r �&create_interactive_only_print_functionr
Zentitlement_factoryr: r_ rJ r ZALREADY_ENABLEDrK rE Z
ENABLE_FAILEDr rL r �UbuntuProError�LOGZ exceptionZCLIEnableDisableProgressr4 r r r ZACCESS_ENABLED_TMPLZ ENABLED_TMPLr, ZENABLE_REBOOT_REQUIRED_TMPLr0 re �msg_code)r" r[ r7 r# r\ rN r' r6 r5 �interactive_only_printZentZ ent_titleZalready_enabledrE r1 r( Z
enable_resultr r/ r* Zerr_code�err_msgZerr_infor2 r^ r3 �_enable_one_service� s� �� � �
�� �
��
���
� ���
�
��
��rm )r] c K s6 g }g }g }g }t jdd�}| jdk}t�|�} t| dd�}
| j} | j} |
r^| r^tj ddd��| t
j � zt �
|� W nD tjtjfk
r� tjd d
d � |�d t
jjt
jjd
�� Y nX t|�j�s
t�� }
| |
j� |�d |
j|
jd
�� t||||||dd� dS t| dg �}t� ||�\}}t |�j }t |�j }d
}t�!||�D ]n}t"|||
| | ||�#d�||� }|j$�r�|�|� |j%�r�d
|d< n&d}|�|� |j&dk �rJ|�|j&� �qJ|�r d}||7 }tj'||d�}| |j� |�d d|j|j|j(d�� t �)|�}|�*� t|||||||d� |�r2dS dS )z^Perform the enable action on a named entitlement.
@return: 0 on success, 1 otherwise
F)Z_schema_versionr rX r7 � �
--access-only� --variant)Zoption1Zoption2zFailed to refresh contractT)�exc_info�system)r` r/ ra )r � r) r' r Nrb rf r )+r ZJSON_SCHEMA_VERSIONrK r rg �getattrr# r\ r ZInvalidOptionCombinationr ZREFRESH_CONTRACT_ENABLEr ZrefreshZConnectivityErrorrh ri Zwarning�appendZE_REFRESH_CONTRACT_FAILURErE r: r Z is_attached_and_contract_validZContractExpiredErrorrj rZ r
Zget_valid_entitlement_namesr r6 r �servicesZ order_entitlements_for_enablingrm �getr r r! Z*create_enable_entitlements_not_found_errorre ZUAContractClientZupdate_activity_token)�argsr" �kwargsrP rQ rR rS rO rN rk r7 r# r\ Z expired_err�namesZentitlements_foundZentitlements_not_foundr6 r5 Zretr[ rT �errZcontract_clientr2 r2 r3 �
action_enableu s� �
� �
��
���
�
��
�
��
�
r| r- � r) z, rb )ZoptionsZstore�+)�help�action�nargsz --assume-yes)Zcommand�
store_true)r r� ro z--betaz--formatZcli)�defaultrX )r r� �choicesr� rp )Z arguments)r Z descriptionr� Z
help_categoryZ
help_positionZargument_groups)N)GrX Zlogging�typingr r r r r Zuaclientr r r r
r r r
r r r Z+uaclient.api.u.pro.services.dependencies.v1r r Z%uaclient.api.u.pro.services.enable.v1r r r Z-uaclient.api.u.pro.status.enabled_services.v1r r Z(uaclient.api.u.pro.status.is_attached.v1r Z uaclient.clir Zuaclient.cli.commandsr r r Zuaclient.cli.parserr Z getLoggerZ replace_top_level_logger_name�__name__ri �bool�strr ZUAConfigZAbstractProgressr4 rL rZ rm Zverify_json_format_argsZ assert_rootZassert_attachedZ&_raise_enable_disable_unattached_error�intr| ZCLI_ROOT_ENABLEZCLI_ENABLE_DESCZ
QUICKSTARTZCLI_ENABLE_SERVICErK �joinZvalid_servicesZCLI_ASSUME_YESZCLI_ENABLE_ACCESS_ONLYZCLI_ENABLE_BETAZCLI_FORMAT_DESCZCLI_ENABLE_VARIANTZenable_commandr2 r2 r2 r3 �<module> s� 0 �� �� -� ]
�
� #
���
��� ������