OwlCyberSecurity - MANAGER
Edit File: disable.cpython-38.pyc
U ��g�* � @ 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 ZservicesZentitlement_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)ra rb z--formatZcli)�defaultr: )ra rb �choicesre r; )Z arguments)ra Z descriptionrb Z help_categoryZargument_groups)<r: ZloggingrV �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__rS ZUAConfig�strr% r7 Zverify_json_format_argsZ assert_rootZassert_attachedZ&_raise_enable_disable_unattached_errorZassert_lock_filer_ ZCLI_ROOT_DISABLEZCLI_DISABLE_DESCZOTHERZCLI_DISABLE_SERVICEr! rT rU ZCLI_ASSUME_YESZCLI_FORMAT_DESCZ CLI_PURGEZdisable_commandr# r# r# r$ �<module> s� ( � � . 6��� � ������