Edit File: persistent_state.cpython-311.pyc
� a�nh� � � � d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z ee � � Z ed� � Z e dz Z e� � Zd ed ee j e j e j f d efd �Zd edefd�Zd efd�Zd� ZdS )� N)� getLogger)�Path)�Literal)�ANTIVIRUS_MODE)�Scopez/var/imunify360z.persistent_state� lock_file�scope�returnc �> � t d| � d�z }|t j k rt � |� � nc|t j k r"t rt � |� � n1|t j k r!t st � |� � |S )z%Register lock file for further usage.�.z.lock)�PERSISTENT_STATE_DIRr �AV_IM360� LOCK_FILES�add�AVr �IM360)r r � _lock_files �\/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/subsys/persistent_state.py�register_lock_filer s� � � &�(<�I�(<�(<�(<�<�J� ��� � ����z�"�"�"�"� �%�(� � �~� ����z�"�"�"�"� �%�+� � �n� ����z�"�"�"� �� � class_name�valuesc � � t } |� dd�� � || � d�z }t j ||� d� � � � dS # t t f$ r'}t � d| |� � Y d}~dS d}~ww xY w)z1Save state to a file in .persistent_state folder.T)�parents�exist_ok�.state�wzFailed to save state: %s %sN) r �mkdir�json�dump�open�AttributeError�OSError�logger�error)r r � folder_path� file_path�es r � save_stater) s� � � '�K�C����$���6�6�6� �Z�"7�"7�"7�7� � � �&�)�.�.��-�-�.�.�.�.�.�� �G� $� C� C� C�� � �2�J��B�B�B�B�B�B�B�B�B�����C���s �AA �B �# B�B c �> � t }|| � d�z }|� � � rm t j |� d� � � � S # t j t t f$ r&}t � d| |� � Y d}~nd}~ww xY wt � � S )z3Load state from a file in .persistent_state folder.r �rzFailed to load state: %s %sN) r �existsr �loadr! �JSONDecodeErrorr# �UnicodeDecodeErrorr$ r% �dict)r r& r' r( s r � load_stater1 * s� � � '�K��� 3� 3� 3�3�I������ G� G��9�Y�^�^�C� 0� 0�1�1� 1���$�g�/A�B� G� G� G� �L�L�6� �A� F� F� F� F� F� F� F� F����� G���� �6�6�Ms �&A � B�( B � Bc �z � t � d� � D ] } | t vr| � � � � dS )z;Remove all unused lock files from .persistent_state folder.z*.lockN)r �globr �unlink)r s r �remove_unused_locksr5 8 sG � �)�.�.�x�8�8� � � � �J� &� &� � � � � � �� � r )r �loggingr �pathlibr �typingr � defence360agent.contracts.configr �!defence360agent.contracts.pluginsr �__name__r$ �BASE_DIRr �setr �strr r r r r0 r) r1 r5 � r r �<module>r@ s: �� � � � � � � � � � � � � � � � � � � � � � � ;� ;� ;� ;� ;� ;� 3� 3� 3� 3� 3� 3� ��8� � �� �4�!� "� "�� �"5�5� � �S�U�U� � �� �"�5�8�U�[�%�.�#H�I� � � � � � � C�3� C�� C� C� C� C� �d� � � � � � � � � r