Ki dZddlmZmZddlmZddlmZmZmZddl m Z m Z ddl m Z ddlmZmZmZmZedZe d ed eed eeefd ed zfdZe ded efdZGddeZGddeZy )aJSerialization adapter base class for converting ManagedEntry objects to/from store-specific formats. This module provides the SerializationAdapter ABC that store implementations should use to define their own serialization strategy. Store-specific adapter implementations should be defined within their respective store modules. )ABCabstractmethod)datetime)AnyLiteralTypeVar)DeserializationErrorSerializationError) bear_enforce) ManagedEntry dump_to_jsonload_from_json verify_dictTkey expected_type dictionaryreturnNcl||vryt|||s|d|j}t|||S)Nz must be a ) isinstance__name__ TypeError)rrrmsgs p/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/key_value/shared/utils/serialization.py key_must_bersF * jo} 5[!7!7 89n c?valuecl tj|S#t$rd|}t|dwxYw)NzInvalid datetime string: message)r fromisoformat ValueErrorr )rrs rparse_datetime_strr#s@:%%e,, :)%1"3/T9:s3c~eZdZUdZdZeddzed<dZeddzed<ddd d eddzd eddzd dfd Zde d e fdZ e de e efd e e effdZde e efd e fdZe de e efd e e effdZ dddddde dede dzde dzded e e eff dZ dddddde dede dzde dzded e f dZy)SerializationAdapteraBase class for store-specific serialization adapters. Adapters encapsulate the logic for converting between ManagedEntry objects and store-specific storage formats. This provides a consistent interface while allowing each store to optimize its serialization strategy. isoformatr&rN _date_formatdictstringr) _value_format date_format value_formatr.r/rc ||_||_yN)r(r,)selfr.r/s r__init__zSerializationAdapter.__init__1s()rjson_strc>t|}|j|S)z(Convert a JSON string to a ManagedEntry.r4data)r load_dict)r2r4 loaded_datas r load_jsonzSerializationAdapter.load_json7s&4h&G ~~;~//rr8cy)zPrepare data for loading. This method is used by subclasses to handle any required transformations before loading the data into a ManagedEntry.Nr2r8s r prepare_loadz!SerializationAdapter.prepare_load=rcp|j|}i}|jdk(rFt|dtx}rt ||d<t|dtx}rt ||d<|jdk(r2t|dt x}r||d<t|dt x}r||d<d|vrd }t | |d}i}t|tr t| }n+t|tr t| }nd }t | t||jd|jdS)z'Convert a dictionary to a ManagedEntry.r7r& created_at)rr)r expires_atrrzValue field not foundrr6objz)Value field is not a string or dictionary)rrBrC) r?r(rstrr#rr rrr)rr get)r2r8managed_entry_protorBrCrrmanaged_entry_values rr9zSerializationAdapter.load_dictCsL  d +.0    +(#L1(#L1 $ )C&s3 3W .0 eS !"0%"@  t $"-%"8 =C&s3 3%*..|<*..|<  rcy)zPrepare data for dumping to a dictionary. This method is used by subclasses to handle any required transformations before dumping the data to a dictionary.Nr=r>s r prepare_dumpz!SerializationAdapter.prepare_dumplr@r)r collectionversionentry exclude_nonerrMrNc||jdk(r |jn |jd}|||d<|||d<|jdk(r|j|d<|j |d<|jdk(r|j |d<|j|d<|r&|jDcic] \}}| || }}}|j| Scc}}w) aConvert a ManagedEntry to a dictionary. Args: entry: The ManagedEntry to serialize exclude_none: Whether to exclude None values from the output key: Optional unsanitized key name to include in the document collection: Optional unsanitized collection name to include in the document version: Document schema version (default: 1) Returns: A dictionary representation of the ManagedEntry with optional metadata fields r))rNrrrMr&rBrCrr7) r, value_as_dict value_as_jsonr(created_at_isoformatexpires_at_isoformatrBrCitemsrK) r2rOrPrrMrNr8kvs r dump_dictzSerializationAdapter.dump_dictrs.,0,>,>&,HU((eNaNa  ?DK  !!+D     +!&!;!;D !&!;!;D     *!&!1!1D !&!1!1D  %)ZZ\CTQQ]AqDCDC  d ++Ds ( C 3C c z|jdk(r d}t|t|j|||||S)aConvert a ManagedEntry to a JSON string. Args: entry: The ManagedEntry to serialize exclude_none: Whether to exclude None values from the output key: Optional unsanitized key name to include in the document collection: Optional unsanitized collection name to include in the document version: Document schema version (default: 1) Returns: A JSON string representation of the ManagedEntry with optional metadata fields rzbdump_json is incompatible with date_format="datetime"; use date_format="isoformat" or dump_dict().)rOrPrrMrNrD)r(r r rY)r2rOrPrrMrNrs r dump_jsonzSerializationAdapter.dump_jsonsP*    *vC$S) )U[^ku@G!HI Ir)T)r __module__ __qualname____doc__r(r__annotations__r,r3rFr r;rr)rr?r9rKboolintrYr[r=rrr%r%&s=HL'12T9G6E*cjk{c|@DdD* * 0#0,0 Ac3hADcNAA ' d38n' ' R}c3h}DcN}}",, !%,,,,,, 4Z ,, $J ,,,, c3h,,b"I !%III 4Z I $J II Irr%ceZdZdZddddeddzded dzd dffd Zd eeefd eeeffd Z d eeefd eeeffdZ xZ S)BasicSerializationAdapterzFBasic serialization adapter that does not perform any transformations.r&r)r-r.r'Nr/r*rc(t|||y)Nr-)superr3)r2r.r/ __class__s rr3z"BasicSerializationAdapter.__init__s [|Lrr8c|Sr1r=r>s rr?z&BasicSerializationAdapter.prepare_load rc|Sr1r=r>s rrKz&BasicSerializationAdapter.prepare_dumprhr) rr\r]r^rr3r)rFrr?rK __classcell__)rfs@rrcrcsPITGMM%&=>EMcjk{c|@DdDM M c3hDcNc3hDcNrrc)r^abcrrrtypingrrrkey_value.shared.errorsr r )key_value.shared.type_checking.bear_sprayr $key_value.shared.utils.managed_entryr r rrrrFtyper)rr#r%rcr=rrrqs$((LBhh CLCQDcNTUX\T\:c:h::RI3RIj  4 r