← Back
Editing: v1.cpython-38.pyc
U Uٵg& � @ 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� - � / � � �� ~� ������ �
Save File
Cancel