File: //lib/python3/dist-packages/uaclient/api/u/pro/services/enable/__pycache__/v1.cpython-310.pyc
o
��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'dededej(deeej( de)deej(ee f f dd�Z*dedee+ fdd�Z, d.de&dee de'fdd �Z- d.de&dedee de'fd d �Z.ed d 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 @ sZ e Zd Zededd�ededdd�ededdd�gZd dd
�dedee de fd 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 c C s || _ || _|| _d S �N)r r r )�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 s0 �������r c @ sr e Zd Zedee�dd�edee�dd�ededd�edee�d d�gZdee dee de
dee fd
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.c C s || _ || _|| _|| _d S r �r, r- r. r )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 s> ����� ����r+ �cfg�progress� entitlement�available_variantsr �returnc C s� d }|D ]}|| |d�}|� � d tjjkr |�� r |} nq|d u r/|jd ur/|j| |d�}|d urc|�dtjdt j
j t j j
|j t j j d�ifg� t jj |jd�}|t|j|jd|jid�fS |d fS ) N)r0 r r �message_operation�msg)r )� variant_namer7 )�title�code�meta)�applicability_statusr �ApplicabilityStatus�
APPLICABLE�variant_auto_select�default_variant�emitr �prompt_for_confirmationr �AUTO_SELECTING_VARIANT�format�TxtColor�BOLDr7 �ENDC� AUTO_SELECTED_VARIANT_WARNINGr r6 �name) r0 r1 r2 r3 r r �v_cls�v�warningr" r" r# �_auto_select_variantb sJ
������������rL c C s dd� t | �jD �S )Nc S s g | ]}|j �qS r" )rH ��.0�sr" r" r# �
<listcomp>� s z+_enabled_services_names.<locals>.<listcomp>)r �enabled_services�r0 r" r" r# �_enabled_services_names� s rS �options�progress_objectc C s t | t� |d�S )N)rU )�_enabler )rT rU r" r" r# �enable� s rW c
s t |�}g }t�� st�� �t|�jst�� �� jdkr t� � �t
|�}t � fdd�t |�j
D �d�}|r=tg g dg d�S tj|� j� jpFd� jd�}|�� \}} |j}
t|tjjk|j |
g�rvt||||
�� � jd �\}} | rv|�| � |� � |_ d} d}
z t j d
d �� |� |�\} }
W d � n1 s�w Y W n t!y� } zt �"� |�d}~ww | s�|
dur�|
j#dur�|
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.
� landscapec 3 s8 � | ]}|j � jkr� jr|jr|j� jkr|V qd S r )rH r r �variant_enabledr7 rM �rT r" r# � <genexpr>� s � �� ��z_enable.<locals>.<genexpr>NFr/ � )r0 rH r r )r0 r1 r2 r3 r zu.pro.services.enable.v1)� lock_holder)r �reasonc S s g | ] }t |t�r|�qS r" )�
isinstancer) )rN r6 r" r" r# rP � s ��z_enable.<locals>.<listcomp>� post_enablerR )1r
r �we_are_currently_rootr �NonRootUserErrorr � is_attached�UnattachedErrorr � NotSupportedrS �nextr rQ r+ r �entitlement_factoryr r r; �variants�allr< r= �
is_variantrL �values�append� calculate_total_enable_steps� total_stepsr � RetryLockrW � Exception�clear_lock_file_if_present�messager �GENERIC_UNKNOWN_ISSUE�EntitlementNotEnabledError� messaging�getr
�finish�sorted�list�set�
difference�_check_for_reboot�warnings)rT r0 rU r1 r} �enabled_services_before�already_enabledr2 �
applicability�_r3 �auto_select_warning�success� fail_reason�er^ �enabled_services_after�post_enable_messages�resultr" rZ r# rV � s�
� � ��
��
�
� ������ �������rV �v1�
EnableServiceT)�versionrH �fn� options_cls�supports_progress�32zw
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
}
)�
introduced_in�requires_network�example_python� result_classr � example_cli� example_jsonr )7�logging�typingr r r r r �uaclientr r r r
r � uaclient.apir r
r �uaclient.api.apir �uaclient.api.data_typesr r �-uaclient.api.u.pro.status.enabled_services.v1r �(uaclient.api.u.pro.status.is_attached.v1r �uaclient.configr �uaclient.data_typesr r r r r � getLogger� replace_top_level_logger_namer% �LOGr r+ �
UAEntitlementr* rL r) rS rW rV �endpointrb rd re �EntitlementNotFoundError�
LockHeldErrorrt �_docr" r" r" r# �<module> s� ,��� ��
�/���
� ����
�~� ������
�