CasperSecurity
o
��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- dBde
j.de+dee j/ fdd�Z0de
j.de,dee dee de,de,de,fdd�Z1d e+d ee,ef d ee, d ee, d eee,ef d!eee,ef d e+fd"d#�Z2de
j.d$e,de,de+d%e+d e+d&eee, dee dee d'e-fd(d)�Z3e j4e j5e �6e j7�d'e8fd*d+����Z9e$d,ej:ej;e9e&j<d-e#e"dej=j>d.�?e j@e
�.� d/��d0�d1d2d3�e"d4ejAj>d,d5�d6d7�e"d8ejBd6d7�e"d9ejCd6d7�e"d:ejDj>d;d<�d1d;d=gd;d>�e"d?ejEd1d7�gd@�gdA�ZFdS )C� 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 d � n1 s,w Y W n tyD } zt�� |�d}~ww |s^|durT|j durT|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)� lock_holder)�progress)�service�reason)�enabled�disabled�reboot_requiredr )r �ProgressWrapperr
�LandscapeEntitlementr
� RetryLock�enable� Exception�clear_lock_file_if_present�messager �GENERIC_UNKNOWN_ISSUEr �EntitlementNotEnabledErrorr )
r" r# r' r$ r) r% r � fail_reason�er+ � r: �5/usr/lib/python3/dist-packages/uaclient/cli/enable.py�_enable_landscape/ s>
�� �����
�r< r* �all_dependencies�enabled_servicesr& �variant�
service_titlec s� g }g }dd� |D �� t �fdd�|D �d �} | d ur/� fdd�| jD �}� fdd�| jD �}|D ]'}
tj| jdd�} t�| |
�} tj j
|| d �}
| sQtj |
d
�sXt j
|| d ��q1|D ] }t�| |�}tjj
||d �}
tj |
d
�sxt j||d ��q[t ��fd d�|D �d �}|d ur��d ur�t�| ���}t�| �|j�}tj| jdd�} tj j
||d �}
| s�tj |
d
�s�t j
||d ��d S d S d S )
Nc S s g | ]}|j �qS r: ��name��.0�sr: r: r; �
<listcomp>f s z2prompt_for_dependency_handling.<locals>.<listcomp>c 3 s � | ]
}|j � kr|V qd S �NrA rC )r* r: r; � <genexpr>i s � z1prompt_for_dependency_handling.<locals>.<genexpr>c s g | ]
}|j � v r|j �qS r: rA rC ��enabled_service_namesr: r; rF l �
�c s g | ]
}|j � vr|j �qS r: rA rC rI r: r; rF q rK z features.block_disable_on_enable)r �
path_to_value)�service_being_enabled�incompatible_service)�msg)rM �required_servicec 3 s. � | ]}|j � kr|jr|j�kr|V qd S rG �rB �variant_enabled� variant_namerC )r* r? r: r; rH � s �
�
��)�next�incompatible_with�
depends_onr �is_config_value_truer" r
� get_titler �INCOMPATIBLE_SERVICE�format�prompt_for_confirmationr � IncompatibleServiceStopsEnable�REQUIRED_SERVICE�RequiredServiceStopsEnablerS )r" r* r= r> r& r? r@ �incompatible_services�required_services� dependenciesrN �cfg_block_disable_on_enable�incompatible_service_title�user_msgrP �required_service_titlerR �to_be_enabled_title�enabled_variant_titler: )rJ r* r? r; � prompt_for_dependency_handling[ s� �
�
������� � �� ��
������
rh � json_output�
json_response�processed_services�failed_services�errors�warningsc C sb | r/|� � |� � |rdnd|d< ||d< ||d< ||d< ||d< ttj|tjdd �� d S d S )
Nr �failure�resultrk rl rm rn T)�cls� sort_keys)�sort�print�json�dumpsr �DatetimeAwareJSONEncoder)ri rj rk rl rm rn r r: r: r; �_print_json_output� s ���rx �ent_name�
assume_yesr' �returnc s� t �|�} tj| |�||d�}
|
j� |
j} t� �fdd�|D �d �} | d urItjj | d�}
| |
j
� | tj j | d�� t ddd||
j
|
jd�d�S |s�z
t
| � |||�| d � W n- tjy� } z t�|� | |j
� | tj j | d�� t ddd d�W Y d }~S d }~ww z_|r�d }nt j|d
�}� d kr�t| |||d �}n tt|�|d
�| |d�}tj| d� |r�| tjj | d�� n | tjj | d�� |j}|r�| tjj dd�� |jD ]}| |� q�t d|d d�W S tj�y$ } z1|j d }|d }|d }|d }| |� | tj j | d�� t ddd||||d�d�W Y d }~S d }~w tj�yU } z$| |j
� | tj j | d�� t ddd||j
|j |j d�d�W Y d }~S d }~ww )N)r? r# r' c 3 s2 � | ]}|j � kr�r|jr|j�kr|V qd S rG rQ rC �� real_namer? r: r; rH � s �
��
��z&_enable_one_service.<locals>.<genexpr>)�titleFr* )�typer* r5 � message_code)r r r! )r& r? r@ )rz r% )r' r$ )r* r? r# )r$ �r" �install)� operationTr+ �coder~ �additional_info�r r* r5 r� r� ) r �&create_interactive_only_print_functionr
�entitlement_factoryrB r~ rT r �ALREADY_ENABLEDrZ rO �
ENABLE_FAILEDr rh r �UbuntuProError�LOG� exception�CLIEnableDisableProgressr< r r r �ACCESS_ENABLED_TMPL� ENABLED_TMPLr. �ENABLE_REBOOT_REQUIRED_TMPLr7 r� �msg_code)r" ry r? r# rz ri r' r> r= �interactive_only_print�ent� ent_title�already_enabledrO r9 r) �
enable_resultr r5 r+ �err_code�err_msg�err_infor: r| r; �_enable_one_service� s �� � �
��
�
���� ���
� ���
�
���
����r� c K s( g }g }g }g }t jdd�}| jdk}t�|�} t| dd�}
| j} | j} |
r/| r/tj ddd��| t
j � zt �
|� W n! tjtjfy\ tjd d
d � |�d t
jjt
jjd
�� Y nw t|�js�t�� }
| |
j� |�d |
j|
jd
�� t||||||dd� dS t| dg �}t� ||�\}}t |�j }t |�j }d
}t�!||�D ]3}t"|||
| | ||�#d�||� }|j$r�|�|� |j%r�d
|d< q�d}|�|� |j&dur�|�|j&� q�|r�d}||7 }tj'||d�}| |j� |�d d|j|j|j(d�� t �)|�}|�*� t|||||||d� |�rdS dS )z^Perform the enable action on a named entitlement.
@return: 0 on success, 1 otherwise
F)�_schema_versionr ru r? � �
--access-only� --variant)�option1�option2zFailed to refresh contractT)�exc_info�system)r r5 r� )r � r* r' r Nr� r� r )+r �JSON_SCHEMA_VERSIONrZ r r� �getattrr# rz r �InvalidOptionCombinationr �REFRESH_CONTRACT_ENABLEr �refresh�ConnectivityErrorr� r� �warning�append�E_REFRESH_CONTRACT_FAILURErO rB r � is_attached_and_contract_valid�ContractExpiredErrorr� rx r
�get_valid_entitlement_namesr r> r �services� order_entitlements_for_enablingr� �getr r r! �*create_enable_entitlements_not_found_errorr� �UAContractClient�update_activity_token)�argsr" �kwargsrk rl rm rn rj ri r� r? r# rz � expired_err�names�entitlements_found�entitlements_not_foundr> r= �retry rp �err�contract_clientr: r: r; �
action_enableu s� �
� �
���
���
�
��
�
��
��
�
r� r2 � z, r� )�options�store�+)�help�action�nargsz --assume-yes)�command�
store_true)r� r� r� z--betaz--format�cli)�defaultru )r� r� �choicesr� r� )� arguments)r� � descriptionr� �
help_category�
help_position�argument_groupsrG )Gru �logging�typingr r r r r �uaclientr r r r
r r r
r r r �+uaclient.api.u.pro.services.dependencies.v1r r �%uaclient.api.u.pro.services.enable.v1r r r �-uaclient.api.u.pro.status.enabled_services.v1r r �(uaclient.api.u.pro.status.is_attached.v1r � uaclient.clir �uaclient.cli.commandsr r r �uaclient.cli.parserr � getLogger� replace_top_level_logger_name�__name__r� �bool�strr �UAConfig�AbstractProgressr< rh rx r� �verify_json_format_args� assert_root�assert_attached�&_raise_enable_disable_unattached_error�intr� �CLI_ROOT_ENABLE�CLI_ENABLE_DESC�
QUICKSTART�CLI_ENABLE_SERVICErZ �join�valid_services�CLI_ASSUME_YES�CLI_ENABLE_ACCESS_ONLY�CLI_ENABLE_BETA�CLI_FORMAT_DESC�CLI_ENABLE_VARIANT�enable_commandr: r: r: r; �<module> s
0 �����
�,������
�\�
�����
� ������
�� �
� #
���
��� �����
�