ѭO"/% d dl Z d dlZd dlZd dlmZ d dlmZmZ ej e Z
d Z de de fdZ
ddde d eeef fd
ZdS ) N)iscoroutinefunction)UnionCallablec t | d 5 }t j || d d d d S # 1 swxY w Y d S )Nwb)openpickledump)pathobjws X/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/serialization.py_dumpr
s
dD Q C s 488r 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 %rgetattrloggerdebugr selfargskwargsresultr r fr s r wrapperz2serialize_attr..decorator..wrapper s] Qt-d---f--F$%%C LL)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..decorator..async_wrapper ss 1T3D333F33333333F$%%C LL)3 5 5 5 $ Mr ) functoolswrapsr )r r r" r r s ` r decoratorz!serialize_attr..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 loadFileNotFoundErrorr warning Exceptionerrorcallable)r r( res r unserializer3 / s M
$
";q>> " " " " " " " " " " " " " " " " " = = =5t<<<<< M M M H!LLLLLLLLM!(++ 988:::9s3 ? 3 ? 7? 7? %B& B/BB)r# loggingr asyncior typingr r getLogger__name__r r strr' objectr3 r& r r r; s
' ' ' ' ' ' " " " " " " " " 8 $ $ C s > CG : : : :h.>(? : : : : : :r