2F] d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z ej e
Zde de fdZd Zd ed
efd Zdd d ed
e e
ef fdZdS )zLJSON persistence helpers for small agent state files (no pickle at runtime). N)iscoroutinefunction)AnyCallableUnionobjreturnc t | t j r d | D S t | t r d | D S t | t
t f r d | D S | S )Nc , g | ]}t | S _to_jsonable.0items X/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/serialization.py
z _to_jsonable.. 333t T""333 c 4 i | ]\ }}|t | S r r )r kvs r
z _to_jsonable.. s$ ;;;tq!<??;;;r c , g | ]}t | S r r r s r r z _to_jsonable.. r r )
isinstance collectionsdequedictitemslisttuple)r s r r
r
s #{()) 433s3333#t <;;syy{{;;;;#e}%% 433s3333 Jr c t j t | }d | }t |dd 5 }| | ddd n # 1 swxY w Y t
j || dS )z-Atomically write ``obj`` to ``path`` as JSON.z{}.tmpwutf-8encodingN)jsondumpsr
formatopenwriteosreplace)pathr payloadtmpr" s r _dumpr0 s j c**++G
//$
C
c3 ) ) ) Q JsDs A++A/2A/r- attrc fd}|S )zZDecorator: after the wrapped method runs, persist ``self.``
to ``path`` as JSON.c t j fd }t j fd }t r|S |S )Nc | g|R i |}t | }t d| t | |S NzWrite %r to %rgetattrloggerdebugr0 selfargskwargsresultr r1 fr- s r wrapperz2serialize_attr..decorator..wrapper& s] Qt-d---f--F$%%C LL)3 5 5 5 $ Mr c K | g|R i | d {V }t | }t d| t | |S r5 r6 r: s r
async_wrapperz8serialize_attr..decorator..async_wrapper. ss 1T3D333F33333333F$%%C LL)3 5 5 5 $ Mr ) functoolswrapsr )r? r@ rB r1 r- s ` r decoratorz!serialize_attr..decorator% s
q ! ! ! r r )r- r1 rE s `` r serialize_attrrF ! s* * r )fallbackrG c t | dd 5 }t j | }ddd n # 1 swxY w Y t |t rt j | S |S # t $ r t d| Y n1t $ r%}t d| Y d}~nd}~ww xY wt | r
| n|S )zRestore an object from ``path`` (JSON); a top-level list becomes a
deque to match the legacy queue API, and missing/unparseable input
returns ``fallback`` (called if callable).rr# r$ Nz Can't find %s to unserializez.Unserialize failed with %r. Returning fallback)
r) r& loadr r r r FileNotFoundErrorr8 warning Exceptionerrorcallable)r- rG rI r es r unserializerQ = s:
$g
.
.
. !)A,,C c4 * $S)) )
= = =5t<<<<< J J J EqIIIIIIIIJ "(++ 988:::9s8 A+ 5 A+ 9A+ 9A+ +%C C B;;C )__doc__r rC r& loggingr+ asyncior typingr r r getLogger__name__r8 r
r0 strrF objectrQ r r r rZ s R R ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 8 $ $c c C s 8 CG : : : :h.>(? : : : : : :r