File: //lib/python3/dist-packages/uaclient/api/u/pro/services/disable/__pycache__/v1.cpython-38.pyc
U
��Jh� �
@ sj d dl Z d dlmZmZ d dlmZmZmZmZm Z d dl
m Z m Z m
Z
d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZmZmZm Z m Z e � e � e ��Z!G d d � d e�Z"G d
d� dee�Z#eee$ d�dd�Z%d&e"ee e#d�dd�Z&d'e"eee e#d�dd�Z'edde'e"dd�Z(d dd e#e
j)d fe
j*d fe
j+d fe
j,d!fe
j-d"fgd#d$d%�Z.dS )(� N)�List�Optional)� entitlements�lock�messages�status�util)�AbstractProgress�ProgressWrapper�
exceptions)� APIEndpoint)�AdditionalInfo)�_enabled_services)� _is_attached)�UAConfig)�
BoolDataValue�
DataObject�Field�StringDataValue� data_listc @ s@ e Zd Zededd�ededdd�gZdd�eed�d d
�Z d S ) �DisableOptions�servicezPro service to disable��doc�purgeFzqAlso remove all packages that were installed from this service. Only supported by some services. (default: false))r �r r c C s || _ || _d S �Nr )�selfr r � r �H/usr/lib/python3/dist-packages/uaclient/api/u/pro/services/disable/v1.py�__init__$ s zDisableOptions.__init__N)
�__name__�
__module__� __qualname__r r r �fields�str�boolr r r r r r s ��
r c @ s2 e Zd Zedee�dd�gZee d�dd�Z dS )�
DisableResult�disabledzList of services disabledr �r( c C s
|| _ d S r r) )r r( r r r r 2 s zDisableResult.__init__N)
r! r" r# r r r r$ r r% r r r r r r' ) s ��r' )�cfg�returnc C s dd� t | �jD �S )Nc S s g | ]
}|j �qS r )�name)�.0�sr r r �
<listcomp>7 s z+_enabled_services_names.<locals>.<listcomp>)r Zenabled_services�r* r r r �_enabled_services_names6 s r1 )�options�progress_objectr+ c C s t | t� |d�S )N)r3 )�_disabler )r2 r3 r r r �disable: s r5 )r2 r* r3 r+ c
C sJ t |�}t�� st�� �t|�js*t�� �tj || j
| j d�}t |�}| j
|krZt
g d�S |j}|dk rl|}|�� |_d}d}z*tjdd�� |�|�\}}W 5 Q R X W n. tk
r� } zt�� | �W 5 d} ~ X Y nX |�s|dk r�|jdk r�|j}
ntj}
tj| j
|
d��t |�} tj|d� |�� t
t t t |�� t | ����d�S ) zt
Disable a Pro service. This will automatically disable any services that
depend on the target service.
)r* r, r r) NFzu.pro.services.disable.v1)Z lock_holder)r �reasonr0 ) r
r Zwe_are_currently_rootr �NonRootUserErrorr Z is_attached�UnattachedErrorr Zentitlement_factoryr r r1 r' Zenabled_variantZ calculate_total_disable_stepsZ total_stepsr Z RetryLockr5 � ExceptionZclear_lock_file_if_present�messager ZGENERIC_UNKNOWN_ISSUE�EntitlementNotDisabledErrorr Zfinish�sorted�list�set�
difference) r2 r* r3 ZprogressZ entitlementZenabled_services_beforeZvariantZsuccessZ fail_reason�er6 Zenabled_services_afterr r r r4 @ sb
�
�
� � ����r4 Zv1ZDisableServiceT)�versionr, �fnZ options_clsZsupports_progressZ32z|
from uaclient.api.u.pro.services.disable.v1 import disable, DisableOptions
result = disable(DisableOptions(service="usg"))
z When called as non-root userzCWhen called on a machine that is not attached to a Pro subscriptionz9When the service argument is not a valid Pro service namez8When another Ubuntu Pro related operation is in progressz!When the service fails to disablez4pro api u.pro.services.disable.v1 --args service=usgz+
{
"disabled": [
"usg"
]
}
)Z
introduced_inZrequires_networkZexample_pythonZ result_classr Z example_cliZ example_json)N)N)/Zlogging�typingr r Zuaclientr r r r r Z uaclient.apir r
r Zuaclient.api.apir Zuaclient.api.data_typesr
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' r% r1 r5 r4 Zendpointr7 r8 ZEntitlementNotFoundErrorZ
LockHeldErrorr; Z_docr r r r �<module> sl
� � �� L� ������