�
�Hcf�
� �� � d dl Z d dlmZmZmZmZ d dlmZmZm Z m
Z
m Z m Z d dl
mZ d dlmZ e j" e j$ e� � Z ej* � Z G d� de� Zy)� N)�Any�Dict�Optional�Tuple)�api� event_logger�
exceptions�messages�system�util)�
UAEntitlement)�ApplicationStatusc � � � e Zd ZdZej
Zej Zej j Z ej Z
defd�Zdefd�Zdej$ defd�Zdej$ defd�Zdeeeej2 f fd�Zdeeeej2 f fd �Z dd
e e e f d e e e f d edef� fd
�
Z � xZ S )�LandscapeEntitlement� landscape�returnc � � y�N� � ��selfs �A/usr/lib/python3/dist-packages/uaclient/entitlements/landscape.py� enable_stepsz!LandscapeEntitlement.enable_steps � � �� c � � yr r r s r �
disable_stepsz"LandscapeEntitlement.disable_steps r r �progressc �� � dg| j z }| j r
d|vr|dgz
}t j d|� |j t j t j j dj |� �� � � t j || j �� y
# t j $ r�}t j |� | j rq|j! d|j" j% � � t j& |j( j% � |j" j% � �� �Y d }~y d }~ww xY w) N�landscape-config�--silentz
Executing: %r� ��command)�pipe_stdouterr�info)�stdout�stderrFT)�
extra_args�
assume_yes�LOG�debugr r �redact_sensitive_logsr
�EXECUTING_COMMAND�format�joinr �subpr �ProcessExecutionError� exception�emitr) �strip�LandscapeConfigFailedr( �r r �cmd�es r �_perform_enablez$LandscapeEntitlement._perform_enable s � �!�"�T�_�_�4�� �?�?�z�� 4� �J�<� �C� � � �/�3�'���� � &� &��*�*�1�1�#�(�(�3�-�1�H�
�
�
� �K�K��D�O�O� <� �� �/�/� � �M�M�!� �����
�
�f�a�h�h�n�n�&6�7� �6�6� �8�8�>�>�+�A�H�H�N�N�4D�� � �� �s �!B1 �1E �BE�E c � � ddg}|j t j j dj |� �� � t j |� |j dt j � y# t j $ rH}t j |� |j dt |� j � � Y d }~�wd }~ww xY w)Nr! z --disabler# r$ r' T)r r
r/ r0 r1 r r2 r r3 r, �errorr5 �strr6 �LANDSCAPE_CONFIG_REMAINSr8 s r �_perform_disablez%LandscapeEntitlement._perform_disable0 s� � �!�;�/����� � &� &� -� -�c�h�h�s�m� -� D�
� 2� �K�K�� �
�
�
�f�h� ?� ?�@���
�/�/� 2� �I�I�a�L� �M�M�&�#�a�&�,�,�.� 1� 1�� 2�s �A: �:C�
>C�Cc � � | j � r't j d� rt j d fS t j
t j fS )Nzlandscape-client)� are_required_packages_installedr �is_systemd_unit_activer �ENABLED�DISABLEDr
� LANDSCAPE_SERVICE_NOT_ACTIVEr s r �application_statusz'LandscapeEntitlement.application_status? sN � �
� 0� 0� 2��-�-�.@�A�%�-�-�t�4� 4� "�*�*��5�5��
r c � � t j � r t j g d�� yy# t j
$ r dt j fcY S w xY w)N)r! z--is-registeredr" T)FN)r �we_are_currently_rootr r2 r r3 r
�LANDSCAPE_NOT_REGISTEREDr s r �enabled_warning_statusz+LandscapeEntitlement.enabled_warning_statusM sV � �
� %� %� '�
�� � �G�� ��
�3�3�
�� �5�5�� �
�s �/ �%A�A� orig_access�deltas� allow_enablec �( �� t �| � ||d�� S )NF)rN )�super�process_contract_deltas)r rL rM rN � __class__s �r rQ z,LandscapeEntitlement.process_contract_deltas_ s# �� � �w�.� ��e� /�
�
r )F)!�__name__�
__module__� __qualname__�namer
�LANDSCAPE_TITLE�title�LANDSCAPE_DESCRIPTION� description�urls�LANDSCAPE_HOME_PAGE� help_doc_url�LANDSCAPE_HELP_TEXT� help_text�intr r r �ProgressWrapper�boolr; r@ r r r � NamedMessagerG rK r r> r rQ �
__classcell__)rR s @r r r s �� � �D� � $� $�E��0�0�K��=�=�4�4�L��,�,�I��c� ��s� ���(;�(;� �� �.
��)<�)<�
��
� � � �(�8�+@�+@�"A�A� B� � � �t�X�h� 3� 3�4�4� 5��, #�
��#�s�(�^�
� �S�#�X��
� �
�
�
�
r r )�logging�typingr r r r �uaclientr r r r
r r �uaclient.entitlements.baser
�(uaclient.entitlements.entitlement_statusr � getLogger� replace_top_level_logger_namerS r, �get_event_logger�eventr r r r �<module>rn sX �� � -� -� J� J� 4� F��g���:��:�:�8�D�E��%� �%�%�'��]
�=� ]
r |