iCUddlmZddlZddlZddlmZddlmZmZm Z m Z ddl m Z mZdZdZd ed <ded  dd Zded  dd ZddZdZddZddZddZGddZy)) annotationsN)Iterable)AnyCallableFinalOptional)inspect to_builtins)schemaschema_components#/$defs/{name}r _REF_TEMPLATE schema_hook ref_templatec<t|f||\\}}|r||d<|S)aGenerate a JSON Schema for a given type. Any schemas for (potentially) shared components are extracted and stored in a top-level ``"$defs"`` field. If you want to generate schemas for multiple types, or to have more control over the generated schema you may want to use ``schema_components`` instead. Parameters ---------- type : type The type to generate the schema for. schema_hook : callable, optional An optional callback to use for generating JSON schemas of custom types. Will be called with the custom type, and should return a dict representation of the JSON schema for that type. ref_template : str, optional A template to use when generating ``"$ref"`` fields. This template is formatted with the type name as ``template.format(name=name)``. This can be useful if you intend to store the ``components`` mapping somewhere other than a top-level ``"$defs"`` field. For example, you might use ``ref_template="#/components/{name}"`` if generating an OpenAPI schema. Returns ------- schema : dict The generated JSON Schema. See Also -------- schema_components rz$defs)r )typerrout componentss G/root/searxng-venv/lib/python3.12/site-packages/msgspec/_json_schema.pyr r s4N+ !FSJ !G Jc  tj|}t|}t|}t ||| t fd|D}|j Dcic]\}}|| j|d} }}|| fScc}}w)aGenerate JSON Schemas for one or more types. Any schemas for (potentially) shared components are extracted and returned in a separate ``components`` dict. Parameters ---------- types : Iterable[type] An iterable of one or more types to generate schemas for. schema_hook : callable, optional An optional callback to use for generating JSON schemas of custom types. Will be called with the custom type, and should return a dict representation of the JSON schema for that type. ref_template : str, optional A template to use when generating ``"$ref"`` fields. This template is formatted with the type name as ``template.format(name=name)``. This can be useful if you intend to store the ``components`` mapping somewhere other than a top-level ``"$defs"`` field. For example, you might use ``ref_template="#/components/{name}"`` if generating an OpenAPI schema. Returns ------- schemas : tuple[dict] A tuple of JSON Schemas, one for each type in ``types``. components : dict A mapping of name to schema for any shared components used by ``schemas``. See Also -------- schema c3@K|]}j|ywN) to_schema).0tgens r z$schema_components..os9CMM!$9sF)mimulti_type_info_collect_component_types_build_name_map_SchemaGeneratortupleitemsr) typesrr type_infoscomponent_typesname_mapschemasclsrrrs @rr r @sN##E*J.z:O/H 8[, ?C9j99G>M=R=R=T393 s}}Q..J J s! Bc8ifd|D] }| S)zFind all types in the type tree that are "nameable" and worthy of being extracted out into a shared top-level components mapping. Currently this looks for Struct, Dataclass, NamedTuple, TypedDict, and Enum types. cpt|tjtjtjtj frB|j vr3||j <|jD]}|jyyt|tjr||j <yt|tjr|jyt|tjr|jyt|tjr|jD] }| yt|tjr%|j |j"yt|tj$r|j&D] }| yyr) isinstancer! StructType TypedDictType DataclassTypeNamedTupleTyper-fieldsrEnumTypeMetadataCollectionType item_type TupleType item_typesDictTypekey_type value_type UnionTyper()rfstcollectrs rrBz)_collect_component_types..collects<   r//1A1A2CTCTU uuJ&$% 155!$AAFFO$'2;; ' !Jquu  2;; ' AFFO 2,, - AKK 2<< (ll    2;; ' AJJ  ALL ! 2<< (gg   )r)r)rrBrs @@rr#r#ws/J0  rcPt|tr |jSt|Sr)r0r__name__repr)objs r _type_reprrHs%c403<<?d3i?rct|dr@|jj}djd|jD}|d|dS|jS)N __origin__z, c32K|]}t|ywr)rH)ras rr z"_get_class_name..s=1A=s[])hasattrrJrEjoin__args__)r-nameargss r_get_class_namerTsPsL!~~&&yy= ==qa  <<rct|dsJt|jd|j}t|dd}|sytj|j d}t |tjr|dk(ry|St |tjtjfr0|j|jdr|jdry|S) Nr-rJ__doc__z zAn enumeration.())rOgetattrr-textwrapdedentstripr0r!r6r4r3 startswithrEendswith)rr-docs r_get_docras 1e   !%%quu -C #y" %C  //#  $ $V ,C!R[[! # # J A))2+;+;< = >>S\\N!, -#,,s2C Jrc< d fd}t}i}|D]Y} t|}||vr-|j|}|j|||||<||vr ||||<U|||<[|j Dcic]\}}|| c}}Scc}}w)zA mapping from nameable subcomponents to a generated name. The generated name is usually a normalized version of the class name. In the case of conflicts, the name will be expanded to also include the full import path. c0tjdd|S)Nz[^a-zA-Z0-9.\-_]_)resubrRs r normalizez"_build_name_map..normalizesvv)355rcF|jd|jS)N.) __module__ __qualname__)r-rhs rfullnamez!_build_name_map..fullnames%CNN+1S-=-=,>?@@r)setrTpopaddr') r*rm conflictsnamesr-rRoldkvrhs @rr$r$s6AIE -. 5=))D/C MM$ #&E(3- 9 #&E(3- E$K #[[] +TQAqD ++ +s Bc.eZdZ d ddZdddZy)r%Nc.||_||_||_yr)r+rr)selfr+rrs r__init__z_SchemaGenerator.__init__s ! &(rc i}t|tjrGtj||j}|j }t|tjrG|rTt |drH|jj|jx}r!|jj||d<|St|tjtjfr |St|tjrd|d<|St|tjrd|d<|St|tj tj"frt|tj rdnd|d<|j$|j$|d <|j&|j&|d <|j(|j(|d <|j*|j*|d <|j,|j,|d<|St|tj.rXd|d<|j0|j0|d<|j2|j2|d<|j4|j4|d<|St|tj6tj8tj:frTd|d<d|d<|j0d|j0dzdzz|d<|j2d|j2dzdzz|d<|St|tj<rd|d<|j>durd|d<|St|tj@r/d|d<|j>durd|d<|S|j>durd|d<|St|tjBr d|d<d|d<|St|tjDr d|d<d|d<|St|tjFr d|d<d |d<|St|tjHr d|d<d!|d<|St|tjJrd"|d<t|jLtjs|jO|jL|d#<|j0|j0|d$<|j2|j2|d%<t|tjPtjRfrd|d&<|St|tjTrad"|d<tW|jXx|d%<|d$<|jXr1|jXDcgc]}|jO|c}|d'<d|d#<|St|tjZrd(|d<t|j\x}tj.rZi}|j2|j2|d<|j0|j0|d<|j4|j4|d<|r||d)<t|j^tjs|jO|j^|d*<|j0|j0|d+<|j2|j2|d,<|St|tj`ri}g} d } |jbD]} | } t| tjr'| j } t| tjr't| tjdr(| jfs| jh} | || jj<| jm| | D cgc]} |jO| }} tW|dk\r|joDcic]9\}}||jj|j|j;}}}|jqDcgc]}|jO|c}| |d-d.}|r|jm|||d/<|S|js||StW|d0k(r:|ju\}} |jm|jO| ||d/<|S||d/<|St|tjvrty|jp|d1<|St|tjzrf|j}d2|jj~t|x}r|j}d3|tyd4|jD|d1<|St|tjdrD|j}d2t|jt|x}r|j}d3|g}g}g}|jhT|jm|jh|jm|jh|jmd1|jjgi|jD]}|jO|j }|jr|jm|jnl|jtjurt|jd5|d6<n5|jttttfvr|j|d6<|jm|j|jm||jfrmd7}tt|jD]\}}|jsnd"|d<||d'<tW||z |d%<|jrCtW||d$<|Sd(|d<tt|||d8<||d9<|jrd|d*<|St|tjtjtjfrf|j}d2t|jt|x}r|j}d3|g}g}g}|jD]}|jO|j }|jr|jm|jn6|jtjurt|jd5|d6<|jm|j|jm|t|tjr(d"|d<||d'<tW||d%<tW||d$<|Sd(|d<tt|||d8<||d9<|St|tjr td:t|tjrY|jr0 tj|j|j|}|std;|jd<|Std=|cc}wcc} wcc}}wcc}w#t$rYKwxYw)>z!Converts a Type to a json-schema.r-rgz$refnullrbooleanintegernumberNminimumexclusiveMinimummaximumexclusiveMaximum multipleOfstring maxLength minLengthpatternbase64contentEncodingTz date-timeformattimeFz partial-timedatedurationuuiddecimalarrayr'maxItemsminItems uniqueItems prefixItemsobject propertyNamesadditionalProperties maxProperties minProperties) propertyNamemapping)anyOf discriminatorrr enumtitle descriptionc34K|]}|jywr)value)res rr z-_SchemaGenerator.to_schema..ts#;AGG#;s)str_keysdefaultr propertiesrequiredz-json-schema doesn't support msgpack Ext typeszGenerating JSON schema for custom types requires either: - specifying a `schema_hook` - annotating the type with `Meta(extra_json_schema=...)` type z is not supportedz!json-schema doesn't support type )Yr0r!r7 _merge_jsonextra_json_schemarrOr+getr-rrAnyTypeRawTypeNoneTypeBoolTypeIntType FloatTypegegtlelt multiple_ofStrType max_length min_lengthr BytesType ByteArrayTypeMemoryViewType DateTimeTypetzTimeTypeDateType TimeDeltaTypeUUIDType DecimalTyper8r9rSetType FrozenSetTyper:lenr;r<r=r>r?r(r1 array_like tag_fieldtagappendr'valuesupdatepopitem LiteralTypesortedr6 setdefaultrErarTr5r encode_namer NODEFAULTr default_factorylistdictrn bytearray enumeratereversedforbid_unknown_fieldszipr2r3r4ExtType TypeError CustomTyperNotImplementedError)rxr check_refr rRir=property_namesstructsotherrsubtype real_typerLoptionsrtrur struct_schemardr`rrrr5field field_scheman_trailing_defaultsr@s rrz_SchemaGenerator.to_schemas !#BKK(^^FA,?,?@FABKK( E*}}((//t/!%!2!2!9!9t!9!Dv a"**bjj1 2 p o2;; '#F6Nl k2;; '&F6Nh gBJJ 5 6*4Q *CYF6Ntt$%DDy!tt-.TT)*tt$%DDy!tt-.TT)*}}('(}}|$P O2:: &%F6N||'&'ll{#||'&'ll{#yy$$%IIy!@ BLL"*:*:B> *%F6N(F8 F E2,, -$F6Nakk2::6"&.."=w||'%&\\z"||'%&\\z"!bjj"*:*:;<(,}%r q2<< ($F6N69!,,6G GF:  !3||DELL(Qq):(Q}%"'wf e2;; '%F6Najj0("**=13&&22:2E2EN;/&&22:2E2EN;/##/080@0@N9-!.> *#AHH-F6Nt s2;; '   gquu~~ 6qk!s!!!-5##;QUU#;;F6Nj i2== )   gquu'= >qk!s!!!-5HEF{{& , Q[[) vw/0 ,#~~ejj9 >>OOE$5$56]]",,6.9%--RV.WL+**tT3 .JJ.3.C.C.EL+ U../ l+ ,||&'#.78J.K*'zz")v(.}%%([3F%Fz"**),VF:&f c"*v'+Cv,>'?|$%-z"**5:F12Z YB,,b.>.>@Q@QR S   gquu'= >qk!s!!!-5EFH ,#~~ejj9 >>OOE$5$56]]",,6.9%--RV.WL+ U../ l+ ,!R../!(v(.}%%(]z"%([z"2 /"*v'+Cv,>'?|$%-z"* )2:: &KL L 2== )^^D,<,ww #/w ww)Nr)r+dict[Any, str]r*Optional[Callable[[type], dict[str, Any]]]rstr)T)rmi.Typerboolreturndict[str, Any])rErkrlryrrCrrr%r%s4CG, ) )@) )frr%)rrrrrrrr)r(z Iterable[Any]rrrrrz1tuple[tuple[dict[str, Any], ...], dict[str, Any]])r)zIterable[mi.Type]rdict[Any, mi.Type])r-rrr)rrrr)r*rrr) __future__rrer[collections.abcrtypingrrrrrWr r!r __all__r__annotations__r r r#rHrTrar$r%rCrrrs" $11( )' u' ?C% . .<. .  .h?C% 4 4<4 4 7 4n$N@  ,