�
Uٵg� � � � S r SSKrSSKrSSKJr SSKJrJrJrJ r J
r
SSK J r
SSKJr SSKJrJr SSKJr SS KJrJr S
rS r\R4 " \
R6 " \ 5 5 r S \4S
jr S \S\ 4S jr S r!g)a%
Update messaging text for use in MOTD and APT custom Ubuntu Pro messages.
Messaging files will be emitted to /var/lib/ubuntu-advantage/message-* which
will be sourced by apt-hook/hook.cc and various /etc/update-motd.d/ hooks to
present updated text about Ubuntu Pro service and token state.
� N)�exists)�contract�defaults�messages�system�util)�_updates)�_enabled_services)�ContractExpiryStatus� _is_attached)�UAConfig)�
machine_token�noticeszmotd-contract-statusz6/usr/lib/update-notifier/update-motd-updates-available�cfgc �� � [ R " U 5 nUR nUR S0 5 R S0 5 R SS 5 n[ R " U 5 nUR UR SS5 U5 nUR S0 5 R S0 5 R SS 5 nUb- XaR :w a UUS S S' UR U5 g g g )N�machineTokenInfo� contractInfo�id� machineToken� � effectiveTo)r �get_machine_token_file�getr �UAContractClient�get_contract_machine�contract_expiry_datetime�write)r �machine_token_file�
orig_token� contract_id�contract_client�resp� resp_expirys �A/usr/lib/python3/dist-packages/uaclient/timer/update_messaging.py�update_contract_expiryr% s� � �&�=�=�c�B��#�1�1�J����)�2�.� ��^�R� � ��T�4� � �
�/�/��4�O� � /� /����~�r�*�K� �D�
���#�R�(� ��^�R� � ��]�D� !� � � � �F�F� F�
� �%�&�~�6� �
� � � ��,�
G� � �returnc �� � [ U 5 nUR ( d g[ R S5 [ R
R
U R S[ 5 nUR nUR n[ R " U 5 nU[ R R [ R R [ R" R 4; a. [% U 5 [ U 5 nUR nUR nU[ R& R [ R( R 4; aE [* R, " [* R. R0 5 [2 R4 " U5 gU[ R R :X as [* R, " [* R. R0 5 [2 R6 " U[8 R: R= U5 R? US9S- 5 gU[ R R :X a� [* R, " [* R. R0 5 [@ RB U- nURD nUc SnOURG S5 n[2 R6 " U[8 RH R= U5 R? UUS9S- 5 gU[ R" R :X GaW [* RJ " [* R. R0 5 S n S
n
[M U 5 RN V s1 s H o�RP iM n n [2 RR " 5 ( a( S U ; a! S n [U U 5 RV RX n
OA[2 RZ " 5 ( a' S U ; a! S n [U U 5 RV R\ n
U
S
:X a) [2 R6 " U[8 R0 S- 5 g[2 R6 " U[8 R^ R= U
5 R? U
U S
9S- 5 gs sn f )z�Emit human-readable status message used by motd.
Used by /etc/update.motd.d/91-contract-ua-esm-status
:param cfg: UAConfig instance for this environment.
Fz&Updating Ubuntu Pro messages for MOTD.r )�remaining_daysz
�Unknownz%d %b %Y)� expired_dater) zn/ar z esm-infrazesm-apps)�pkg_num�serviceT)0r � is_attached�LOG�info�os�path�join�data_dir� MOTD_CONTRACT_STATUS_FILE_NAME�contract_status�contract_remaining_daysr r r �ACTIVE_EXPIRED_SOON�value�EXPIRED_GRACE_PERIOD�EXPIREDr% �ACTIVE�NONEr �remove�Notice�CONTRACT_EXPIREDr �ensure_file_absent�
write_filer �CONTRACT_EXPIRES_SOON� pluralize�formatr �!CONTRACT_EXPIRY_GRACE_PERIOD_DAYSr �strftime� CONTRACT_EXPIRED_GRACE_PERIOD�addr
�enabled_services�name� is_current_series_active_esm� api_u_pro_packages_updates_v1�summary�num_esm_infra_updates�is_current_series_lts�num_esm_apps_updates�CONTRACT_EXPIRED_WITH_PKGS)
r �is_attached_info� motd_contract_status_msg_path�
expiry_statusr) r �grace_period_remaining�exp_dt�
exp_dt_strr- r, �s�enabled_services_namess
r$ �update_motd_messagesr[ ; s� � � $�C�(�� � '� '���H�H�
5�6�$&�G�G�L�L� � � �j�"@�%�!� %�4�4�M�%�=�=�N�&�=�=�c�B��� �0�0�6�6� �1�1�7�7� �$�$�*�*�� �
�s�#�'��,��(�8�8�
�)�A�A��� �#�#�)�)� �!�!�'�'�� � ���w�~�~�6�6�7��!�!�"?�@�F �E
�.�B�B�H�H� H����w�~�~�6�6�7���� )� � *� *� 4� 4�^� D� K� K� -�
L�
� �
�
�@ �s
�.�C�C�I�I� I����w�~�~�6�6�7� � 6� 6� � G� � $�<�<�� �>�"�J� � � ��4�J���� )� � 2� 2� <� <� �
��f� '� 5� � � �
�
�^ �K
�.�6�6�<�<� <�� � �G�N�N�3�3�4����� .�c� 2� C� C�"
� C�q�F�F� C� � "
� � .� .� 0� 0�� 4�4�%��7����'�/�/� �� �
)�
)�
+�
+�� 3�3�$��7����'�.�.� � �a�<� � � �-��)�)�F�2�
� �
� � �-��3�3�=�=�g�F�M�M� #� #� N� � � �
� ��?"
s �
Q c � � [ [ 5 ( a [ R " [ S/5 g g ! [ a n [
R
U 5 S n A g S n A ff = f)Nz--force)r �UPDATE_NOTIFIER_MOTD_SCRIPTr �subp� Exceptionr/ � exception)�excs r$ � refresh_motdrb � sM � � �)�*�*� � �K�K�4�i�@� A� +�� � � �M�M�#� � �� �s � 4 �
A �A�A )"�__doc__�loggingr1 �os.pathr �uaclientr r r r r �&uaclient.api.u.pro.packages.updates.v1r rM �-uaclient.api.u.pro.status.enabled_services.v1r
�(uaclient.api.u.pro.status.is_attached.v1r r �uaclient.configr
�uaclient.filesr r r5 r] � getLogger� replace_top_level_logger_name�__name__r/ r% �boolr[ rb � r& r$ �<module>rq s� ��� � � � ?� ?�� L�� %� 1�!7� �<� � ����:�:�8�D�E��-�� -�6f�h� f�4� f�R
r&