Server : Apache/2.4.52 (Ubuntu) System : Linux webserver 6.8.0-49-generic #49~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Wed Nov 6 17:42:15 UTC 2 x86_64 User : www-data ( 33) PHP Version : 8.1.2-1ubuntu2.21 Disable Function : NONE Directory : /lib/python3/dist-packages/uaclient/cli/__pycache__/ |
o
���f�B � @ 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 m!Z! e�"e�#e$��Z%ed d
e&fd e&fd eee'ef fg�Z( d,d
e
j)de&dee j* fdd�Z+d
e
j)de'dee dee de'de'de'fdd�Z,de&dee'ef d ee' d ee' d eee'ef d eee'ef d
e&fd d!�Z-d
e
j)d"e'de'de&d#e&de&d$eee' dee dee d%e(fd&d'�Z.e j/e j0e �1e j2�d%e3fd(d)����Z4d
e
j)fd*d+�Z5dS )-� 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� constants�_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( � r7 �5/usr/lib/python3/dist-packages/uaclient/cli/enable.py�_enable_landscape- s>
�� �����
�r9 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 r7 ��name��.0�sr7 r7 r8 �
<listcomp>d s z2prompt_for_dependency_handling.<locals>.<listcomp>c 3 s � | ]
}|j � kr|V qd S �Nr> r@ )r' r7 r8 � <genexpr>g s � z1prompt_for_dependency_handling.<locals>.<genexpr>c s g | ]
}|j � v r|j �qS r7 r> r@ ��enabled_service_namesr7 r8 rC j �
�c s g | ]
}|j � vr|j �qS r7 r> r@ rF r7 r8 rC o rH z features.block_disable_on_enable)r �
path_to_value)�service_being_enabled�incompatible_service)�msg)rJ �required_servicec 3 s. � | ]}|j � kr|jr|j�kr|V qd S rD �r? �variant_enabled� variant_namer@ )r' r< r7 r8 rE � s �
�
��)�next�incompatible_with�
depends_onr �is_config_value_truer r
� get_titler �INCOMPATIBLE_SERVICE�format�prompt_for_confirmationr � IncompatibleServiceStopsEnable�REQUIRED_SERVICE�RequiredServiceStopsEnablerP )r r' r: r; r# r< r= �incompatible_services�required_services� dependenciesrK �cfg_block_disable_on_enable�incompatible_service_title�user_msgrM �required_service_titlerO �to_be_enabled_title�enabled_variant_titler7 )rG r' r< r8 � prompt_for_dependency_handlingY s� �
�
������� � �� ��
������
re � 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�resultrh ri rj rk T)�cls� sort_keys)�sort�print�json�dumpsr �DatetimeAwareJSONEncoder)rf rg rh ri rj rk r r7 r7 r8 �_print_json_output� s ���ru �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 rD rN r@ �� real_namer< r7 r8 rE � s �
��
��z&_enable_one_service.<locals>.<genexpr>)�titleFr' )�typer' r2 � message_code)r r r )r# r< r= )rw r" )r$ r! )r' r< r )r! �r �install)� operationTr( �coder{ �additional_info�r| r' r2 r} r� ) r �&create_interactive_only_print_functionr
�entitlement_factoryr? r{ rQ r �ALREADY_ENABLEDrW rL �
ENABLE_FAILEDr re r �UbuntuProError�LOG� exception�CLIEnableDisableProgressr9 r r r �ACCESS_ENABLED_TMPL� ENABLED_TMPLr+ �ENABLE_REBOOT_REQUIRED_TMPLr4 r� �msg_code)r rv r< r rw rf r$ r; r: �interactive_only_print�ent� ent_title�already_enabledrL r6 r& �
enable_resultr r2 r( �err_code�err_msg�err_infor7 ry r8 �_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 rr r< � �
--access-only� --variant)�option1�option2zFailed to refresh contractT)�exc_info�system)r| r2 r} )r � r' r$ r Nr~ r� r )+r �JSON_SCHEMA_VERSIONrW r r� �getattrr rw r �InvalidOptionCombinationr �REFRESH_CONTRACT_ENABLEr �refresh�ConnectivityErrorr� r� �warning�append�E_REFRESH_CONTRACT_FAILURErL r? r � is_attached_and_contract_valid�ContractExpiredErrorr� ru 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 �kwargsrh ri rj rk rg rf r� r< r rw � expired_err�names�entitlements_found�entitlements_not_foundr; r: �retrv rm �err�contract_clientr7 r7 r8 �
action_enables s� �
� �
���
���
�
��
�
��
��
�
r� c
C s� | j dtjd�}|jtd� tj|_tjj tj
dd�|_ d|_ tj
|j_tj|j_|jdddtjj d �tj|d
��d �d � |jd
dtjj dd�d� |jddtjd� |jddtjd� |jddddgdtjj dd�d� |jddtjd� |S )Nr/ )�help)�actionz enable <service> [<service>])r? �commandr' �store�+z, r~ )�options)r� �nargsr� z --assume-yes�
store_true)r� )r� r� r� z--betaz--format�clirr )�default)r� �choicesr� r� r� ) �
add_parserr �CLI_ROOT_ENABLE� set_defaultsr� �CLI_ENABLE_DESC� descriptionr �
USAGE_TMPLrW �NAME�usage�prog�CLI_ARGS� _positionalsr{ � CLI_FLAGS�
_optionals� add_argument�CLI_ENABLE_SERVICE�joinr
�valid_services�CLI_ASSUME_YES�CLI_ENABLE_ACCESS_ONLY�CLI_ENABLE_BETA�CLI_FORMAT_DESC�CLI_ENABLE_VARIANT)�
subparsersr �parserr7 r7 r8 r� � sR �
��
��� ��r� rD )6rr �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 r � getLogger� replace_top_level_logger_name�__name__r� �bool�strr �UAConfig�AbstractProgressr9 re ru r� �verify_json_format_args� assert_root�assert_attached�&_raise_enable_disable_unattached_error�intr� r� r7 r7 r7 r8 �<module> s� 0 �����
�,������
�\�
�����
� ������
�� �
� #