MhXxddlZddlZddlmZddlmZmZddddd d d d Zejd Z GddeZ dS)N)JsonSchemaDefinitionException) CodeGenerator enforce_listNoneTypeboolzint, float, Decimalintstrz list, tupledict)nullbooleannumberintegerstringarrayobjectz (?z5CodeGeneratorDraft04.generate_type..Xs($P$PQ%=a%@$P$P$P$P$P$Pr\zUnknown type: {}Nrrr z or isinstance({variable}, bool))variablez&if not isinstance({variable}, ({})){}:{name} must be {}z or rule) r _definitionjoinKeyErrorrr$ _variablelexc)rStypes python_typesrrextras r[r:z"CodeGeneratorDraft04.generate_typeMssT-f566 P99$P$P%$P$P$PPPLL P P P/0B0I0I#0N0NOO O P   e!3!3%9O9O6==t~=VVE VV4K^pt ) u u u u v vr\c n|xjdz c_|j}|d||jdD]}|d|d5|dd 5|||j|jd |d |d d d n #1swxYwY|dd d d n #1swxYwY|d|d5|ddd d d d S#1swxYwYd S)a Means that value have to be valid by any of those definitions. It can also be valid by all of them. .. code-block:: python { 'anyOf': [ {'type': 'number', 'minimum': 10}, {'type': 'number', 'maximum': 5}, ], } Valid values for this definition are 3, 4, 5, 10, 11, ... but not 8 for example. rz"{variable}_any_of_count{count} = 0countrz&if not {variable}_any_of_count{count}:Froptimizetry:rTr|z#{variable}_any_of_count{count} += 1N%except JsonSchemaValueException: passz,{name} cannot be validated by any definitionrkrRrqrmr~rprrr)rSrrs r[r=z$CodeGeneratorDraft04.generate_any_ofsH !!Q&!!) 35AAA#/8 @ @O@X]^^ @ @VVFUV33OO11/4>SWSfx|1}}}FF@FNNNOOOOOOOOOOOOOOO>???  @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ VVz$CodeGeneratorDraft04.generate_one_ofsP !!Q&!!) 35AAA#/8 @ @O@X]^^ @ @VVFUV33OO11/4>SWSfx|1}}}FF@FNNNOOOOOOOOOOOOOOO>???  @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ VV=UV K K s sQG HHEu\cjqH r r r s s s s s s s s s s s s s s s s s ssHC!.;B5) C!5B9 9C!<B9 =C!!C% (C% D--D14D1c|jd}|dur|dddS|durdS|s|dddS|dd 5|||j|jdddn #1swxYwY|d |d 5|ddddddS#1swxYwYdS) aT Means that value have not to be valid by this definition. .. code-block:: python {'not': {'type': 'null'}} Valid values for this definition are 'hello', 42, {} ... but not None. Since draft 06 definition can be boolean. False means nothing, True means everything is invalid. r T{name} must not be thererkFNz-{name} must NOT match a disallowed definitionrrrzelse:)rmrrrqr~rpr)rSnot_definitions r[r?z!CodeGeneratorDraft04.generate_nots)%0 T ! ! HH/eH < < < < < u $ $ F V HHD5H Q Q Q Q Q// c c--ndndNabbb c c c c c c c c c c c c c c c FF: ; ; ; V VHuUUU V V V V V V V V V V V V V V V V V Vs$#"BBBC++C/2C/c~|d5|t|jdtt fst d|d5|dddddn #1swxYwYddddS#1swxYwYdS)Nif isinstance({variable}, str):r!zminLength must be a numberz if {variable}_len < {minLength}:z={name} must be longer than or equal to {minLength} charactersrkrqcreate_variable_with_lengthrwrmr floatrrrrSs r[r@z(CodeGeneratorDraft04.generate_min_lengths^ VV5 6 6 l l  , , . . .d.{;c5\JJ R34PQQQ:;; l lX_jkkk l l l l l l l l l l l l l l l  l l l l l l l l l l l l l l l l l l6A B26B B2B B2!B "B22B69B6c~|d5|t|jdtt fst d|d5|dddddn #1swxYwYddddS#1swxYwYdS)Nrr"zmaxLength must be a numberz if {variable}_len > {maxLength}:z>{name} must be shorter than or equal to {maxLength} charactersrkrrs r[rAz(CodeGeneratorDraft04.generate_max_lengths^ VV5 6 6 m m  , , . . .d.{;c5\JJ R34PQQQ:;; m mY`klll m m m m m m m m m m m m m m m  m m m m m m m m m m m m m m m m m mrc|d5|jd}|dddd}td|}t j||j|<|dt|5| d |d dddn #1swxYwYddddS#1swxYwYdS) Nrr#\z\\"z\"z\\Zz-if not REGEX_PATTERNS[{}].search({variable}):z{name} must match pattern {}rk) rqrmreplace DOLLAR_FINDERsubrecompile_compile_regexpsreprrr)rSr# safe_patternend_of_string_fixed_patterns r[rBz%CodeGeneratorDraft04.generate_patterns VV5 6 6 W W&y1G"??488@@eLLL*7*;*;FG*L*L '-/Z8S-T-TD !' *GgWW W W7IVVV W W W W W W W W W W W W W W W  W W W W W W W W W W W W W W W W W Ws6BC%(C  C% C C%C C%%C),C)c~|jsdS|d5|jd}||jvr|j|}t |t r|||dz|n1|d|5|d|ddddn #1swxYwYn||jvr(|j|}|||dz|n|dkrd g|_ |d d 5|d dddn #1swxYwY|d5|dddddn #1swxYwYn"td |ddddS#1swxYwYdS)z Means that value have to be in specified format. For example date, email or other. .. code-block:: python {'format': 'email'} Valid value for this definition is user@example.com but not @username Nrr$ _re_patternz(if not custom_formats["{}"]({variable}):rjrkregexz import rerFrzre.compile({variable})zexcept Exception:z{name} must be a valid regexzUnknown format: {}) r6rqrmr5rwr _generate_formatrr FORMAT_REGEXS_extra_imports_linesrr$)rSformat_ custom_format format_regexs r[rCz$CodeGeneratorDraft04.generate_formats   F VV5 6 6 Z Z&x0G$... $ 4W = mS11N))'7]3JMZZZZ JGTTNN!4gHMMMNNNNNNNNNNNNNNND...#1': %%gw/F UUUUG##-8M)VVFUV3355FF3444555555555555555VV/00LLHH;(HKKKLLLLLLLLLLLLLLL44H4O4OPW4X4XYYY+ Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z ZsA+F2 B/# F2/B3 3F26B3 7AF2D3' F23D7 7F2:D7 ;F2E7+ F27E; ;F2>E; ?&F22F69F6c|jd|krn||jvrtj||j|<|d|5|d|dddddS#1swxYwYdSdS)Nr$z.if not REGEX_PATTERNS["{}"].match({variable}):rjrk)rmrrrrqrr)rS format_name regexp_nameregexps r[rz%CodeGeneratorDraft04._generate_formats  H % 4 4$"77757Z5G5G%k2H+VV J J,kIII J J J J J J J J J J J J J J J J J J 5 4s A33A7:A7c,|d5t|jdttt jfstd|jddrE|d5| dddddn #1swxYwYnD|d 5| d ddddn #1swxYwYddddS#1swxYwYdS) N1if isinstance({variable}, (int, float, Decimal)):r%zminimum must be a numberexclusiveMinimumFzif {variable} <= {minimum}:z${name} must be bigger than {minimum}rkzif {variable} < {minimum}:z0{name} must be bigger than or equal to {minimum} rqrwrmr rdecimalDecimalrgetrrrs r[rDz%CodeGeneratorDraft04.generate_minimum"s VVG H H a ad.y9C;XYY P34NOOO##$6>> aVV9::UUHHC)HTTTUUUUUUUUUUUUUUUVV899aaHHOV_H```aaaaaaaaaaaaaaa a a a a a a a a a a a a a a a a a aZA2D B, D ,B0 0D 3B0 4D  C1% D 1C5 5D 8C5 9D  D D c,|d5t|jdttt jfstd|jddrE|d5| dddddn #1swxYwYnD|d 5| d ddddn #1swxYwYddddS#1swxYwYdS) Nrr&zmaximum must be a numberexclusiveMaximumFzif {variable} >= {maximum}:z%{name} must be smaller than {maximum}rkzif {variable} > {maximum}:z1{name} must be smaller than or equal to {maximum}rrs r[rEz%CodeGeneratorDraft04.generate_maximum-s VVG H H b bd.y9C;XYY P34NOOO##$6>> bVV9::VVHHD9HUUUVVVVVVVVVVVVVVVVV899bbHHPW`Haaabbbbbbbbbbbbbbb b b b b b b b b b b b b b b b b b brc|d5t|jdttt jfstdt|jdtr|dn|d|d5|dddddn #1swxYwY|d 5|d ddddn #1swxYwYddddS#1swxYwYdS) Nrr'zmultipleOf must be a numberzBquotient = Decimal(repr({variable})) / Decimal(repr({multipleOf}))z$quotient = {variable} / {multipleOf}zif int(quotient) != quotient:z'{name} must be multiple of {multipleOf}rkz-if {variable} / {multipleOf} == float("inf"):zinifinity reached) rqrwrmr rrrrrrrs r[rFz)CodeGeneratorDraft04.generate_multiple_of8s' VVG H H A Ad.|[\\ S34QRRR$*<8%@@ ?[\\\\=>>>788 W WBVVV W W W W W W W W W W W W W W WGHH A A,<@@@ A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A AsZB"D88C D8C D8#C $D8<D  D8 D$ $D8'D$ (D88D<?D<c||d5t|jdtt fst d||d5|dddddn #1swxYwYddddS#1swxYwYdS)Nif {variable}_is_list:r(zminItems must be a numberzif {variable}_len < {minItems}:z-{name} must contain at least {minItems} itemsrk create_variable_is_listrqrwrmr rrrrrrs r[rGz'CodeGeneratorDraft04.generate_min_itemsHso $$&&& VV, - - [ [d.z:S%LII Q34OPPP  , , . . .9:: [ [HzZZZ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [  [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [6A C B." C.B2 2C5B2 6CC  C c||d5t|jdtt fst d||d5|dddddn #1swxYwYddddS#1swxYwYdS)Nrr)zmaxItems must be a numberzif {variable}_len > {maxItems}:z:{name} must contain less than or equal to {maxItems} itemsrkrrs r[rHz'CodeGeneratorDraft04.generate_max_itemsQsp $$&&& VV, - - h hd.z:S%LII Q34OPPP  , , . . .9:: h hU\fggg h h h h h h h h h h h h h h h  h h h h h h h h h h h h h h h h h hrc|jd}|sdS||d5|d||d5|dddddn #1swxYwYddddS#1swxYwYdS)a With Python 3.4 module ``timeit`` recommended this solutions: .. code-block:: python >>> timeit.timeit("len(x) > len(set(x))", "x=range(100)+range(100)", number=100000) 0.5839540958404541 >>> timeit.timeit("len({}.fromkeys(x)) == len(x)", "x=range(100)+range(100)", number=100000) 0.7094449996948242 >>> timeit.timeit("seen = set(); any(i in seen or seen.add(i) for i in x)", "x=range(100)+range(100)", number=100000) 2.0819358825683594 >>> timeit.timeit("np.unique(x).size == len(x)", "x=range(100)+range(100); import numpy as np", number=100000) 2.1439831256866455 r*Nrzdef fn(var): return frozenset(dict((k, fn(v)) for k, v in var.items()).items()) if hasattr(var, "items") else tuple(fn(v) for v in var) if isinstance(var, (dict, list)) else str(var) if isinstance(var, bool) else varzNif {variable}_len > len(set(fn({variable}_x) for {variable}_x in {variable})):z {name} must contain unique itemsrk)rmrrqrrr)rSunique_definitions r[rIz*CodeGeneratorDraft04.generate_unique_itemsZs{!,];   F $$&&& VV, - - Q Q FF4 5 5 5  , , . . .hii Q Q;-PPP Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Qs5?B6:B B6B" "B6%B" &B66B:=B:c |jd}|durdS||d5||durF|d5|dddddn #1swxYwYnt |t rt|D]\}}|d |5|d |||d |j |d |j |dddn #1swxYwY|j rBt |tr-d |vr)|dt|d d|jvr|jddurT|d t|5|dddddn #1swxYwYnF|dt|5||jdd |j d |j |j }|dkr|ddddn #1swxYwYn|r|d5||d |j d |j |j }|dkr|ddddn #1swxYwYddddS#1swxYwYdS)a Means array is valid only when all items are valid by this definition. .. code-block:: python { 'items': [ {'type': 'integer'}, {'type': 'string'}, ], } Valid arrays are those with integers or strings, nothing else. Since draft 06 definition can be also boolean. True means nothing, False means everything is invalid. r+TNrFzif {variable}:rrkzif {variable}_len > {}:z!{variable}__{0} = {variable}[{0}]{}__{}z{}[{}]defaultzelse: {variable}.append({})additionalItemsz({name} must contain only specified itemszFfor {variable}_x, {variable}_item in enumerate({variable}[{0}:], {0}):z{}_itemz {}[{{{}_x}}]rpassz;for {variable}_x, {variable}_item in enumerate({variable}):)rmrrqrrrrwrx enumerater~r$rprr7r rlen)rSitems_definitionidxitem_definitionrs r[rJz#CodeGeneratorDraft04.generate_items{s$ +G4 t # # F $$&&& VV, - -' +' +  , , . . .5((VV,--GGHH7gHFFFGGGGGGGGGGGGGGG,d33" +,56F,G,G ` `(C 93??BCHHH55+$OODNC@@$OOD,?EE(`Z-N-N`S\`oSoSo^>^___$(888'(9:eCC!VV$=sCS?T?TUU__ HH%OV]H^^^_______________"VV$lnqsCoDoDEE//$($A$A $ 01B C ) 0 0 @ @ . 5 5d6I4> Z Z%%E %zz $v///////////////$+ ]^^++ $ = =,%,,T^<<*11$2Et~VV!! !A:: FF6NNN+++++++++++++++A' +' +' +' +' +' +' +' +' +' +' +' +' +' +' +' +' +' +s.M+B MB MB AMA E< ME ME B MG=1 M=H MH (M-A4J-! M-J1 1M4J1 5MA)M9 MM M M MM!$M!c||d5t|jdtt fst d||d5|dddddn #1swxYwYddddS#1swxYwYdS)Nif {variable}_is_dict:r,zminProperties must be a numberz$if {variable}_len < {minProperties}:z7{name} must contain at least {minProperties} propertiesrk create_variable_is_dictrqrwrmr rrrrrrs r[rKz,CodeGeneratorDraft04.generate_min_propertiessp $$&&& VV, - - j jd.?#uNN V34TUUU  , , . . .>?? j jRYhiii j j j j j j j j j j j j j j j  j j j j j j j j j j j j j j j j j jrc||d5t|jdtt fst d||d5|dddddn #1swxYwYddddS#1swxYwYdS)Nrr-zmaxProperties must be a numberz$if {variable}_len > {maxProperties}:zD{name} must contain less than or equal to {maxProperties} propertiesrkrrs r[rLz,CodeGeneratorDraft04.generate_max_propertiessp $$&&& VV, - - w wd.?#uNN V34TUUU  , , . . .>?? w w_fuvvv w w w w w w w w w w w w w w w  w w w w w w w w w w w w w w w w w wrc Nd5tjdtt fst dtjdttjdkrt dj ddsCfdjdD}|r(t d j |d d 5d } d jdd| dddn #1swxYwYddddS#1swxYwYdS)Nrr.zrequired must be an arrayz%required must contain unique elementsr2Tc g|]UjdivtfdjdiDSVS)r0c3BK|]}tj|VdSrc)rsearch)rerprops r[rgzDCodeGeneratorDraft04.generate_required...s/#v#vuBIeT$:$:#v#v#v#v#v#vr\r1)rmrany)rerrSs @r[ z:CodeGeneratorDraft04.generate_required..s   D$4$8$8r$J$JJJ ##v#v#v#vHXH\H\]prtHuHu#v#v#v v vKKJJr\z){}: items {} are required but not allowedz>{variable}__missing_keys = set({required}) - {variable}.keys()zif {variable}__missing_keys:z5str(sorted({variable}__missing_keys)) + " properties"z{name} must contain )rlr)rrqrwrmrxryrrsetrr$rprrrz)rS not_possiblers` r[rMz&CodeGeneratorDraft04.generate_requiredsa $$&&& VV, - -  d.z:T5MJJ Q34OPPP4#J/00CDEE J     $ 0 <     J78c8j8jkoky|H9I9IJJJ FFS T T T677  Q/8H8T1U1U\fv}~~~               !                  s6DF<:F6 FF F F FF!Fc T||d5||jdD]\}}t jdd|}t|ttfs(td |j ||d| |5|d| ||d|| |||d |j |d |j| |d d d d n #1swxYwY|jrVt|trAd|vr=|d| |t#|d d d d d S#1swxYwYd S)a Means object with defined keys. .. code-block:: python { 'properties': { 'key': {'type': 'number'}, }, } Valid object is containing key called 'key' and value any number. rr0z($[^a-zA-Z]|[^a-zA-Z0-9])rhz{}[{}] must be objectzif "{}" in {variable}_keys:z{variable}_keys.remove("{}")z#{variable}__{0} = {variable}["{1}"]rz{}.{}Tr|Nrzelse: {variable}["{}"] = {})rrqcreate_variable_keysrmr+rrrwr rrr$rprzr~rr7r)rSkeyprop_definitionkey_names r[rOz(CodeGeneratorDraft04.generate_propertiess $$&&& VV, - - i i  % % ' ' '(,(8(F(L(L(N(N i i$_6">CHH!/D$<@@r78O8V8VW[Wego8p8pqqqVV9466#;;GGFF9466#;;GGGFF@(DFFSVKKXXX11' AAt':DFF3KKHH(, 2$iOT)J)Jiy\kOkOkFF8$&&++tO\eLfGgGghhh i i i i i i i i i i i i i i i i i i is8B=H'B2F% H%F) )H,F) -A"HH!$H!c r||d5||jdD]!\}}t j||j|<"|d5|jdD]\}}|dt|5|d5|ddddn #1swxYwY| |d |j d |j |j d dddn #1swxYwY dddn #1swxYwYddddS#1swxYwYdS) a4 Means object with defined keys as patterns. .. code-block:: python { 'patternProperties': { '^x': {'type': 'number'}, }, } Valid object is containing key starting with a 'x' and value any number. rr1z9for {variable}_key, {variable}_val in {variable}.items():z-if REGEX_PATTERNS[{}].search({variable}_key):z%if {variable}_key in {variable}_keys:z&{variable}_keys.remove({variable}_key)Nz{}_val {}.{{{}_key}}Tr|) rrqrrmr+rrrrr~r$rpr)rSr#rTs r[rPz0CodeGeneratorDraft04.generate_pattern_propertiess $$&&& VV, - -    % % ' ' ''+'78K'L'R'R'T'T E E#13G1D1D%g..STT  +/+;PPRSSS55/&--dn==+2243FWWVV122yyHHZawHxxxyyyyyyyyyyyyyyy! y y y y y y y y y y y y y y y y y ys-F/$AF/5E A#D;/ E;D? ?ED? E F/E F/E F/3F F/F F/F F//F36F3c \||d5d}|jdD]/\}}|gks|durd}|d||5|dur|d|dnt |tr|D]}|d||5|d ||||dd d d n #1swxYwYn#|||j |j d d d d n #1swxYwY1|r|d d d d d S#1swxYwYd S) a Means when object has property, it needs to have also other property. .. code-block:: python { 'dependencies': { 'bar': ['foo'], }, } Valid object is containing only foo, both bar and foo or none of them, but not object with only bar. Since draft 06 definition can be boolean or empty array. True and empty array means nothing, False means that key cannot be there at all. rTr/Fzif "{}" in {variable}:z{} in {name} must not be thererkzif "{}" not in {variable}:z#{name} missing dependency {} for {}Nr|r) rrqrmr+rzrrrwrxr~rpr)rSis_emptyrvaluesvalues r[rNz*CodeGeneratorDraft04.generate_dependenciesGs$ $$&&& VV, - -  H#/?EEGG y y VR<<6T>> VV4dffSkkBByy!A3^\\\\#FD11y%+AAE!%(DdffUmm!T!TAA $)NPTPVPVW\P]P]_c_e_efi_j_jq!A!A!AAAAAAAAAAAAAAAAA55fdndNasw5xxxyyyyyyyyyyyyyyy v!                  s\AF!A E0(AD4 ( E04D88E0;D8<(E0$ F!0E4 4F!7E4 8F!!F%(F%)!__name__ __module__ __qualname__rr4propertyr_r:r;r<r=r>r?r@rArBrCrrDrErFrGrHrIrJrKrLrMrOrPrQrN __classcell__)rZs@r[rrsWDBNfy' M-1"$\`vz & & & & & &DX KKK,LLL vvv&SSS<sss>VVV6lllmmmWWW!Z!Z!ZHJJJ a a a b b bAAA [[[hhhQQQB>+>+>+@jjjwww,iiiB>"y"y"yH#######r\r) rr exceptionsr generatorrrrdrrrr\r[rs 55555522222222 #  <(( T T T T T =T T T T T r\