o i@sddlmZddlmZmZmZmZmZddlm Z m Z ddl m Z ddl mZmZddlmZddlmZddlmZed eeeBd ZGd d d eeZd S))Sequence)AnyGeneric SupportsFloatTypeVar get_origin)DeserializationErrorSerializationError) bear_spray) BaseModelValidationError) TypeAdapter)PydanticSerializationError) AsyncKeyValueT)boundc @seZdZdZe  d'dedeededBde ddf d d Z d e ee fdedBfd d Z d ede ee ffddZdddededBdedBfddZdddeededBdeedBfddZdddded ededBdedBddf ddZddddeedeededBdeedBdBddf ddZdddededBde fdd ZdddeededBdefd!d"ZdddededBdeedBedBffd#d$ZdddeededBdeeedBedBffd%d&ZdS)(PydanticAdapterz^Adapter around a KVStore-compliant Store that allows type-safe persistence of Pydantic models.NF key_valuepydantic_modeldefault_collectionraise_on_validation_errorreturncCs@||_t|}|duot|t|_tt||_||_||_ dS)a9Create a new PydanticAdapter. Args: key_value: The KVStore to use. pydantic_model: The Pydantic model to use. default_collection: The default collection to use. raise_on_validation_error: Whether to raise a ValidationError if the model is invalid. N) _key_valuer issubclassr_is_list_modelr r _type_adapter_default_collection_raise_on_validation_error)selfrrrroriginr m/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/key_value/aio/adapters/pydantic/adapter.py__init__s  zPydanticAdapter.__init__valuec Cslz|jr|j|dgWS|j|WSty5}z|jr*d|}t||WYd}~dSd}~ww)NitemsInvalid Pydantic model: )rrvalidate_pythongetr rrrr#emsgr r r!_validate_model.s  zPydanticAdapter._validate_modelc Cs\z|jrd|jj|ddiWS|jj|ddWSty-}z d|}t||d}~ww)Nr$json)moder%)rr dump_pythonrr r(r r r!_serialize_model:s  z PydanticAdapter._serialize_model) collectionkeyr0cs6|p|j}|jj||dIdH}r|j|dSdS)a0Get and validate a model by key. Returns: The parsed model instance, or None if not present. Raises: DeserializationError if the stored data cannot be validated as the model and the PydanticAdapter is configured to raise on validation error. r1r0Nr#)rrr'r+)rr1r0r#r r r!r'Ds   zPydanticAdapter.getkeysc4|pj}jj||dIdH}fdd|DS)aKBatch get and validate models by keys, preserving order. Returns: A list of parsed model instances, or None if missing. Raises: DeserializationError if the stored data cannot be validated as the model and the PydanticAdapter is configured to raise on validation error. r4r0Ncs g|] }|r j|dndqSr3Nr+.0r#rr r! cs z,PydanticAdapter.get_many..)rrget_many)rr4r0valuesr r;r!r=Us zPydanticAdapter.get_many)r0ttlr?cs6|p|j}|j|d}|jj||||dIdHdS)zoSerialize and store a model. Propagates SerializationError if the model cannot be serialized. r3)r1r#r0r?N)rr/rput)rr1r#r0r? value_dictr r r!r@es  zPydanticAdapter.putr>cs<|pj}fdd|D}jj||||dIdHdS)zJSerialize and store multiple models, preserving order alignment with keys.csg|]}j|dqS)r3)r/r9r;r r!r<vsz,PydanticAdapter.put_many..)r4r>r0r?N)rrput_many)rr4r>r0r? value_dictsr r;r!rBps zPydanticAdapter.put_manyc"|p|j}|jj||dIdHS)zGDelete a model by key. Returns True if a value was deleted, else False.r2N)rrdelete)rr1r0r r r!rEz zPydanticAdapter.deletecrD)zDDelete multiple models by key. Returns the count of deleted entries.r6N)rr delete_many)rr4r0r r r!rGrFzPydanticAdapter.delete_manycsN|p|j}|jj||dIdH\}}|durdS|j|d}r%||fSdS)zvGet a model and its TTL seconds if present. Returns (model, ttl_seconds) or (None, None) if missing. r2N)NNr3)rrr?r+)rr1r0entryttl_infovalidated_modelr r r!r?s zPydanticAdapter.ttlcr5)zKBatch get models with TTLs. Each element is (model|None, ttl_seconds|None).r6Ncs(g|]\}}|rj|dnd|fqSr7r8)r:rHrIr;r r!r<s(z,PydanticAdapter.ttl_many..)rrttl_many)rr4r0entriesr r;r!rKs zPydanticAdapter.ttl_many)NF)__name__ __module__ __qualname____doc__r rtyperstrboolr"dictrr+r/r'listr=rr@rrBrEintrGtuplefloatr?rKr r r r!rsJ  $ ,.  $0<rN)collections.abcrtypingrrrrrkey_value.shared.errorsrr )key_value.shared.type_checking.bear_sprayr pydanticr r pydantic.type_adapterr pydantic_corer!key_value.aio.protocols.key_valuerrrr r r r!s