o i6% @sddlZddlZddlmZmZmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZddlmZmZmZmZmZeeZe dedZd Z d Z!Gd d d eeeZ" d)de ededefddZ#erwGdddZ$n&erddl%m&Z&ddl'm(Z(m)Z)ddl*m+Z+Gddde(Z$nGdddZ$   d*dedeee ede,de,def dd Z-d!ee.efdee.e eefffd"d#Z/dedee efd$d%Z0d&e"edefd'd(Z1dS)+N) TYPE_CHECKINGAnyDictGenericListOptionalSetTypeTypeVarUnioncast) BaseModel)DEFS_KEY PYDANTIC_V2JsonSchemaModemodels_json_schema v1_schema) ComponentsOpenAPI ReferenceSchemaschema_validate PydanticType)boundz#/components/schemas/z#/components/schemas/{model}c@seZdZUdZeeed<dS)PydanticSchemazBSpecial `Schema` class to indicate a reference from pydantic class schema_classN)__name__ __module__ __qualname____doc__r r__annotations__r"r"c/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/openapi_pydantic/v3/v3_0/util.pyr$s  r validationclsdefaultreturncCs<t|ds|S|jd|}|dvrtd|tt|S)a Get the JSON schema mode for a model class. The mode can be either "validation" or "serialization". In validation mode, computed fields are dropped and optional fields remain optional. In serialization mode, computed and optional fields are required. model_configjson_schema_mode)r$ serializationzinvalid json_schema_mode: )hasattrr(get ValueErrorr r)r%r&moder"r"r#get_mode+s  r/c@ eZdZdSGenerateOpenAPI30SchemaNrrrr"r"r"r#r2> r2)Enum)GenerateJsonSchemaJsonSchemaValue) core_schemac@s8eZdZdZdejdefddZdejdefddZ dS) r2z-Modify the schema generation for OpenAPI 3.0.schemar'cCs||d}d|d<|S)zGenerates a JSON schema that matches a schema that allows null values. In OpenAPI 3.0, types can not be None, but a special "nullable" field is available. r9Tnullable)generate_inner)selfr9inner_json_schemar"r"r#nullable_schemaIs z'GenerateOpenAPI30Schema.nullable_schemacCsdd|dD}dd|D}|thkr|ddS|thkr$|ddS|thkr.|d dS|thkr8|d dS|thkrB|d dSd |iS) zGenerates a JSON schema that matches a literal value. In OpenAPI 3.0, the "const" keyword is not supported, so this version of this method skips that optimization. cSs g|] }t|tr |jn|qSr") isinstancer5value).0vr"r"r# \sz:GenerateOpenAPI30Schema.literal_schema..expectedcSsh|]}t|qSr")type)rAer"r"r# `sz9GenerateOpenAPI30Schema.literal_schema..string)enumrEintegernumberbooleanarrayrI)strintfloatboollist)r<r9rDtypesr"r"r#literal_schemaVs          z&GenerateOpenAPI30Schema.literal_schemaN) rrrr r8NullableSchemar7r> LiteralSchemarTr"r"r"r#r2Fs  c@r0r1r3r"r"r"r#r2rr4Topen_apischema_classes"scan_for_pydantic_schema_referenceby_aliasc Cst|trdnd}|dd}|r!t|}|rth||}n|}|s%|S|jdddtd|trEtd d |D|tt d \}}nt ||t d }|j sSt |_ |j jr|j jD]} | |tvrttd | dt dt | dq[|j jt||St||j _|S)aJ Construct a new OpenAPI object, utilising pydantic classes to produce JSON schemas. :param open_api: the base `OpenAPI` object :param schema_classes: Pydantic classes that their schema will be used "#/components/schemas" values :param scan_for_pydantic_schema_reference: flag to indicate if scanning for `PydanticSchemaReference` class is needed for "#/components/schemas" value updates :param by_alias: construct schema by alias (default is True) :return: new OpenAPI object with "#/components/schemas" values updated. If there is no update in "#/components/schemas" values, the original `open_api` will be returned. model_copycopyT)deepcSs|jS)N)r)xr"r"r#sz6construct_open_api_with_schema_class..)keyzschema_classes: %scSsg|]}|t|fqSr")r/)rAcr"r"r#rCsz8construct_open_api_with_schema_class..)rZ ref_templateschema_generator)rZ ref_prefix"z" already exists in z. The value of "z" will be overwritten.)getattrr_handle_pydantic_schemarRsortloggerdebugrrbr2rrd componentsrschemasrwarningupdate_validate_schemas) rWrXrYrZ copy_func new_open_apiextracted_schema_classes_key_mapschema_definitions existing_keyr"r"r#$construct_open_api_with_schema_classusL       rvrtcCsdd|tDS)z9Convert JSON Schema definitions to parsed OpenAPI objectscSsi|] \}}|t|qSr")r)rAr` schema_dictr"r"r# sz%_validate_schemas..)ritems)rtr"r"r#ros rocs.tdtddffdd |tS)a This function traverses the `OpenAPI` object and 1. Replaces the `PydanticSchema` object with `Reference` object, with correct ref value; 2. Extracts the involved schema class from `PydanticSchema` object. **This function will mutate the input `OpenAPI` object.** :param open_api: the `OpenAPI` object to be traversed and mutated :return: a list of schema classes extracted from `PydanticSchema` objects objr'Ncs.t|tr9t|tr dnd}|D]&}||}t|tr2td||||t | |j q|qdSt|t rgt |D]"\}}t|tr`td|t |||< |j qB|qBdSt|tr|D]$\}}t|trtd|t |||< |j qp|qpdSdS)Nmodel_fields_set__fields_set__zPydanticSchema found in %s: %szPydanticSchema found in list: zPydanticSchema found in dict: )r?r rfr__getattribute__rrirj __setattr___construct_ref_objaddrrR enumeratedictry)rzfieldsfield child_objindexelemr`r@ _traversepydantic_typesr"r#rs:             z*_handle_pydantic_schema.._traverse)setrrR)rWr"rr#rgsrgpydantic_schemacCsDtdd|jjdd}tdidt|i}td||S) a Construct a reference object from the Pydantic schema name characters in the schema name that are invalid/problematic for JSONschema $ref names will get replaced with underscores. Especially needed for Pydantic generic Models with brackets "[]" see: https://github.com/pydantic/pydantic/blob/aee6057378ccfec02126bf9c984a9b6d6b411777/pydantic/json_schema.py#L2031 z[^a-zA-Z0-9.\-_]_.__z$refzref_obj=Nr") resubrrreplacerrdrirj)rref_nameref_objr"r"r#rs r)r$)NTT)2loggingrtypingrrrrrrrr r r r pydanticr openapi_pydantic.compatrrrrrrrrrr getLoggerrrirrdrbrr/r2rIr5pydantic.json_schemar6r7 pydantic_corer8rQrvrNrorgrr"r"r"r#s`4      , A   2