Ki-ddlZddlmZmZddlmZddlmZmZm Z m Z m Z ddl m Z mZddlmZddlmZddlmZdd lmZej.eZe d ZGd d eeeZy) N)ABCabstractmethod)Sequence)AnyGeneric SupportsFloatTypeVaroverload)DeserializationErrorSerializationError)ValidationError) TypeAdapter)PydanticSerializationError) AsyncKeyValueTc &eZdZUdZeed<eed<eeed<e dzed<eed<e de fd Z d e e e fdedzfd Zd ede e e ffd Zedd de de dzdedefdZedddde de dzdddedzfdZdddde de dzdedzdedzfdZedd dee de dzdedeefdZeddddee de dzdddeedzfdZddddee de dzdedzdeeeedzzfdZdddde d ede dzdedzddf dZddddee deede dzdedzddf dZdd de de dzdefdZdd dee de dzdefdZdd de de dzdeedzedzffd Zdd dee de dzdeeedzedzffd!Zy)"BasePydanticAdapteraBBase adapter using Pydantic TypeAdapter for validation and serialization. This abstract base class provides shared functionality for adapters that use Pydantic's TypeAdapter for validation and serialization. Concrete subclasses must implement _get_model_type_name() to provide appropriate error messages. _key_value_is_list_model _type_adapterN_default_collection_raise_on_validation_errorreturncy)zReturn the model type name for error messages. Returns: A string describing the model type (e.g., "Pydantic model", "dataclass"). N)selfs p/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/key_value/aio/adapters/pydantic/base.py_get_model_type_namez(BasePydanticAdapter._get_model_type_name s valuec  |jrd|vrd|jrd|jd}t|tj d|j|jdddy |j j|dS|j j|S#t$r}|jr4|jd }d|jd |}t|||jd }tj d |j|jt||d dYd }~y d }~wwxYw)aValidate and deserialize a dict into the configured model type. This method handles both single models and list models. For list models, it expects the value to contain an "items" key with the list data, following the convention used by `_serialize_model`. If validation fails and `raise_on_validation_error` is False, returns None instead of raising. Args: value: The dict to validate and convert to a model. Returns: The validated model instance, or None if validation fails and errors are suppressed. Raises: DeserializationError: If validation fails and `raise_on_validation_error` is True. itemsInvalid z! payload: missing 'items' wrapperz#Missing 'items' wrapper for list %szmissing 'items' wrapper) model_typeerrorF)extraexc_infoN) include_input: zValidation failed for %s)r$ error_counterrorsT) rrrr loggerr%rvalidate_pythonr r+len)rr msgedetails error_detailss r_validate_modelz#BasePydanticAdapter._validate_model)sa ' ""%'66 ()B)B)D(EEfg2377LL=113*.*C*C*E%>"'! ))99%.II%%55e< < ..(((7 !:!:!< =RyI*3/Q6HH5H9M LL*))+"&";";"=#&}#5+   % s%A3B/6B/B// E8BEEc |jrd|jj|diS|jj|dS#t$r'}d|j d|}t ||d}~wwxYw)avSerialize a model to a dict for storage. This method handles both single models and list models. For list models, it wraps the serialized list in a dict with an "items" key (e.g., {"items": [...]}) to ensure consistent dict-based storage format across all value types. This wrapping convention is expected by `_validate_model` during deserialization. Args: value: The model instance to serialize. Returns: A dict representation of the model suitable for storage. Raises: SerializationError: If the model cannot be serialized. r"json)moder#r)N)rr dump_pythonrrr )rr r0r/s r_serialize_modelz$BasePydanticAdapter._serialize_modelbs" 1""!3!3!?!?F!?!STT%%11%f1E E) 1T6689A3?C$S)q 0 1s*A A A:"A55A:) collectionkeyr9defaultc KywNrrr:r9r;s rgetzBasePydanticAdapter.get|s VY)r9r;c Kywr=rr>s rr?zBasePydanticAdapter.gets gjr@cK|xs |j}|jj||d{}||j|}||S|S7w)a<Get and validate a model by key. Args: key: The key to retrieve. collection: The collection to use. If not provided, uses the default collection. default: The default value to return if the key doesn't exist or validation fails. Returns: The parsed model instance if found and valid, or the default value if key doesn't exist or validation fails. Raises: DeserializationError: If the stored data cannot be validated as the model and the adapter is configured to raise on validation error. Note: When raise_on_validation_error=False and validation fails, returns the default value (which may be None). When raise_on_validation_error=True and validation fails, raises DeserializationError. r:r9Nr )rrr?r3)rr:r9r;r validateds rr?zBasePydanticAdapter.getsd& ;4#;#; oo))cj)II  ,,5,9I$   Js1AAAkeysc Kywr=rrrFr9r;s rget_manyzBasePydanticAdapter.get_manys lor@c Kywr=rrHs rrIzBasePydanticAdapter.get_manys ~Ar@cK|xs |j}|jj||d{}g}|D]=}||j||j |}|j||n|?|S7Jw)aBatch get and validate models by keys, preserving order. Args: keys: The list of keys to retrieve. collection: The collection to use. If not provided, uses the default collection. default: The default value to return for keys that don't exist or fail validation. Returns: A list of parsed model instances, with default values for missing keys or validation failures. Raises: DeserializationError: If the stored data cannot be validated as the model and the adapter is configured to raise on validation error. Note: When raise_on_validation_error=False and validation fails for any key, that position in the returned list will contain the default value (which may be None). The method returns a complete list matching the order and length of the input keys, with defaults substituted for missing or invalid entries. rFr9NrD)rrrIappendr3)rrFr9r;valuesresultr rEs rrIzBasePydanticAdapter.get_manys( ;4#;#; 48OO4L4LRVcm4L4n.n!# OE} g& 00u0=  9+@igN  O  /os1BA>A B)r9ttlrPcK|xs |j}|j|}|jj||||d{y7w)zoSerialize and store a model. Propagates SerializationError if the model cannot be serialized. rD)r:r r9rPN)rr8rput)rr:r r9rP value_dicts rrRzBasePydanticAdapter.putsM  ;4#;#; %)%:%:%:%G oo!!c X[!\\\sAAA ArNcK|xs |j}|Dcgc]}|j|}}|jj||||d{ycc}w7 w)zJSerialize and store multiple models, preserving order alignment with keys.rD)rFrNr9rPN)rr8rput_many)rrFrNr9rPr value_dictss rrUzBasePydanticAdapter.put_manysf ;4#;#; ]c,dTYT-B-B-B-O,d ,doo&&DQ[ad&eee-eesA!A$A!A A!cxK|xs |j}|jj||d{S7w)zGDelete a model by key. Returns True if a value was deleted, else False.rCN)rrdelete)rr:r9s rrXzBasePydanticAdapter.deletes6;4#;#; __++ +KKKK 1:8:cxK|xs |j}|jj||d{S7w)zDDelete multiple models by key. Returns the count of deleted entries.rLN)rr delete_many)rrFr9s rr[zBasePydanticAdapter.delete_manys6;4#;#; __00dz0RRRRrYcK|xs |j}|jj||d{\}}|y|j|}|||fSy7#w)a,Get a model and its TTL seconds if present. Args: key: The key to retrieve. collection: The collection to use. If not provided, uses the default collection. Returns: A tuple of (model, ttl_seconds). Returns (None, None) if the key is missing or validation fails. Note: When validation fails and raise_on_validation_error=False, returns (None, None) even if TTL data exists. When validation fails and raise_on_validation_error=True, raises DeserializationError. rCN)NNrD)rrrPr3)rr:r9entryttl_infovalidated_models rrPzBasePydanticAdapter.ttlso ;4#;#; !% 3 3 3 SSx =..U.;  &#X. .Ts1AA$AcK|xs |j}|jj||d{}|Dcgc]\}}||j|nd|fc}}S7/cc}}ww)zKBatch get models with TTLs. Each element is (model|None, ttl_seconds|None).rLNrD)rrttl_manyr3)rrFr9entriesr]r^s rrazBasePydanticAdapter.ttl_manys};4#;#; JN//JbJbhlzDJbKEEEt{|apafhpe6G%%E%2TS[\||EE|s 1A+A# A+"A%A+%A+)__name__ __module__ __qualname____doc__r__annotations__boolrrstrrrdictrr3r8r r?rlistrIrrRrUrXintr[tuplefloatrPrarrrrrstq>!t# $$ c  7T#s(^7D7r1a1DcN14>BYSYtYQYSTYY >BTXjSjtjTj]^ae]ejj>BX\StQQUXabeiai:NRo8C=oto]^ocghicjoo NRdhA8C=AtA]aAmqrsvzrzm{AANRhl8C=t]^ae]equvwqx{ABEIAI|JrJBIMjn ]S ] ]3: ]S`cgSg ]sw ]UYvzfSMf+3A;fGJTzf_los_sf fBFLLC$JL$L RVShsmSC$JSZ]S ?CStuQQUXW\_cWcMcGd<OS}8C=}t}W[\abcfjbjlqtxlxbx\yWz}rr)loggingabcrrcollections.abcrtypingrrrr r key_value.shared.errorsr r pydanticr pydantic.type_adapterr pydantic_corer!key_value.aio.protocols.key_valuer getLoggerrcr,rrrrrrysT#$AAL$-4;   8 $ CLy}'!*cy}r