0x1949 Team - FAZEMRX - MANAGER
Edit File: enable.cpython-310.pyc
o �(�fD � @ 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 ����� �,������ �\� ����� � ������ �� � � # ��� ��� ����� �