File: //usr/lib/python3/dist-packages/uaclient/cli/__pycache__/enable.cpython-38.pyc
U
���hAK � @ 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& d d l'm(Z( e�)e�*e+��Z,ed d
e-fde-fdeee.ef fg�Z/e
j0e.e-d�dd�Z1d:e
j0e-ee j2 d�dd�Z3e
j0e.ee ee e.e.e.d�dd�Z4e-ee.ef ee. ee. eee.ef eee.ef e-d�dd�Z5e
j0e.e.e-e-e-eee. ee ee e/d �
d d �Z6e j7e j8e �9e j:�e;d �d d!����Z<e$d"ej=ej>e<e&j?d#e#e"d$ej@jAd%�Be jCe
�0� d&��d'�d(d)d*�e"d+ejDd,d-�e"d.ejEjAd"d/�d,d-�e"d0ejFd,d-�e"d1ejGd,d-�e"d2ejHjAd3d4�d(d3d5gd3d6�e"d7ejId(d-�gd8�gd9�ZJdS );� 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)�
machine_token�_EnableOneServiceResult�success� needs_reboot�error)�cfg�variant�
assume_yesc
C st t �|�}t�| �}t�| |�� �}|s6|tj� d S t | �j
}t | �j }|D ] } t
| | j|d||d ||d� qNd S )NF) r# �ent_namer$ � access_onlyr% � json_output�
extra_args�enabled_services�all_dependencies)r �&create_interactive_only_print_functionr Zget_machine_token_filer Z get_enabled_by_default_servicesr
r ZNO_SERVICES_TO_AUTO_ENABLEr r* r �services�_enable_one_service�name)
r# r$ r% r( �interactive_only_printZmachine_token_fileZservices_to_be_enabledr* r+ Zenable_by_default_service� r1 �5/usr/lib/python3/dist-packages/uaclient/cli/enable.py�_auto_enable_services0 s2 �
�
�r3 )r# r' �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' r) FNz 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) r4 r7 r5 r Z fail_reason�er9 r1 r1 r2 �_enable_landscapeS s>
�� � �rA )r# r8 r+ r* r6 r$ �
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 r1 �r/ ��.0�sr1 r1 r2 �
<listcomp>� s z2prompt_for_dependency_handling.<locals>.<listcomp>c 3 s | ]}|j � kr|V qd S �NrC rD )r8 r1 r2 � <genexpr>� s
z1prompt_for_dependency_handling.<locals>.<genexpr>c s g | ]}|j � kr|j �qS r1 rC rD ��enabled_service_namesr1 r2 rG � s
�c s g | ]}|j � kr|j �qS r1 rC rD rJ r1 r2 rG � s
�z features.block_disable_on_enable)r Z
path_to_value)�service_being_enabled�incompatible_service)�msg)rL �required_servicec 3 s, | ]$}|j � kr|jr|j�kr|V qd S rH �r/ �variant_enabled� variant_namerD )r8 r$ r1 r2 rI � s
�)�nextZincompatible_withZ
depends_onr Zis_config_value_truer# r
Z get_titler ZINCOMPATIBLE_SERVICE�formatZprompt_for_confirmationr Z IncompatibleServiceStopsEnableZREQUIRED_SERVICEZRequiredServiceStopsEnablerR )r# r8 r+ r* r6 r$ rB Zincompatible_servicesZrequired_servicesZ dependenciesrM Zcfg_block_disable_on_enableZincompatible_service_titleZuser_msgrO Zrequired_service_titlerQ Zto_be_enabled_titleZenabled_variant_titler1 )rK r8 r$ r2 � prompt_for_dependency_handling s� �
�
�� ���� � � ��
���
��rU �r( �
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�resultrX rY rZ r[ T)�clsZ sort_keys)�sort�print�json�dumpsr ZDatetimeAwareJSONEncoderrV r1 r1 r2 �_print_json_output� s ��rb )
r# r& r$ r' r% r( r) r* r+ �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)r$ r' r) c 3 s0 | ](}|j � kr�r$|jr|j�kr|V qd S rH rP rD �Z real_namer$ r1 r2 rI s
�z&_enable_one_service.<locals>.<genexpr>)�titleFr8 )�typer8 r> � message_code)r r! r" )r6 r$ rB )r% r5 )r) r4 )r8 r$ r' )r4 �r# �install)Z operationTr9 �codere �additional_info�rf r8 r> rg rk ) r r, r
Zentitlement_factoryr/ re rS r ZALREADY_ENABLEDrT rN Z
ENABLE_FAILEDr rU r �UbuntuProError�LOGZ exceptionZCLIEnableDisableProgressrA r r r ZACCESS_ENABLED_TMPLZ ENABLED_TMPLr; ZENABLE_REBOOT_REQUIRED_TMPLr? rk �msg_code)r# r& r$ r' r% r( r) r* r+ r0 ZentZ ent_titleZalready_enabledrN r@ r7 Z
enable_resultr! r> r9 Zerr_code�err_msgZerr_infor1 rd r2 r. � s� �� � �
�� �
��
���
� ���
�
��
��r. )rc c K s� g }g }g }g }t jdd�}| jdk}t�|�} t| dd�}
| j} | j} | j}
|
rd| rdt j
ddd��|
rz|
rzt j
dd d��| r�|
r�t j
dd d��| t j � zt
j|dd
� W nD t jt jfk
r� tjd d d
� |�dt jjt jjd�� Y nX t|�j�s@t �� }| |j� |�d|j|jd�� t||||||dd� dS |
�rVt |dd |d�S t| dg �}|�spt � � �t � ||�\}}t |�j!}t"|�j#}d }t �$||�D ]n}t%|||
| | ||�&d�||� }|j'�r�|�|� |j(�rd |d< n&d}|�|� |j)dk �r�|�|j)� �q�|�rZd}||7 }t j*||d�}| |j� |�dd|j|j|j+d�� t
�,|�}|�-� t|||||||d� |�r�dS dS )z^Perform the enable action on a named entitlement.
@return: 0 on success, 1 otherwise
F)Z_schema_versionr! r` r$ � �
--access-only� --variant)Zoption1Zoption2�--auto)�verbosezFailed to refresh contractT)�exc_info�system)rf r> rg )r � )r# r$ r% r( r8 r) r! Nrh rl r ).r ZJSON_SCHEMA_VERSIONrT r r, �getattrr' r% �autor ZInvalidOptionCombinationr ZREFRESH_CONTRACT_ENABLEr ZrefreshZConnectivityErrorrm rn Zwarning�appendZE_REFRESH_CONTRACT_FAILURErN r/ r Z is_attached_and_contract_validZContractExpiredErrorro rb r3 ZNoServicesToEnabler
Zget_valid_entitlement_namesr r* r r- Z order_entitlements_for_enablingr. �getr r! r" Z*create_enable_entitlements_not_found_errorrk ZUAContractClientZupdate_activity_token)�argsr# �kwargsrX rY rZ r[ rW r( r0 r$ r' r% rz Z expired_err�namesZentitlements_foundZentitlements_not_foundr* r+ Zretr& r\ �errZcontract_clientr1 r1 r2 �
action_enable� s �
� � � �
��
��� �
�
��
�
��
�
r� r< � r8 z, rh )ZoptionsZstore�*)�help�action�nargsrr �
store_true)r� r� z --assume-yes)Zcommandrt z--betaz--formatZcli)�defaultr` )r� r� �choicesr� rs )Z arguments)r� Z descriptionr� Z
help_categoryZ
help_positionZargument_groups)N)Kr` 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 Zuaclient.filesr Z getLoggerZ replace_top_level_logger_name�__name__rn �bool�strr ZUAConfigr3 ZAbstractProgressrA rU rb r. Zverify_json_format_argsZ assert_rootZassert_attachedZ&_raise_enable_disable_unattached_error�intr� ZCLI_ROOT_ENABLEZCLI_ENABLE_DESCZ
QUICKSTARTZCLI_ENABLE_SERVICErT �joinZvalid_servicesZCLI_ENABLE_ACCESS_ONLYZCLI_ASSUME_YESZCLI_ENABLE_AUTOZCLI_ENABLE_BETAZCLI_FORMAT_DESCZCLI_ENABLE_VARIANTZenable_commandr1 r1 r1 r2 �<module> s� 0 �� � ' �� -� ]
�
� #
���
� ��� ������