Ki ddlmZddlmZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZedZGd d eeZy ) )Sequence) is_dataclass)AnyTypeVarget_args get_origin) bear_spray) TypeAdapter)BasePydanticAdapter) AsyncKeyValueTc feZdZUdZeeed<e d dedee de dzde ddf d Z de fd Z y) DataclassAdapteraAdapter around a KVStore-compliant Store that allows type-safe persistence of dataclasses. This adapter works with both standard library dataclasses and Pydantic dataclasses, leveraging Pydantic's TypeAdapter for robust validation and serialization. _inner_typeN key_valuedataclass_typedefault_collectionraise_on_validation_errorreturnc||_t|}|duxr"t|txrt |t |_|j r)t|}|sd|d}t||d|_ n||_ t|js|jd}t|tt||_ ||_||_y)a-Create a new DataclassAdapter. Args: key_value: The AsyncKeyValue to use. dataclass_type: The dataclass type to use. Can be a single dataclass or list[dataclass]. default_collection: The default collection to use. raise_on_validation_error: Whether to raise a DeserializationError if validation fails during reads. Otherwise, calls will return None if validation fails. Raises: TypeError: If dataclass_type is not a dataclass type. Nz List type z must have a type argumentrz is not a dataclass) _key_valuer isinstancetype issubclassr_is_list_modelr TypeErrorrrr r _type_adapter_default_collection_raise_on_validation_error)selfrrrroriginargsmsgs t/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/key_value/aio/adapters/dataclass/adapter.py__init__zDataclassAdapter.__init__s($N+$D0nZ5MnR\]cemRn   N+D">"22LMn$#AwD -D D,,-%%&&9:CC. (^N;#5 *C'cy)z.Return the model type name for error messages. dataclass)r s r$_get_model_type_namez%DataclassAdapter._get_model_type_nameEsr&)NF)__name__ __module__ __qualname____doc__rr__annotations__r r r strboolr%r*r)r&r$rrs{ c *.*/ )D )DQ)D $J )D $( )D  )D)DVcr&rN)collections.abcr dataclassesrtypingrrrr)key_value.shared.type_checking.bear_sprayr pydantic.type_adapterr $key_value.aio.adapters.pydantic.baser !key_value.aio.protocols.key_valuer r rr)r&r$r9s8$$55@-D; CL9*1-9r&