관리-도구
편집 파일: serialization.cpython-311.pyc
� IMg� � � � d dl Z d dlZd dlZd dlmZ d dlmZmZ ej e � � Z d� Z de de fd�Z dd�de d eeef fd �ZdS ) � N)�iscoroutinefunction)�Union�Callablec � � t | d� � 5 }t j ||� � d d d � � d S # 1 swxY w Y d S )N�wb)�open�pickle�dump)�path�obj�ws �X/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/serialization.py�_dumpr s� � � �d�D� � � �Q�� �C����� � � � � � � � � � � � ���� � � � � � s �4�8�8r �attrc � � �� �� fd�}|S )z� Make decorator to serialize an object or object's attribute :param path: path to file to serialize into :param attr: attribute name to serialize c � �� � t j � � � �� �fd�� � }t j � � � �� �fd�� � }t � � � r|S |S )Nc � �� �| g|�R i |��}t | �� � }t � d|�� � t �|� � |S �NzWrite %r to %r��getattr�logger�debugr ��self�args�kwargs�resultr r �fr s ���r �wrapperz2serialize_attr.<locals>.decorator.<locals>.wrapper s] �� ��Q�t�-�d�-�-�-�f�-�-�F��$��%�%�C� �L�L�)�3�� 5� 5� 5� �$�� � � ��M� c � � �K � �| g|�R i |��� d {V ��}t | �� � }t � d|�� � t �|� � |S r r r s ���r � async_wrapperz8serialize_attr.<locals>.decorator.<locals>.async_wrapper ss �� � � � �1�T�3�D�3�3�3�F�3�3�3�3�3�3�3�3�F��$��%�%�C� �L�L�)�3�� 5� 5� 5� �$�� � � ��Mr )� functools�wrapsr )r r r" r r s ` ��r � decoratorz!serialize_attr.<locals>.decorator s� ��� � � �� � � � � � � � � � � � � �� � � � � � � � � � � � �q� !� !� !� � ��r � )r r r% s `` r �serialize_attrr' s* ��� �� � � � � �* �r )�fallbackr( c �d � t | d� � 5 }t j |� � cddd� � S # 1 swxY w Y nX# t $ r t � d| � � Y n1t $ r%}t � d|� � Y d}~nd}~ww xY wt |� � r |� � n|S )z� Restore an object from file :param path: path to read from :param fallback: return or call it if unserialization fails :return: �rbNz Can't find %s to unserializez1Unserialize failed with %r. Returning an fallback) r r �load�FileNotFoundErrorr �warning� Exception�error�callable)r r( �r�es r � unserializer3 / s � �M� �$�� � � "���;�q�>�>� "� "� "� "� "� "� "� "� "� "� "� "���� "� "� "� "� "�� � =� =� =����5�t�<�<�<�<�<� � M� M� M�� � �H�!�L�L�L�L�L�L�L�L�����M����!�(�+�+� 9�8�8�:�:�:��9s3 �? �3� ? �7�? �7�? �%B�& B�/B�B)r# �loggingr �asyncior �typingr r � getLogger�__name__r r �strr' �objectr3 r& r r �<module>r; s� �� � � � � ���� � � � � '� '� '� '� '� '� "� "� "� "� "� "� "� "� �� �8� $� $�� � � � �C� �s� � � � �> CG� :� :� :�� :��h��.>�(?� :� :� :� :� :� :r