jUddlmZddlZddlZddlZddlmZmZmZddl m Z m Z ddl m Z mZmZddlmZedZdZd ed <Gd d eZd$dZd%dZd%dZd&dZd'dZd(dZd)d!ZGd"d#ZdS)*) annotationsN)AnyLiteralcast)array_schema_configobject_schema_config) MISSING_VALUEJSONReturnTypeMissingValueType)match_pattern_properties)standardsalvageztuple[SchemaRepairMode, ...]SUPPORTED_SCHEMA_REPAIR_MODESceZdZdZdS)SchemaDefinitionErrorz6Raised when schema metadata is invalid or unsupported.N)__name__ __module__ __qualname____doc__GC:\Projects\DSCons\.venv\Lib\site-packages\json_repair\schema_repair.pyrrs@@@@rrmode str | NonereturnSchemaRepairModec|dS|dkrdS|dkrdSdt}td|d)Nrr, z#schema_repair_mode must be one of: .)joinr ValueError)rexpecteds rnormalize_schema_repair_moder$sX |z zz yyyy677H F8FFF G GGrrcp tjdS#t$r}td|d}~wwxYw)N jsonschemaz6jsonschema is required when using schema-aware repair. importlib import_module ImportErrorr"excs r_require_jsonschemar-sN\&|444 \\\QRRX[[\ 505cp tjdS#t$r}td|d}~wwxYw)Npydanticz.pydantic is required when using schema models.r'r+s r_require_pydanticr1&sNT&z222 TTTIJJPSSTr.nodect|trd|D}|d}t|trU|dd||d<|dd}|durd|d<nt|tr||d<|St|tr d|DS|S)Nc4i|]\}}|t|Sr#_prepare_schema_for_validation_node.0keyvalues r z7_prepare_schema_for_validation_node../s'eee*#uc>uEEeeeritems prefixItemsadditionalItemsFc,g|]}t|Srr5r8items r z7_prepare_schema_for_validation_node..;s!KKKd3D99KKKr) isinstancedictr<getlistpop)r2 normalizedr<additional_itemss rr6r6-s$ eeX\XbXbXdXdeee w'' eT " " 7 NN7D ) ) )(-J} %)~~.?FF 5((&+ 7##,d33 7&6 7#$LKKdKKKK Krpathstr type[Any]cd|vrtd|dd\}}tj|}|j|}|t |tstd|d|d|S)N:z4Schema model must be in the form 'module:ClassName'.rzSchema model 'z' not found in module 'z'.)r"splitr(r)__dict__rErCtype)rJ module_name class_namemodulemodels rload_schema_modelrV?s $OPPP"jja00K  $[ 1 1F!?..z::E }Jud33}\*\\[\\\]]] Lrr:objectr c|tust|trdSt|trTi}|D];\}}t|t st dt|||<<|St|tr d|DS|(t|t tttfr|St d)NzObject keys must be strings.c,g|]}t|Srnormalize_missing_valuesr@s rrBz,normalize_missing_values..Us!AAA4(..AAArzValue is not JSON compatible.) r rCr rDr<rKr"r\rFintfloatbool)r:rHr9rAs rr\r\Js E3C!D!Dr%02  = =ICc3'' A !?@@@6t<&>&@&@ 6> * * ?$// bAAJj$// 7 ,6 L)%288:: ? ? e$$&&","7"7b"A"A!/4887&(O'6Jt$//(4161F1F1H1HOI..16OI.. _ ` ``rceZdZ d=d>d Zd?d Zd@dZdAdZdBdZdCdZdDdZ dEdZ dEdZ dFdZ dGdZ dHdZdId"ZdJd%ZdKd&ZdLd+ZdMd,ZdMd-ZdNd0ZdOd3ZdPd4ZdQd5ZdRd6ZdSd8ZdTd:ZdUd;Zd?Q"R"RGI+/rtextrJcR|j|j||ddSdS)N)rcontext)rxappend)rrrJs r_logzSchemaRepairer._logs4 8  HOOTd;; < < < < < rdict[str, Any]rc,t|}|j|}||d|ur|dS||}|}||jur|n||}||f|j|<|S)Nrr)r`)idrrE_prepare_schema_for_validation_get_root_validatorr~evolve)rr` cache_keycached_validatorprepared_schemaroot_validator validators r_get_validatorzSchemaRepairer._get_validatorsvJJ 044Y??  ',i(rc|j|jS||j}t}|j|}|||_|jSr})rrr~r- validators validator_for)rprepared_root_schemar& validator_clss rrz"SchemaRepairer._get_root_validatorsi   +' '#BB4CSTT(** "-;;>4 U??5''// I&&u--...rc,||}|durdS|durtdt}||} ||dS#|jj$r}t|j|d}~wwxYw)NTF!Schema does not allow any values.)rr"r-rvalidate exceptionsValidationErrormessage)rr:r`r&rr,s rrzSchemaRepairer.validates$$V,, T>> F U??@AA A(** ''//  3   u % % % % %$4 3 3 3S[))s 2 3sA**B9BB object | NonecJ|dSt|tr|St|tstd|D]&}t|tstd't d|}t }d|vr|d}t|tstdt|}||vrtd|||| |}t|tr|S|}d|v|S)NTSchema must be an object.zSchema keys must be strings.rz$refz$ref must be a string.zCircular $ref detected: ) rCr_rDrrKrsetradd _resolve_ref)rr`r9rqseen_schema_idsref schema_idresolveds rrzSchemaRepairer.resolve_schemasS >4 fd # # M&$'' E'(CDD D L LCc3'' L+,JKKK L+V44 $'EE ##f%Cc3'' F+,DEEE;IO+++,Ls,L,LMMM    * * *((--H(D)) "K ##rdict[str, Any] | bool | Nonec|ttsdSd}|dkrdSt|trd|vrdSt fddDS)NFrQrWTc3 K|]}|vV dSr}r)r8r9r`s r z2SchemaRepairer.is_object_schema..s'ttS3&=ttttttr)ripatternPropertiesadditionalPropertiesrequired)rrCrDrErFanyrr` schema_types ` ris_object_schemazSchemaRepairer.is_object_schemas$$V,,&$'' 5jj(( ( " "4 k4 ( ( X-D-D4tttt,sttttttrc||}t|tsdS|d}|dkrdSt|trd|vrdSd|vS)NFrQarrayTr<)rrCrDrErFrs ris_array_schemazSchemaRepairer.is_array_schemasy$$V,,&$'' 5jj(( ' ! !4 k4 ( ( W -C-C4&  rrc|d}t|tr||kSt|tr||vS|dkr||S||S)NrQrW)rErCrKrFrr)rr`r declared_types r_allows_schema_typez"SchemaRepairer._allows_schema_types 6** mS ) ) 0 K/ / mT * * 0-/ / ( " "((00 0##F+++rc\||do||d S)NrWr)r)rr`s r_can_salvage_list_as_objectz*SchemaRepairer._can_salvage_list_as_objects2''99k$BZBZ[acjBkBk>kkrc ||}|durt|S|durtd|st|S|tur|||Sd|vr\|d}|st|S|||d|}|ddD]}||||}|Sd|vr|||d|Sd |vr|||d |S|d }|/||rd }n| |rd }t|tr| ||||S|d kr| |||}nZ|d kr||||}n>+E22 2 U??@AA A 3+E22 2 M ! !%%fd33 3 f  J 7/666(( 1 tDDH'^ H H ,,Xy$GGO f  %%eVG_dCC C f  %%eVG_dCC C 6**  $$V,, ( ( %%f-- ( ' mT * * O**5-NN N H $ $**5&$??HH g % %))%>>HH  s + + 7**5-FFHH/66H%%h===rschemaslist[dict[str, Any] | bool]c$d}|D]^} |tj|||}||||cS#t$r }|}Yd}~Wd}~wwxYw|rt t ||t d)NzNo schema matched the value.)rcopydeepcopyrr"rK)rr:rrJ last_errorr candidater,s rrzSchemaRepairer._repair_union's'+   ! !I ! --dmE.B.BItTT  i333!    ! ! !  !  >S__--: =7888s?A  A!AA!types list[str]cdd}|D]~}i|d|i} |tj||||}||||}||||cS#t $r } | }Yd} ~ wd} ~ wwxYw|rt t ||t d)NrQz!No schema type matched the value.)_repair_by_typerrrrr"rK) rr:rr`rJrr branch_schemarr,s rrz!SchemaRepairer._repair_type_union5s(,  ! !K;v;v{;;M ! 00u1E1E{TZ\`aa  229mTRR  i777!    ! ! !  !  >S__--: =<===sAA++ B5A<<Bc|dkr||||S|dkr||||S||||S)NrrW)rrr)rr:rr`rJs rrzSchemaRepairer._repair_by_typeLs` ' ! !%%eVT:: : ( " "&&ufd;; ;""5+t<<.repair_or_drop}s| #T..xiPPPP(    # # ## F RRR"{{  #s-A  A [])startTc34K|]}t|VdSr}r[r@s rrz/SchemaRepairer._repair_array..s,-^-^QU.Ft.L.L-^-^-^-^-^-^rz.Dropped extra array item not covered by schemaminItems Array at z does not meet minItems.)rrrrrrKrr)rrFrCrr\rzr items_schema enumeratelenrrIrDextendrEr")rr:r`rJr< schema_configrrepaired_itemsidxrr keep_itemrepaired_valuetailoffsetrA_item min_itemsrs` @rrzSchemaRepairer._repair_arraymsw00    < E    eT " " 6*/EE II> E E E-e445E.); +F33  # # # # # # #  % 1-4d;; '79(1-2L(M(M>>$Cc%jj((#' 0 0# 0 0 0I0>uSz;Xa0b0b-I~ >&--n===u::M$> ? ??? ]%?!@!@!B!BCD!-"@$GG m,5d#mF`BaBa,b,b,bFFLFD+/(;(;&(;(;(;I8Ft]Mkmv8w8w5I~(F . 5 5n E E E F '74??=CaCi&---^-^Y]-^-^-^^^^^-6t3}GaCbCb-c-c-cmmMFE II&V[_XkXkbhXkXkXkllll&!#!*5!1!1>>IC#' 0 0# 0 0 0I0>t]E_aj0k0k-I~ >&--n===&JJz**  SZZ)%;%;GGGGHH H rc|jdkrttr||rn|||}||nR|dkrLt dkr9tdt rd|d||t |ddtt s(td|d tj d t|}|jdkr|j rzt }|j D]a}||vr|j|}|$|d |} ||| \} } | r| ||<|d | b|fd |j D} | r(td |dd| i} |jD]\}}|d |} |vr!|||| | |<1t|t rCd|vr?||j vr6||d| d| |<|d| D]$\}}||jvr|d |} g}g}|jrt-|j|\}}|D]}|d|d| |rG|||d| }|ddD]}|||| }|| |<t|jt r.||t1d|j| | |<|jdus|jt3|| |<|d| &|d}|&t | |krtd|d| S)Nr$rrz:Unwrapped single-item root array to object while salvagingz/Unwrapped JSON string to object to match schemaz8Repaired malformed JSON string to object to match schemazExpected object at z, got r z0Filled missing required property while salvagingcg|]}|v| Srrr7s rrBz1SchemaRepairer._repair_object..s$VVVCSPUEUEUCEUEUEUrzMissing required properties at z: rrjz+Inserted default value for missing propertyz-Skipped unsupported patternProperties regex ''rTz,Dropped extra property not covered by schema minProperties Object at z does not meet minProperties.)rzrCrFr_map_list_to_objectrrDrrr"rQrr rrirE"_fill_missing_required_for_salvager!r<r_copy_json_valuepattern_propertiesr additional_propertiesrr\)rr:r`rJmappedrvalue_with_salvage_fillsr9 prop_schemakey_pathfilled filled_valuemissing_requiredr raw_valuematchedunsupported_patternspatternrmin_propertiess ` rrzSchemaRepairer._repair_objects  #y 0 05$'' 10088 1--eVTBBF!UqZa$5O5Oa VX\]]]00    = F   %&& XV4VVtE{{?SVVVWW W,V44  "i / /M4J /'+E{{ $$- \ \222+6::3?? &"??S??'+'N'N{\d'e'e$ \4@,S1IIPRZ[[[,EVVVV=+AVVV  fdtddtyyQaGbGbddee e.0 - 8 > > @ @ S S CHe|| $ 1 1%*k8 T T K.. S9 3K3KPS[h[qPqPq $ 5 5k)6LhXa b b  GRRR#kkmm P PNCm...H!#G.0 / p0HIikn0o0o--/ ` ` T'TTTV^____ !%!2!29gaj(!S!S#*122;^^K%)%6%6~{T\%]%]NN . -=tDD  $ 1 1)=+NOO!! 2d::m>a>i 8 C C  IIDh O O O OO44  %#h--.*H*HM$MMMNN Nr list[Any] dict[str, JSONReturnType] | Nonec`|d}t|tr|sdSi}|D].\}}t|tst d|||</t |}t|t|krdSi} t|D]O\} }|d|} | || ||| | |<4#t $rt$rYdSwxYw| d|| S)Nriz-Schema object property names must be strings.r z/Mapped array to object by schema property order) rErCrDr<rKrrFkeysrrrr"r) rr:r`rJrityped_propertiesr9rrrrrs rr z"SchemaRepairer._map_list_to_objectsnZZ -- *d++ : 4+- * 0 0 2 2 0 0 Cc3'' ]+,[\\\$/ S ! !$))++,, u::T " "4,.!$  HCH "//c > >(($HH H V   .K C !AD!A!A!ABBB IIBD I I I((QvFF F    II94 @ @ @(( ):D)LL L 6** mT * * G,   --.M.M.M.MtTTTTT!HEdEEEFF F  $$V,, ( ( %%f-- ( ' H $ $ II> E E E2 1 1 1 II3T : : :1 I % % II7 > > >5 G # # :..I Z !XT!X!Xi!X!X!XYYY II=t D D DI H $ $#ZZ88N e !cd!c!c~!c!c!cddd II> E E EI F " " II6 = = =4A$AAABBBs2D DDc|dkrt|tr|St|ttfr:t|ts%|d|t|St d|d|dkrt|trt d|dt|tr|St|trL|r%|d|t|St d|dt|tr t|}n#t $rd}YnwxYw||d||S t|}n&#t $r}t d|d|d}~wwxYw|st d|d|d|t|St d|d|d krt|trt d |dt|ttfr|St|trO t|}n&#t $r}t d |d|d}~wwxYw|d ||St d |d|d krt|tr|St|trL|}|d vr|d|dS|dvr|d|dSt|ttfr>t|ts)|dvr%|d|t |St d|d|dkr|dSt d|dtd|d|d)Nr%zCoerced number to stringzExpected string at r r&zExpected integer at zCoerced number to integerzCoerced string to integerr'zExpected number at zCoerced string to numberr()trueyesyon1zCoerced string to booleanT)falsenonoffrfF)rrzCoerced number to booleanzExpected boolean at r)zExpected null at zUnsupported schema type z at ) rCrKr]r^r_rr" is_integerlowerr) rr:rrJ int_valuenumr, float_valuelowereds rrzSchemaRepairer._coerce_scalarws ( " "%%%  %#u.. "z%7N7N " 4d;;;5zz!:4:::;; ; ) # #%&& A !?!?!?!?@@@%%%  %'' A##%%&II94@@@u::% !?!?!?!?@@@%%% % #E II!%%% $III%(II94@@@$$N,,CC!NNN$%CD%C%C%CDD#MN~~''E$%CD%C%C%CDDD 5t<<<3xx;D;;;<< < ( " "%&& @ !>t!>!>!>???%#u..  %%% #M"',,KK!MMM$%B4%B%B%BCCLM 4d;;;"":4:::;; ; ) # #%&&  %%% !++--===II94@@@4>>>II94@@@ 5%#u.. #z%7N7N #SX\bSbSb 5t<<<E{{";D;;;<< < & }t888899 9#$W{$W$WPT$W$W$WXXXsBE E#"E#F F4F//F48J J+J&&J+cd|vr||dkrtd|dd|vr||dvrtd|d|S)Nr z Value at z does not match const.r!z does not match enum.)r")rr:r`rJs rrz SchemaRepairer._apply_enum_constsm f  &/!9!9EEEEFF F V  VF^ ; ;DDDDEE E rrc|dstd||dd}|j}|D]_}|dddd}t |tr||vrtd|||}`t |tr|S|durdS|d urd Std|) Nz#/zUnsupported $ref: /z~1z~0~zUnresolvable $ref: TF) startswithrlstriprOr~replacerCrD)rrpartscurrentpart resolved_parts rrzSchemaRepairer._resolve_refs~~d## D'(BS(B(BCC C 4  &&s++' - -D LLs33;;D#FFMgt,, I W0L0L+,G#,G,GHHHm,GG gt $ $ N d??4 e  5#$?#$?$?@@@rlabelc.|(t|ttttfr|St|t rfdt |DSt|tryi}|D]`\}}t|ts't dd |d|||<a|St dd)Nc Pg|]"\}}|d|d#S)rr)r )r8rrArErJrs rrBz3SchemaRepairer._copy_json_value..sAkkkYSRVD))$40@0@#0@0@0@%HHkkkrz value at z contains a non-string key.r z is not JSON compatible.) rCrKr]r^r_rFrrDr<r" capitalizer )rr:rJrEcopiedr9rAs` `` rr zSchemaRepairer._copy_json_values, =JusC.EFF=L eT " " lkkkkkkZcdiZjZjkkk k eT " " 02F"[[]] R R T!#s++i$(8(8(:(:%g%gd%g%g%ghhh"33DT//C//5QQs ME,,..XX$XXXYYYrclt|}t|tstd|S)Nr)r6rCrDr")rr`rHs rrz-SchemaRepairer._prepare_schema_for_validations68@@ *d++ :899 9rN)r)r`rarxryrzrKrr{)rrKrJrKrr{)r`rrrrr)r:r r`rarr_)r:r r`rarr{)r`rrra)r`rrr_)r`rrrKrr_)r`rrr_)r:rr`rrJrKrr )r:rrrrJrKrr ) r:rrrr`rrJrKrr ) r:rrrKr`rrJrKrr ) r:rrrrJrKrrKrrKrr)r:rr`rrJrKrr )r:rr`rrJrKrr)r`rWrJrKrr)r`rrJrKrr )r:rrrKrJrKrr )r:r r`rrJrKrr )rrKrra)r:rrJrKrErKrr )r`rWrr)rrrrrrrrrrrrrrrrrrrrrr r rrrrr rrrrrwrw|sK #- 0 0 0 0 0====    $$$$//// 3 3 3 34 u u u u ! ! ! ! , , , ,llll.>.>.>.>` 9 9 9 9>>>>.====4>>>>@UUUUn<68C8C8C8CtHYHYHYHYTAAAA$ Z Z Z Zrrw)rrrrrK)r2rrr)rJrKrrL)r:rWrr )r`rrra) __future__rrr(rtypingrrr parser_schemarr utils.constantsr r r utils.pattern_propertiesr rr__annotations__r"rr$r-r1r6rVr\rurwrrrrRs"""""""  %%%%%%%%%%DDDDDDDDLLLLLLLLLL>>>>>>01>UUUUUAAAAAJAAAHHHH\\\\TTTT$6666"aaaaBp p p p p p p p p p r