�
�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 �f � e Zd Z ede� eded� eded� gZddd�dedee de fd�Z
y)�
EnableOptions�service�variantF� access_onlyN)r r c �. � || _ || _ || _ y �N)r r r )�selfr r r s �G/usr/lib/python3/dist-packages/uaclient/api/u/pro/services/enable/v1.py�__init__zEnableOptions.__init__ s � � �� � �� �&��� ) �__name__�
__module__� __qualname__r r r �fields�strr �boolr � r! r r r sZ � �
�i� �)�
�i� �%�0�
�m�]�E�2��F� "&� !� '� � '� �#� � '�
� 'r! r c � � e Zd Z ed ee� � ed ee� � ede� ed ee� � gZdee dee de
dee fd�Z y)� EnableResult�enabled�disabled�reboot_requiredr c �< � || _ || _ || _ || _ y r �r+ r, r- r )r r+ r, r- r s r r zEnableResult.__init__1 s! � � �� � ��
� .�� � ��
r! N) r" r# r$ r r r r r% r r& r' r r( r! r r* r* ) sz � �
�i��?�3�4�
�j�)�O�4�5�
� � �/�
�j�)�O�4�5� �F� !� �c�� !� �s�)� !�
� !� �s�)�
!r! r* �cfg�returnc �f � t | � j D �cg c] }|j �� c}S c c}w r )r �enabled_services�name)r0 �ss r �_enabled_services_namesr6 ? s% � � -�c� 2� C� C� D�q�A�F�F� D�D�� Ds �.�options�progress_objectc �. � t | t � |�� S )N)r8 )�_enabler )r7 r8 s r �enabler; C s � � �7�H�J�� H�Hr! c �| � t |� }t j � st j � �t |� j st j � �| j dk( rt j � �t |� }| j |v rt g g dg �� S t j || j | j xs d�� } ||dd| j | j �� }|j! � |_ d}d } t% j& d� � 5 |j) |� \ }}d d d � |sL|�|j. �
|j. }
nt0 j2 }
t j4 | j |
�
� �t |� } |j6 j9 d g � xs g D � cg c] } t; | t< � r| �� }
} t? j> |� � |jA � t tC tE tG | � jI tG |� � � � tC tE tG |� jI tG | � � � � |jK � |
�� S # 1 sw Y ��IxY w# t* $ r} t% j, � | �d } ~ ww xY wc c} w )
N� landscapeFr/ � )r0 r4 r T)�
assume_yes�
allow_beta� called_namer zu.pro.services.enable.v1)� lock_holder)r �reason� post_enable)r0 )&r r �we_are_currently_rootr �NonRootUserErrorr � is_attached�UnattachedErrorr � NotSupportedr6 r* r �entitlement_factoryr r � calculate_total_enable_steps� total_stepsr � RetryLockr; � Exception�clear_lock_file_if_present�messager �GENERIC_UNKNOWN_ISSUE�EntitlementNotEnabledError� messaging�get�
isinstancer& r �finish�sorted�list�set�
difference�_check_for_reboot)r7 r0 r8 �progress�enabled_services_before�ent_cls� entitlement�success� fail_reason�erC �enabled_services_after�msg�post_enable_messagess r r: r: I s� � �
��/�H� � %� %� '��)�)�+�+� �� � (� (��(�(�*�*����+�%��%�%�'�'� 5�c� :�����1�1���� !��
�
� �.�.� �g�o�o�w���/D�"��G� � ����O�O��'�'� �K� '�C�C�E�H���G��K��
�^�^�2�
� @� $/�#5�#5�h�#?� �G�[� @� � � "�{�':�':�'F� �(�(�F� �3�3�F��3�3��O�O�F�
�
� 5�S� 9�� �)�)�-�-�m�R�@�F�B� � � �c�3� � � �� � �M�M�c�� �O�O�� �� ��*�+�6�6��/�0��
�
� � ��+�,�7�7��.�/��
�
� $�5�5�7�%�! � �9 @� @�� � � �'�'�)������ s6 �J �)J�>J �J9�J�
J � J6�J1�1J6�v1�
EnableServiceT)�versionr4 �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& r6 r; r: �endpointr( r! r �<module>rz s �� � !� ?� ?� F� F� (� 2� K� A� $�� � �g���:��:�:�8�D�E��'�J� '�&!�:�~� !�,E�� E�d�3�i� E�
KO�I�
�I�-5�6F�-G�I��I� 37�X�
�X� �X� � .�/�X� � X�v � � �� �� �r! |