Current File : //lib/python3/dist-packages/uaclient/files/__pycache__/state_files.cpython-312.pyc
�
�Hcf� � � � d dl Z d dlmZmZmZmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
mZmZ d dlmZmZ d dlmZmZmZ dZ G d� d e
� Zd
eeef d eeef fd �Z ee eed
�� e�� Z G d� de
� Z ee edd�� ej: � Z G d� de
� Z ee edd�� ej: � Z G d� de
� Z! G d� de
� Z" ee" ed d
�� ej: � Z# ed ejH � � Z% ed d
ejH � � Z& G d!� d"e
� Z' ee' ed#� ej: �$� Z( ed%� Z) G d&� d'e
� Z* ee* ed(ejV d�� ej: � Z, G d)� d*e
� Z- ee- ed+d
�� ej: � Z. e ed,d
�� �-� Z/ ed.ejV d�� Z0d/� Z1y)0� N)�Any�Dict�List�Optional)�defaults)�
BoolDataValue�
DataObject�DatetimeDataValue�Field� IntDataValue�StringDataValue� data_list)�DataObjectFile�DataObjectFileFormat)� ProJSONFile�UAFile�
UserCacheFilezservices-once-enabledc �. � e Zd Z eded� gZdefd�Zy)�ServicesOnceEnabledData� fips_updatesFc � � || _ y �N)r )�selfr s �</usr/lib/python3/dist-packages/uaclient/files/state_files.py�__init__z ServicesOnceEnabledData.__init__ s
� � (��� N)�__name__�
__module__� __qualname__r r �fields�boolr � r r r r s! � �
�n�m�U�3��F�)�T� )r r �data�returnc �r � i }| j � D ]! }d|v rd||j dd� < � d||< �# |S )N�-T�_)�keys�replace)r# � updated_data�keys r �%_services_once_enable_preprocess_datar, sL � � �L��y�y�{� %�� �#�:�26�L����S�#�.� /� $�L�� � %� �r F)�name�private)�data_object_cls�ua_file�preprocess_datac �z � e Zd Z ed ee� d� ed ee� d� gZ ddeee deee fd�Z
y)�RetryAutoAttachOptions�enableF� enable_betaNc � � || _ || _ y r )r4 r5 )r r4 r5 s r r z RetryAutoAttachOptions.__init__= s � �
�� �&��r )NN) r r r r r r
r r r �strr r" r r r3 r3 7 s\ � �
�h� �/�2�E�:�
�m�Y�� 7��?��F� '+�+/�'���c��#�'� �d�3�i�(�'r r3 z retry-auto-attach-options.jsonT)r. c �J � e Zd Z ede� eded�� gZdedee fd�Z
y)�RetryAutoAttachState�interval_index�failure_reasonF��requiredc � � || _ || _ y r )r: r; )r r: r; s r r z RetryAutoAttachState.__init__V s � �
-��� ,��r N) r r r r r r
r �intr r7 r r" r r r9 r9 P s< � �
� � �-�
� ��%�@��F�
-��-� !��
�-r r9 z retry-auto-attach-state.jsonc �6 � e Zd Z ede� ede� gZd� Zy)�
TimerJobState�next_run�last_runc � � || _ || _ y r )rB rC )r rB rC s r r zTimerJobState.__init__l s � � ��
� ��
r N)r r r r r
r r r" r r rA rA f s$ � �
�j�+�,�
�j�+�,��F�
!r rA c �r � e Zd Z eded�� eded�� eded�� gZdee dee dee fd�Zy)�AllTimerJobsState�meteringFr< �update_messaging�update_contract_infoc �. � || _ || _ || _ y r )rG rH rI )r rG rH rI s r r zAllTimerJobsState.__init__x s � � !��
� 0�� �$8��!r N)r r r r rA r r r r" r r rF rF q s_ � �
�j�-�%�8�
� �-�%�@�
�$�m�e�D��F� 9��=�)�9� #�=�1�9� '�}� 5� 9r rF zjobs-status.jsonzapt-news)� directoryz apt-news-raw)r. rK c � � e Zd Z ede� ede� ede� ede� eded�� ede� gZdedededede e
de j f d �Z y
) �LivepatchSupportCacheData�version�flavor�arch�codename� supportedFr< � cached_atc �X � || _ || _ || _ || _ || _ || _ y r )rN rO rP rQ rR rS )r rN rO rP rQ rR rS s r r z"LivepatchSupportCacheData.__init__� s. � � �� � �� ��� � ��
�"���"��r N)
r r r r r
r r
r r7 r r! �datetimer r" r r rM rM � s� � �
�i� �)�
�h��(�
�f�o�&�
�j�/�*�
�k�=�5�9�
�k�,�-�
�F�#��#� �#� � #�
� #� �D�>�
#� �$�$�#r rM z#livepatch-kernel-support-cache.json)� file_formatzmarker-reboot-cmds-requiredc �h � e Zd Z ede� ede� ede� ede� gZdedededefd�Zy)�AnboxCloudData�anbox_images_url�anbox_images_resource_token�anbox_cloud_apt_token�anbox_cloud_apt_urlc �< � || _ || _ || _ || _ y r )rY rZ r[ r\ )r rY rZ r[ r\ s r r zAnboxCloudData.__init__� s$ � � !1�� �+F��(�%:��"�#6�� r N)r r r r r
r r7 r r" r r rX rX � s` � �
� �/�2�
�+�_�=�
�%��7�
�#�_�5� �F�
7� �
7� &)�
7� #�
7�
!�
7r rX zanbox-cloud-credentialsc �@ � e Zd Z ede� gZdej fd�Zy)�AttachmentData� attached_atc � � || _ y r )r` )r r` s r r zAttachmentData.__init__� s
� �&��r N)r r r r r
r rU r r" r r r_ r_ � s& � �
�m� .�/��F�'�H�$5�$5� 'r r_ zattachment.jsonz status.json)�pro_filez
machine-idc �� � t j � t j � t j � t j � t
j � t j � y r )�machine_id_file�delete�status_cache_file�attachment_data_file� anbox_cloud_credentials_file�reboot_cmd_marker_filer" r r �delete_state_filesrj � sJ � ���� �� � � �� � �!� �'�'�)��!�!�#�� � � r )2rU �typingr r r r �uaclientr �uaclient.data_typesr r r
r r r
r �uaclient.files.data_typesr r �uaclient.files.filesr r r �SERVICES_ONCE_ENABLEDr r7 r, �services_once_enabled_filer3 �JSON� retry_auto_attach_options_filer9 � retry_auto_attach_state_filerA rF �timer_jobs_state_file� MESSAGES_DIR�apt_news_contents_file�apt_news_raw_filerM �livepatch_support_cacheri rX �DEFAULT_PRIVATE_DATA_DIRrh r_ rg rf rd rj r" r r �<module>r{ s9 �� � ,� ,� �� � � K� C� C�/� �)�j� )�
�
�s�C�x�.�
� �#�s�(�^�
� ,�+� �
"��
� :�
� � '�Z� '� "0��
�(��� � � �
"� � -�:� -� .��
� )�4�8�� � � � �!�J� !�9�
� 9�$ '��
� �u�-�� � �� � �
�h�6K�6K�L� ���E�X�-B�-B�� �
#�
� #�6 )� ��7�8�$�)�)�� � � =�>� �7�Z� 7�* .��
�!��)�)���
� � � � �'�Z� '� &��
� �e�,�� � �� � �
�
���� � �� �%�%� ��� r
Mini Shell