File: //lib/python3/dist-packages/uaclient/api/u/pro/services/enable/__pycache__/v1.cpython-38.pyc
U
��Jh& � @ s� d dl Z d dlmZmZmZmZmZ d dlmZm Z m
Z
m Z m Z d dl
mZmZmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d d lmZ d d
l m Z m Z m Z m Z m!Z! e �"e �#e$��Z%G d d � d e �Z&G d
d� de e�Z'eeej(eeej( e)eej(ee f d�dd�Z*eee+ d�dd�Z,d*e&ee e'd�dd�Z-d+e&eee e'd�dd�Z.edd e.e&d d �Z/d d d e'ej0d!fej1d"fej2d#fej3d$fej4d%fej5d&fgd'd(d)�Z6dS ),� N)�Iterable�List�Optional�Tuple�Type)� entitlements�lock�messages�status�util)�AbstractProgress�ProgressWrapper�
exceptions)� APIEndpoint)�AdditionalInfo�ErrorWarningObject)�_enabled_services)� _is_attached)�UAConfig)�
BoolDataValue�
DataObject�Field�StringDataValue� data_listc @ sV e Zd Zededd�ededdd�ededdd�gZd dd
�eee e d �d d
�Z
d S )�
EnableOptions�servicezPro service to be enabled��doc�variantFz2Optional variant of the Pro service to be enabled.� access_onlyz^If true and the target service supports it, only enable access to the service (default: false)N)r r �r r r c C s || _ || _|| _d S �Nr )�selfr r r � r# �G/usr/lib/python3/dist-packages/uaclient/api/u/pro/services/enable/v1.py�__init__* s zEnableOptions.__init__) �__name__�
__module__� __qualname__r r r �fields�strr �boolr% r# r# r# r$ r s* �����r c @ sl e Zd Zedee�dd�edee�dd�ededd�edee�d d�gZee ee e
ee d
�d d �Z d
S )� EnableResult�enabledz#List of services that were enabled.r �disabledz$List of services that were disabled.�reboot_requiredz?True if one of the services that was enabled requires a reboot.r zaList of information message strings about the service that was just enabled. Possibly translated.�r- r. r/ r c C s || _ || _|| _|| _d S r! r0 )r" r- r. r/ r r# r# r$ r% T s zEnableResult.__init__N) r& r' r( r r r r r) r r* r+ r% r# r# r# r$ r, 6 s4 ����� �r, )�cfg�progress� entitlement�available_variantsr �returnc C s� d }|D ]4}|| |d�}|� � d tjjkr|�� r|} q>q|d kr^|jd k r^|j| |d�}|d k r�|�dtjdt j
j t j j
|j t j j d�ifg� t jj |jd�}|t|j|jd|jid�fS |d fS d S ) N)r1 r r Zmessage_operation�msg)r )� variant_namer7 )�title�code�meta)�applicability_statusr �ApplicabilityStatus�
APPLICABLEZvariant_auto_selectZdefault_variant�emitr Zprompt_for_confirmationr ZAUTO_SELECTING_VARIANT�formatZTxtColorZBOLDr7 ZENDCZ AUTO_SELECTED_VARIANT_WARNINGr r6 �name) r1 r2 r3 r4 r r Zv_cls�vZwarningr# r# r$ �_auto_select_variantb sL
��� ���������rB )r1 r5 c C s dd� t | �jD �S )Nc S s g | ]
}|j �qS r# )r@ ��.0�sr# r# r$ �
<listcomp>� s z+_enabled_services_names.<locals>.<listcomp>)r �enabled_services�r1 r# r# r$ �_enabled_services_names� s rI )�options�progress_objectr5 c C s t | t� |d�S )N)rK )�_enabler )rJ rK r# r# r$ �enable� s rM )rJ r1 rK r5 c
s t |�}g }t�� s t�� �t|�js.t�� �� jdkr@t� � �t
|�}t � fdd�t |�j
D �d�}|rztg g dg d�S tj|� j� jp�d� jd�}|�� \}} |j}
t|tjjk|j |
g�r�t||||
�� � jd �\}} | r�|�| � |� � |_ d} d}
z*t j d
d �� |� |�\} }
W 5 Q R X W n0 t!k
�rX } zt �"� |�W 5 d}~X Y nX | �s�|
dk �r~|
j#dk �r~|
j#}nt$j%}tj&� j|d ��t
|�}d
d� |j'�(dg ��p�g D �}t)j)|d� |�*� tt+t,t-|��.t-|����t+t,t-|��.t-|����|�/� |d�}||_0|S )z�
Enable a Pro service. This will automatically disable incompatible services
and enable required services that that target service depends on.
Z landscapec 3 s6 | ].}|j � jkr� jr*|jr|j� jkr|V qd S r! )r@ r r Zvariant_enabledr7 rC �rJ r# r$ � <genexpr>� s �z_enable.<locals>.<genexpr>NFr0 � )r1 r@ r r )r1 r2 r3 r4 r zu.pro.services.enable.v1)Z lock_holder)r �reasonc S s g | ]}t |t�r|�qS r# )�
isinstancer* )rD r6 r# r# r$ rF � s
�z_enable.<locals>.<listcomp>Z post_enablerH )1r
r Zwe_are_currently_rootr �NonRootUserErrorr Z is_attached�UnattachedErrorr � NotSupportedrI �nextr rG r, r Zentitlement_factoryr r r; Zvariants�allr< r= Z
is_variantrB �values�appendZ calculate_total_enable_stepsZ total_stepsr Z RetryLockrM � ExceptionZclear_lock_file_if_present�messager ZGENERIC_UNKNOWN_ISSUE�EntitlementNotEnabledErrorZ messaging�getr
Zfinish�sorted�list�set�
differenceZ_check_for_reboot�warnings)rJ r1 rK r2 rb Zenabled_services_beforeZalready_enabledr3 Z
applicability�_r4 Zauto_select_warningZsuccessZ fail_reason�erQ Zenabled_services_afterZpost_enable_messages�resultr# rN r$ rL � s�
� � ��
���
� �� �������rL Zv1Z
EnableServiceT)�versionr@ �fnZ options_clsZsupports_progressZ32zw
from uaclient.api.u.pro.services.enable.v1 import enable, EnableOptions
result = enable(EnableOptions(service="usg"))
z When called as non-root userzCWhen called on a machine that is not attached to a Pro subscriptionz_When called for a service that doesn't support being enabled via API (currently only Landscape)zxWhen the service argument is not a valid Pro service name or if the variant is not a valid variant of the target servicez8When another Ubuntu Pro related operation is in progressz When the service fails to enablez3pro api u.pro.services.enable.v1 --args service=usgzp
{
"disabled": [],
"enabled": [
"usg"
],
"messages": [],
"reboot_required": false
}
)Z
introduced_inZrequires_networkZexample_pythonZ result_classr Z example_cliZ example_json)N)N)7Zlogging�typingr r r r r Zuaclientr r r r
r Z uaclient.apir r
r Zuaclient.api.apir Zuaclient.api.data_typesr r Z-uaclient.api.u.pro.status.enabled_services.v1r Z(uaclient.api.u.pro.status.is_attached.v1r Zuaclient.configr Zuaclient.data_typesr r r r r Z getLoggerZ replace_top_level_logger_namer& ZLOGr r, Z
UAEntitlementr+ rB r* rI rM rL ZendpointrS rT rU ZEntitlementNotFoundErrorZ
LockHeldErrorr\ Z_docr# r# r# r$ �<module> s� - � / � � �� ~� ������ �