OwlCyberSecurity - MANAGER
Edit File: enable.cpython-38.pyc
U ���hAK � @ 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& d d l'm(Z( e�)e�*e+��Z,ed d e-fde-fdeee.ef fg�Z/e j0e.e-d�dd�Z1d:e j0e-ee j2 d�dd�Z3e j0e.ee ee e.e.e.d�dd�Z4e-ee.ef ee. ee. eee.ef eee.ef e-d�dd�Z5e j0e.e.e-e-e-eee. ee ee e/d � d d �Z6e j7e j8e �9e j:�e;d �d d!����Z<e$d"ej=ej>e<e&j?d#e#e"d$ej@jAd%�Be jCe �0� d&��d'�d(d)d*�e"d+ejDd,d-�e"d.ejEjAd"d/�d,d-�e"d0ejFd,d-�e"d1ejGd,d-�e"d2ejHjAd3d4�d(d3d5gd3d6�e"d7ejId(d-�gd8�gd9�ZJdS );� 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)� machine_token�_EnableOneServiceResult�success� needs_reboot�error)�cfg�variant� assume_yesc C st t �|�}t�| �}t�| |�� �}|s6|tj� d S t | �j }t | �j }|D ] } t | | j|d||d ||d� qNd S )NF) r# �ent_namer$ � access_onlyr% � json_output� extra_args�enabled_services�all_dependencies)r �&create_interactive_only_print_functionr Zget_machine_token_filer Z get_enabled_by_default_servicesr r ZNO_SERVICES_TO_AUTO_ENABLEr r* r �services�_enable_one_service�name) r# r$ r% r( �interactive_only_printZmachine_token_fileZservices_to_be_enabledr* r+ Zenable_by_default_service� r1 �5/usr/lib/python3/dist-packages/uaclient/cli/enable.py�_auto_enable_services0 s2 � � �r3 )r# r' �progress_objectc C s� t �|�}tj| d||d�}d}d}z,tjdd�� |j|d�\}}W 5 Q R X W n. tk r~ } zt�� |�W 5 d}~X Y nX |s�|dk r�|j dk r�|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' r) FNz cli.enable._enable_landscape)Z lock_holder)�progress)�service�reason)ZenabledZdisabled�reboot_requiredr )r ZProgressWrapperr ZLandscapeEntitlementr Z RetryLock�enable� ExceptionZclear_lock_file_if_present�messager ZGENERIC_UNKNOWN_ISSUEr �EntitlementNotEnabledErrorr ) r# r' r) r4 r7 r5 r Z fail_reason�er9 r1 r1 r2 �_enable_landscapeS s> �� � �r@ )r# r8 r+ r* r6 r$ � service_titlec s� g }g }dd� |D �� t �fdd�|D �d �} | d k r^� fdd�| jD �}� fdd�| jD �}|D ]N} tj| jdd�} t�| | �} tj j || d �} | s�tj | d �sbt j || d ��qb|D ]:}t�| |�}tjj ||d �} tj | d �s�t j||d ��q�t ��fd d�|D �d �}|d k �r��d k �r�t�| ���}t�| �|j�}tj| jdd�} tj j ||d �} | �srtj | d ��s�t j ||d ��d S ) Nc S s g | ] }|j �qS r1 �r/ ��.0�sr1 r1 r2 � <listcomp>� s z2prompt_for_dependency_handling.<locals>.<listcomp>c 3 s | ]}|j � kr|V qd S �NrB rC )r8 r1 r2 � <genexpr>� s z1prompt_for_dependency_handling.<locals>.<genexpr>c s g | ]}|j � kr|j �qS r1 rB rC ��enabled_service_namesr1 r2 rF � s �c s g | ]}|j � kr|j �qS r1 rB rC rI r1 r2 rF � s �z features.block_disable_on_enable)r Z path_to_value)�service_being_enabled�incompatible_service)�msg)rK �required_servicec 3 s, | ]$}|j � kr|jr|j�kr|V qd S rG �r/ �variant_enabled� variant_namerC )r8 r$ r1 r2 rH � s �)�nextZincompatible_withZ depends_onr Zis_config_value_truer# r Z get_titler ZINCOMPATIBLE_SERVICE�formatZprompt_for_confirmationr Z IncompatibleServiceStopsEnableZREQUIRED_SERVICEZRequiredServiceStopsEnablerQ )r# r8 r+ r* r6 r$ rA Zincompatible_servicesZrequired_servicesZ dependenciesrL Zcfg_block_disable_on_enableZincompatible_service_titleZuser_msgrN Zrequired_service_titlerP Zto_be_enabled_titleZenabled_variant_titler1 )rJ r8 r$ r2 � prompt_for_dependency_handling s� � � �� ���� � � �� ��� ��rT �r( � json_response�processed_services�failed_services�errors�warningsr c C s^ | rZ|� � |� � |r dnd|d< ||d< ||d< ||d< ||d< ttj|tjdd �� d S ) Nr Zfailure�resultrW rX rY rZ T)�clsZ sort_keys)�sort�print�json�dumpsr ZDatetimeAwareJSONEncoderrU r1 r1 r2 �_print_json_output� s ��ra ) r# r& r$ r' r% r( r) r* r+ �returnc s� t �|�} tj| |�||d�} | j� | j} t� �fdd�|D �d �} | d k r�tjj | d�} | | j � | tj j | d�� t ddd|| j | jd�d�S |�szt | � |||�| d � W n^ tjk �r } z<t�|� | |j � | tj j | d�� t ddd d� W Y �S d }~X Y nX z�|�r d }n t j|d �}� d k�rHt| |||d �}ntt|�|d �| |d�}tj| d� |�r�| tjj | d�� n| tjj | d�� |j}|�r�| tjj dd�� |jD ]}| |� �q�t d|d d�W S tjk �rZ } z^|j d }|d }|d }|d }| |� | tj j | d�� t ddd||||d�d� W Y �S d }~X Y nf tjk �r� } zD| |j � | tj j | d�� t ddd||j |j |j d�d� W Y �S d }~X Y nX d S )N)r$ r' r) c 3 s0 | ](}|j � kr�r$|jr|j�kr|V qd S rG rO rC �Z real_namer$ r1 r2 rH s �z&_enable_one_service.<locals>.<genexpr>)�titleFr8 )�typer8 r= � message_code)r r! r" )r6 r$ rA )r% r5 )r) r4 )r8 r$ r' )r4 �r# Zinstall)Z operationTr9 �coderd �additional_info�re r8 r= rf ri ) r r, r Zentitlement_factoryr/ rd rR r ZALREADY_ENABLEDrS rM Z ENABLE_FAILEDr rT r �UbuntuProError�LOGZ exceptionZCLIEnableDisableProgressr@ r r r ZACCESS_ENABLED_TMPLZ ENABLED_TMPLr: ZENABLE_REBOOT_REQUIRED_TMPLr> ri �msg_code)r# r& r$ r' r% r( r) r* r+ r0 ZentZ ent_titleZalready_enabledrM r? r7 Z enable_resultr! r= r9 Zerr_code�err_msgZerr_infor1 rc r2 r. � s� �� � � �� � �� ��� � ��� � �� ��r. )rb c K s� g }g }g }g }t jdd�}| jdk}t�|�} t| dd�} | j} | j} | j} | rd| rdt j ddd��| rz| rzt j dd d��| r�| r�t j dd d��| t j � zt j|dd � W nD t jt jfk r� tjd d d � |�dt jjt jjd�� Y nX t|�j�s@t �� }| |j� |�d|j|jd�� t||||||dd� dS | �rVt |dd |d�S t| dg �}|�spt � � �t � ||�\}}t |�j!}t"|�j#}d }t �$||�D ]n}t%||| | | ||�&d�||� }|j'�r�|�|� |j(�rd |d< n&d}|�|� |j)dk �r�|�|j)� �q�|�rZd}||7 }t j*||d�}| |j� |�dd|j|j|j+d�� t �,|�}|�-� t|||||||d� |�r�dS dS )z^Perform the enable action on a named entitlement. @return: 0 on success, 1 otherwise F)Z_schema_versionr! r_ r$ � � --access-only� --variant)Zoption1Zoption2�--auto)�verbosezFailed to refresh contractT)�exc_info�system)re r= rf )r � )r# r$ r% r( r8 r) r! Nrg rj r ).r ZJSON_SCHEMA_VERSIONrS r r, �getattrr' r% �autor ZInvalidOptionCombinationr ZREFRESH_CONTRACT_ENABLEr ZrefreshZConnectivityErrorrk rl Zwarning�appendZE_REFRESH_CONTRACT_FAILURErM r/ r Z is_attached_and_contract_validZContractExpiredErrorrm ra r3 ZNoServicesToEnabler Zget_valid_entitlement_namesr r* r r- Z order_entitlements_for_enablingr. �getr r! r" Z*create_enable_entitlements_not_found_errorri ZUAContractClientZupdate_activity_token)�argsr# �kwargsrW rX rY rZ rV r( r0 r$ r' r% rx Z expired_err�namesZentitlements_foundZentitlements_not_foundr* r+ Zretr& r[ �errZcontract_clientr1 r1 r2 � action_enable� s � � � � � �� ��� � � �� � �� � r r; � r8 z, rg )ZoptionsZstore�*)�help�action�nargsrp � store_true)r� r� z --assume-yes)Zcommandrr z--betaz--formatZcli)�defaultr_ )r� r� �choicesr� rq )Z arguments)r� Z descriptionr� Z help_categoryZ help_positionZargument_groups)N)Kr_ Zlogging�typingr r r r r Zuaclientr r r r r r r r r r Z+uaclient.api.u.pro.services.dependencies.v1r r Z%uaclient.api.u.pro.services.enable.v1r r r Z-uaclient.api.u.pro.status.enabled_services.v1r r Z(uaclient.api.u.pro.status.is_attached.v1r Z uaclient.clir Zuaclient.cli.commandsr r r Zuaclient.cli.parserr Zuaclient.filesr Z getLoggerZ replace_top_level_logger_name�__name__rl �bool�strr ZUAConfigr3 ZAbstractProgressr@ rT ra r. Zverify_json_format_argsZ assert_rootZassert_attachedZ&_raise_enable_disable_unattached_error�intr ZCLI_ROOT_ENABLEZCLI_ENABLE_DESCZ QUICKSTARTZCLI_ENABLE_SERVICErS �joinZvalid_servicesZCLI_ENABLE_ACCESS_ONLYZCLI_ASSUME_YESZCLI_ENABLE_AUTOZCLI_ENABLE_BETAZCLI_FORMAT_DESCZCLI_ENABLE_VARIANTZenable_commandr1 r1 r1 r2 �<module> s� 0 �� � ' �� -� ] � � # ��� � ��� ������