Edit File: backup_info_sender.cpython-311.pyc
� a�nh9 � � � d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl m Z d dl m Z d dlmZmZ d d lmZmZ d d lmZmZ ee� � Z e ed � � � � � � � � Zd Z G d� de � � Z dS )� N)�suppress)� timedelta)� getLogger)�Union)� MessageType)� MessageSource)�get_current_backend�get_last_backup_timestamp)� load_state� save_state)�Scope�recurring_check� )�hours� c �� � e Zd ZdZej Zd� Zd� Ze de e e f de fd�� � Zdde e e f fd �Zd � Z ee� � d � � � Z ed � � d � � � Zd� ZdS )�BackupInfoSenderz.Send user backup statistics to CH periodicallyc � �L K � || _ || _ t j � � | _ | � � � | _ | j � | � � � � � | _ | j � | � � � � � | _ d S �N) �_loop�_sink�asyncio�Event� _send_event�load_last_send_timestamp�_last_send_timestamp� create_task� _recurring_check_data_to_send� _check_task�_recurring_send_stat�_send_stat_task)�self�loop�sinks �_/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/backup_info_sender.py� create_sourcez BackupInfoSender.create_source s� � � � ��� ��� �"�=�?�?���$(�$A�$A�$C�$C��!� �:�1�1� � .� .� 0� 0� � ��� $�z� 5� 5� � %� %� '� '� � �� � � � c � �� K � | j | j fD ]O}|� � � t t j � � 5 |� d {V �� d d d � � n # 1 swxY w Y �P| � � � d S r )r r! �cancelr r �CancelledError�save_last_send_timestamp)r" �tasks r% �shutdownzBackupInfoSender.shutdown( s� � � � ��%�t�';�<� � �D� �K�K�M�M�M��'�0�1�1� � �� � � � � � � � � � � � � � � � � � ���� � � � �� �%�%�'�'�'�'�'s � A�A � A � timestamp�returnc �F � t | t t f� � o| dk S )Nr )� isinstance�int�float)r. s r% �is_valid_timestampz#BackupInfoSender.is_valid_timestamp/ s � ��)�c�5�\�2�2�D�y�1�}�Dr' N�tsc � � |�| j n|}| � |� � s t � d|� � d S t dd|i� � d S )NzInvalid timestamp: %sr �last_send_timestamp)r r4 �logger�warningr )r" r5 r. s r% r+ z)BackupInfoSender.save_last_send_timestamp3 s` � �13��D�-�-�� ��&�&�y�1�1� � �N�N�2�I� >� >� >� �F��%�(=�y�'I�J�J�J�J�Jr' c � � t d� � � d� � }| � |� � s t � d� � d}|S )Nr r7 z(Invalid timestamp loaded, resetting to 0r )r �getr4 r8 r9 )r" r. s r% r z)BackupInfoSender.load_last_send_timestamp: sS � � � 1�2�2�6�6�7L�M�M� ��&�&�y�1�1� � �N�N�E� F� F� F��I��r' c � � K � t j � � | j z t k r| j � � � d S d S r )�timer � SEND_INTERVALr �set)r" s r% r z.BackupInfoSender._recurring_check_data_to_sendA sC � � � � �9�;�;��2� 2�m� C� C� � � � � "� "� "� "� "� D� Cr' r c � � K � | j � � � � d {V �� | � � � � d {V �� n2# t $ r%}t � d|� � Y d }~nd }~ww xY wt j � � | _ | j � � � d S # t j � � | _ | j � � � w xY w)Nz!Failed to collect backup info: %s) r �wait�_send_server_config� Exceptionr8 � exceptionr= r �clear)r" �es r% r z%BackupInfoSender._recurring_send_statF s� � � � �� �#�#�%�%�%�%�%�%�%�%�%� %��*�*�,�,� ,� ,� ,� ,� ,� ,� ,� ,��� E� E� E� � � � @�!� D� D� D� D� D� D� D� D����� E���� )-� � � �D� %� � � "� "� $� $� $� $� $�� )-� � � �D� %� � � "� "� $� $� $� $���s, �> �B# � A-�A(�#B# �(A-�-B# �#3Cc � � K � t j t � � t � � � d {V ���� � }| j � |� � � d {V �� d S )N)�backup_provider_type�last_backup_timestamp)r � BackupInfor r r �process_message)r" � confg_msgs r% rB z$BackupInfoSender._send_server_configR sq � � � � �*�!4�!6�!6�(A�(C�(C�"C�"C�"C�"C�"C�"C� � � � � �j�(�(��3�3�3�3�3�3�3�3�3�3�3r' r )�__name__� __module__� __qualname__�__doc__r �IM360�SCOPEr& r- � staticmethodr r2 r3 �boolr4 r+ r r �RECURRING_CHECK_INTERVALr r rB � r' r% r r s � � � � � �8�8� �K�E� � � �(� (� (� �E�e�C��J�&7� E�D� E� E� E� �\�E�K� K�5��e��+<� K� K� K� K�� � � �_�-�.�.�#� #� /�.�#� �_�Q��� %� %� �� %�4� 4� 4� 4� 4r' r ) r r= � contextlibr �datetimer �loggingr �typingr �"defence360agent.contracts.messagesr �!defence360agent.contracts.pluginsr �%defence360agent.subsys.backup_systemsr r �'defence360agent.subsys.persistent_stater r �defence360agent.utilsr r rM r8 r2 � total_secondsr> rU r rV r' r% �<module>ra sY �� ���� � � � � � � � � � � � � � � � � � � � � � � � � � � � � :� :� :� :� :� :� ;� ;� ;� ;� ;� ;�� � � � � � � � K� J� J� J� J� J� J� J� 8� 8� 8� 8� 8� 8� 8� 8� ��8� � ����I�I�B�'�'�'�5�5�7�7�8�8� � � �@4� @4� @4� @4� @4�}� @4� @4� @4� @4� @4r'