�
�Hcf+ � � � 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 j< e j> e � � Z! G d � d e� Z" G d
� dee� Z#dedee$ fd�Z% dde"dee de#fd�Z& dde"dedee de#fd�Z' edde'e"d�� Z(y)� N)�List�Optional)� entitlements�lock�messages�status�util)�AbstractProgress�ProgressWrapper�
exceptions)� APIEndpoint)�AdditionalInfo)�_enabled_services)� _is_attached)�UAConfig)�
BoolDataValue�
DataObject�Field�StringDataValue� data_listc �H � e Zd Z ede� eded� gZdd�dedefd�Z y)�DisableOptions�service�purgeF)r c � � || _ || _ y �N)r r )�selfr r s �H/usr/lib/python3/dist-packages/uaclient/api/u/pro/services/disable/v1.py�__init__zDisableOptions.__init__ s � � �� ���
� N)
�__name__�
__module__� __qualname__r r r �fields�str�boolr � r r r r s7 � �
�i� �)�
�g�}�e�,��F�
7<� �3� �t� r r c �> � e Zd Z ed ee� � gZdee fd�Z y)�
DisableResult�disabledc � � || _ y r �r* )r r* s r r zDisableResult.__init__& s � � ��
r N)
r! r" r# r r r r$ r r% r r' r r r) r) ! s( � �
�j�)�O�4�5��F�!�D��I� !r r) �cfg�returnc �f � t | � j D �cg c] }|j �� c}S c c}w r )r �enabled_services�name)r- �ss r �_enabled_services_namesr3 * s% � � -�c� 2� C� C� D�q�A�F�F� D�D�� Ds �.�options�progress_objectc �. � t | t � |�� S )N)r5 )�_disabler )r4 r5 s r �disabler8 . s � � �G�X�Z� � I�Ir c �" � t |� }t j � st j � �t |� j st j � �t |� }t j || j �� }| j |vr t g �� S ||d| j | j �� }|j }|�|}|j � |_ d}d } t# j$ d�� 5 |j' |� \ }} d d d � |sL| �| j, �
| j, } nt. j0 } t j2 | j | �� �t |� } t5 j4 |� � |j7 � t t9 t; t= |� j? t= | � � � � �� S # 1 sw Y ��xY w# t( $ r}
t# j* � |
�d }
~
ww xY w)
N)r- r1 r, T)�
assume_yes� called_namer Fzu.pro.services.disable.v1)� lock_holder)r �reason)r- ) r r �we_are_currently_rootr �NonRootUserErrorr � is_attached�UnattachedErrorr3 r �entitlement_factoryr r) r �enabled_variant� calculate_total_disable_steps� total_stepsr � RetryLockr8 � Exception�clear_lock_file_if_present�messager �GENERIC_UNKNOWN_ISSUE�EntitlementNotDisabledErrorr �finish�sorted�list�set�
difference)
r4 r- r5 �progress�enabled_services_before�ent_cls� entitlement�variant�success� fail_reason�er= �enabled_services_afters
r r7 r7 4 s� � �
��/�H� � %� %� '��)�)�+�+� �� � (� (��(�(�*�*� 5�c� :���.�.�3�W�_�_�M�G� ��� 5�5� ��
�
� � ���O�O��m�m� �K� �)�)�G��� � �&�D�D�F�H���G��K��
�^�^�3�
� A� $/�#6�#6�x�#@� �G�[� A� � � "�{�':�':�'F� �(�(�F� �3�3�F��4�4��O�O�F�
�
� 5�S� 9��
�M�M�c�� �O�O�� �� ��+�,�7�7��.�/��
�
� � �- A� A�� � � �'�'�)�����s0 �-G* �G �G* � G'�#G* �* H�3H � H�v1�DisableServiceT)�versionr1 �fn� options_cls�supports_progressr ))�logging�typingr r �uaclientr r r r r � uaclient.apir
r r �uaclient.api.apir
�uaclient.api.data_typesr �-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) r% r3 r8 r7 �endpointr' r r �<module>rn s �� � !� ?� ?� F� F� (� 2� K� A� $�� � �g���:��:�:�8�D�E���Z� �!�J�� !�E�� E�d�3�i� E�
LP�J�
�J�.6�7G�.H�J��J� 37�G�
�G� �G� � .�/�G� � G�T � � �� �� �r |