File: //proc/self/root/lib/python3/dist-packages/uaclient/cli/__pycache__/disable.cpython-38.pyc
U
��Jh�* � @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z m
Z
m Z m Z m
Z
mZmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d dlmZmZm Z d d l m Z d d
l m Z e �!� Z"e�#e�$e%��Z&ej'e(ee ee( e(e(d �d d
�Z)dd�dd�Z*ej+ej,e�-ej.�e�/d�dd� ����Z0e de
j1e
j2e0e j3eede
j4j5d�6e
j7e�'� d��d�ddd �ed e
j8j5dd �d d �ed!e
j9j5d"d#�dd"d$gd"d%�ed&e
j:d d �gd'�gd(�Z;dS ))� N)�Any�Dict�List)�config�contract� entitlements� event_logger�
exceptions�messages�status�util)�ProgressWrapper)�ServiceWithDependencies�
_dependencies)�_enabled_services)�cli_util)� ProArgument�ProArgumentGroup�
ProCommand)� HelpCategory)�CanDisableFailure)�cfg�service�all_dependencies�enabled_service_names� called_name�
service_titlec C s� g }|D ]<}|j |ks|j |kr"q|jD ]}|j |kr(|�|j � q(q|D ]:} t�| | �}
tjj||
d�} tj | d�sJt
j ||
d��qJd S )N)Zservice_being_disabled�dependent_service)�msg) �nameZ
depends_on�appendr Z get_titler
ZDEPENDENT_SERVICE�formatr Zprompt_for_confirmationr Z DependentServiceStopsDisable) r r r r r r Zdependent_services�sZ requirementr Zdependent_service_titleZuser_msg� r# �6/usr/lib/python3/dist-packages/uaclient/cli/disable.py� prompt_for_dependency_handling s*
�� �r% T)�
update_statusc C s� | j }|dk r|} |r t� }nttj|d��}| �|�\}}|s�t�| j� |dk r�t|t �r�|j
dk r�t� |j
j � tj
|j
j |j
j| jd� n t�| j� |r�tj|d� |S )a Perform the disable action on a named entitlement.
:param entitlement_name: the name of the entitlement to enable
:param cfg: the UAConfig to pass to the entitlement
:param json_output: output should be json only
@return: True on success, False otherwise
N��
assume_yes)Z error_msgZ
error_coder �r )�enabled_variantr
r �CLIEnableDisableProgress�disable�eventZservice_failedr �
isinstancer �message�infor �errorZservice_processedr ) Z entitlementr � json_outputr( r& �variant�progress�ret�reasonr# r# r$ �perform_disable> s.
�
� r7 z pro disablec K s~ g }g }g }g }t jddd�}| jdk}| j} t�|�}
| jrP| rPtjddd��t | dg �} t
� | |�\} }
d d
� t |�j
D �}t|�j}d }| D �]�}t
j||| jd �}|j}|d
k r�|}| �s:z t||j||||jd� W nZ tjk
�r8 } z8t�|� |
|j� |
tjj|jd�� d}W Y �q�W 5 d
}~X Y nX |�rHt� }nttj | d��}|� � |_ z�|� |�\}}t j |d� |�s�d}|�!|� |d
k �rt"|t#��r|j$d
k �r|
|j$j� |�!d|j|j$j|j$jd�� n2|�!|� |�%� }|�rd |d< |
tj&jdd�� W q� tjk
�r� } zNd}|�!|� |
|j� |
tj'j|jd�� |�!d|j|j|j(|j)d�� W 5 d
}~X Y q�X q�|
�r d}dd�*t
j+|d�� d }d�*t,j-|dddd ��}tj.d d�*|
�|d �}|
|j� |�!d d
|j|j(|j)d�� t/�0|�} | �1� |�rp|�2� |�2� |�r2dnd |d!< ||d"< ||d#< ||d$< ||d%< t3t4j5|t6j7d d&�� |�rzd'S d(S ))zbPerform the disable action on a list of entitlements.
@return: 0 on success, 1 otherwise
ZsuccessF)Z_schema_version�result� needs_reboot�json�--purge� --assume-yes)Zoption1Zoption2r c S s g | ]
}|j �qS r# )r )�.0r" r# r# r$ �
<listcomp>� s z"action_disable.<locals>.<listcomp>T)r r �purgeN)r r )�titler' r) )�typer r/ � message_coder9 zdisable operation)� operation)rA r r/ rB �additional_infozTry �, �.�
�P )�widthZbreak_long_wordsZbreak_on_hyphensr, )rC Zinvalid_service� service_msg�systemZfailurer8 �processed_services�failed_services�errors�warnings)�clsZ sort_keysr � )8r ZJSON_SCHEMA_VERSIONr! r( r Z&create_interactive_only_print_functionr? r ZInvalidOptionCombination�getattrr Zget_valid_entitlement_namesr Zenabled_servicesr �servicesZentitlement_factoryr* r% r r@ ZUbuntuProError�LOGZ exceptionr r
Z
ENABLE_FAILEDr
r+ Z calculate_total_disable_stepsZ total_stepsr, r r r. r r/ Z_check_for_rebootZENABLE_REBOOT_REQUIRED_TMPLZDISABLE_FAILED_TMPLZmsg_coderD �join�valid_services�textwrapZwrapZInvalidServiceOpErrorr ZUAContractClientZupdate_activity_token�sort�printr: �dumpsr ZDatetimeAwareJSONEncoder) �argsr �kwargsrL rM rN rO Z
json_responser2 r( Zinteractive_only_print�namesZentitlements_foundZentitlements_not_foundr r r5 Zent_nameZentr3 �er4 Z disable_retr6 Zent_reboot_requiredZ valid_namesrJ �errZcontract_clientr# r# r$ �action_disablel s( �
�
�
��
��
�
�
� ��
��
���
����
��
��r` r, r rE r) )ZoptionsZstore�+)�help�action�nargsr< )Zcommand�
store_true)rb rc z--formatZcli)�defaultr: )rb rc �choicesrf r; )Z arguments)rb Z descriptionrc Z
help_categoryZargument_groups)<r: ZloggingrW �typingr r r Zuaclientr r r r r r
r r Z uaclient.apir
Z+uaclient.api.u.pro.services.dependencies.v1r r Z-uaclient.api.u.pro.status.enabled_services.v1r Z uaclient.clir Zuaclient.cli.commandsr r r Zuaclient.cli.parserr Z(uaclient.entitlements.entitlement_statusr Zget_event_loggerr- Z getLoggerZ replace_top_level_logger_name�__name__rT ZUAConfig�strr% r7 Zverify_json_format_argsZ assert_rootZassert_attachedZ&_raise_enable_disable_unattached_errorZassert_lock_filer` ZCLI_ROOT_DISABLEZCLI_DISABLE_DESCZOTHERZCLI_DISABLE_SERVICEr! rU rV ZCLI_ASSUME_YESZCLI_FORMAT_DESCZ CLI_PURGEZdisable_commandr# r# r# r$ �<module> s� (
� � .
6���
� ������