od}sddlZddlZddlmZddlmZmZddddd d d d Zejd Z GddeZ dS)N)JsonSchemaDefinitionException) CodeGenerator enforce_listNoneTypeboolz int, floatintstrz list, tupledict)nullbooleannumberintegerstringarrayobjectz (?z5CodeGeneratorDraft04.generate_type..Ws($P$PQ%=a%@$P$P$P$P$P$PrYzUnknown type: {}Nrrr z or isinstance({variable}, bool))variablez&if not isinstance({variable}, ({})){}:{name} must be {}z or rule) r _definitionjoinKeyErrorrr$ _variablelexc)rRtypes python_typesrnextras rXr9z"CodeGeneratorDraft04.generate_typeLs|T-f566 P99$P$P%$P$P$PPPLL P P P/0B0I0I#0N0NOO O P   We!3 W%9O W6==t~=VVE VV4K^pt ) u u u u v vrYc 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:rTrxz#{variable}_any_of_count{count} += 1N%except JsonSchemaValueException: passz,{name} cannot be validated by any definitionrgrQrmrirzrlr{rn)rRrr|s rXr<z$CodeGeneratorDraft04.generate_any_ofsH !!Q&!!) 35AAA#/8 @ @O@X]^^ @ @VVFUV33OO11/4>SWSfx|1}}}FF@FNNNOOOOOOOOOOOOOOO>???  @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ VVSWSfx|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--D14D1cX|jd}|dur|dddS|durdS|sM|d|j5|ddddddS#1swxYwYdS|d d 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 therergFNzif {}:z-{name} must NOT match a disallowed definitionrrrzelse:)rirnrmrlrzr{)rRnot_definitions rXr>z!CodeGeneratorDraft04.generate_notsJ)%0 T ! V HH/eH < < < < < u $ V F V$.11 V VHuUUU V V V V V V V V V V V V V V V V V V// 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 Vs6A33A7:A7"CC  C :DD#&D#cp|d5|t|jdtst 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} charactersrgrmcreate_variable_with_lengthrsrir rrnrRs rXr?z(CodeGeneratorDraft04.generate_min_lengthsZ VV5 6 6 l l  , , . . .d.{;SAA 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 l6AB+/B B+B B+B B++B/2B/cp|d5|t|jdtst 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} charactersrgrrs rXr@z(CodeGeneratorDraft04.generate_max_lengthsZ VV5 6 6 m m  , , . . .d.{;SAA 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 {}rg) rmrireplace DOLLAR_FINDERsubrecompile_compile_regexpsreprrn)rRr# safe_patternend_of_string_fixed_patterns rXrAz%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\|d5|jd}||jvr|j|}t|tr|||dz|n)|d|5|d|ddddn #1swxYwYn||jvr(|j|}|||dz|n|dkr|d d 5|d dddn #1swxYwY|d 5|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 rr$ _re_patternz(if not custom_formats["{}"]({variable}):rfrgNregexrFrzre.compile({variable})zexcept Exception:z{name} must be a valid regexzUnknown format: {}) rmrir5rsr _generate_formatrn FORMAT_REGEXSrr$)rRformat_ custom_format format_regexs rXrBz$CodeGeneratorDraft04.generate_formatsVV5 6 6 Z Z&x0G$.. Z $ 4W = mS11N))'7]3JMZZZZ JGTTNN!4gHMMMNNNNNNNNNNNNNNND.. Z#1': %%gw/F UUUUG# ZVVFUV3355FF3444555555555555555VV/00LLHH;(HKKKLLLLLLLLLLLLLLL44H4O4OPW4X4XYYY) Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z ZsA+F!B& F!&B* *F!-B* .AF!D" F!"D& &F!)D& *F!E& F!&E* *F!-E* .&F!!F%(F%c|jd|krn||jvrtj||j|<|d|5|d|dddddS#1swxYwYdSdS)Nr$z.if not REGEX_PATTERNS["{}"].match({variable}):rfrg)rirrrrmrn)rR format_name regexp_nameregexps rXrz%CodeGeneratorDraft04._generate_formats  H % 4 J$"77 H57Z5G5G%k2H+VV J J,kIII J J J J J J J J J J J J J J J J J J J Js A33A7:A7c|d5t|jdttfst d|jddrE|d5|dddddn #1swxYwYnD|d 5|d ddddn #1swxYwYddddS#1swxYwYdS) N(if isinstance({variable}, (int, float)):r%zminimum must be a numberexclusiveMinimumFzif {variable} <= {minimum}:z${name} must be bigger than {minimum}rgzif {variable} < {minimum}:z0{name} must be bigger than or equal to {minimum}rmrsrir floatrgetrnrs rXrCz%CodeGeneratorDraft04.generate_minimums VV> ? ? a ad.y9C<HH P34NOOO##$6>> aVV9::UUHHC)HTTTUUUUUUUUUUUUUUUVV899aaHHOV_H```aaaaaaaaaaaaaaa a a a a a a a a a a a a a a a a a aZA'C>=B! C>!B% %C>(B% )C>C& C>&C* *C>-C* .C>>DDc|d5t|jdttfst d|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}rgzif {variable} > {maximum}:z1{name} must be smaller than or equal to {maximum}rrs rXrDz%CodeGeneratorDraft04.generate_maximum*s VV> ? ? b bd.y9C<HH P34NOOO##$6>> bVV9::VVHHD9HUUUVVVVVVVVVVVVVVVVV899bbHHPW`Haaabbbbbbbbbbbbbbb b b b b b b b b b b b b b b b b b brcH|d5t|jdttfst dt|jdtrD|jdtj |j d<|dn|d|d5| d d dddn #1swxYwYddddS#1swxYwYdS) Nrr'zmultipleOf must be a numberzfrom decimal import DecimalDecimalzBquotient = Decimal(repr({variable})) / Decimal(repr({multipleOf}))z$quotient = {variable} / {multipleOf}zif int(quotient) != quotient:z'{name} must be multiple of {multipleOf}rg) rmrsrir rr_extra_imports_linesappenddecimalr_extra_imports_objectsrnrs rXrEz)CodeGeneratorDraft04.generate_multiple_of5s VV> ? ? W Wd.|>>788 W WBVVV 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 Ws6CDC?3 D?D DD DDDc||d5t|jdtst 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} itemsrgcreate_variable_is_listrmrsrir rrrnrs rXrFz'CodeGeneratorDraft04.generate_min_itemsDsk $$&&& VV, - - [ [d.z:C@@ Q34OPPP  , , . . .9:: [ [HzZZZ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [  [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [6AB?B' B?'B+ +B?.B+ /B??CCc||d5t|jdtst d||d5|dddddn #1swxYwYddddS#1swxYwYdS)Nrr)zmaxItems must be a numberzif {variable}_len > {maxItems}:z:{name} must contain less than or equal to {maxItems} itemsrgrrs rXrGz'CodeGeneratorDraft04.generate_max_itemsMsl $$&&& VV, - - h hd.z:C@@ 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 itemsrg)rirrmrrn)rRunique_definitions rXrHz*CodeGeneratorDraft04.generate_unique_itemsVs{!,];   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}:rrgzif {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}):)rirrmrrnrsrt enumeraterzr$rlr{r6r rlen)rRitems_definitionidxitem_definitionrs rXrIz#CodeGeneratorDraft04.generate_itemsws+$ +G4 t #  F $$&&& VV, - -' +' +  , , . . .5(% +VV,--GGHH7gHFFFGGGGGGGGGGGGGGG,d33" +,56F,G,G ` `(C 93??BCHHH55+$OODNC@@$OOD,?EE(`Z-N-N`S\`oSo`^>^___$(88 /'(9:eC /!VV$=sCS?T?TUU__ HH%OV]H^^^_______________"VV$lnqsCoDoDEE//$($A$A $ 01B C ) 0 0 @ @ . 5 5d6I4> Z Z%%E %z/ $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|jdtst 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} propertiesrgcreate_variable_is_dictrmrsrir rrrnrs rXrJz,CodeGeneratorDraft04.generate_min_propertiessl $$&&& VV, - - j jd.?EE 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|jdtst 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} propertiesrgrrs rXrKz,CodeGeneratorDraft04.generate_max_propertiessl $$&&& VV, - - w wd.?EE 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 ||d5t|jdtt fst d||d5|d| |jdddddn #1swxYwYddddS#1swxYwYdS)Nrr.zrequired must be an arrayz6if not all(prop in {variable} for prop in {required}):z!{name} must contain {} propertiesrg) rrmrsrirtrurrrnrvrs rXrLz&CodeGeneratorDraft04.generate_requireds $$&&& VV, - - u ud.z:T5MJJ Q34OPPP  , , . . .PQQ u ub>bisttt u u u u u u u u u u u u u u u  u u u u u u u u u u u u u u u u u us6A C% 7C  C% C C%C C%%C),C)c 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])rdz{}[{}] must be objectzif "{}" in {variable}_keys:z{variable}_keys.remove("{}")z#{variable}__{0} = {variable}["{1}"]rz{}.{}TrxNrzelse: {variable}["{}"] = {})rrmcreate_variable_keysrir+rrrsr rrr$rlrvrzr{r6r)rRkeyprop_definitionkey_names rXrNz(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\kOkiFF8$&&++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}}Trx) rrmrrir+rrrrrzr$rlr{)rRr#rSs rXrOz0CodeGeneratorDraft04.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 therergzif "{}" not in {variable}:z#{name} missing dependency {} for {}Nrxr) rrmrir+rvrnrsrtrzrlr{)rRis_emptyrvaluesvalues rXrMz*CodeGeneratorDraft04.generate_dependencies6s$ $$&&& VV, - -  H#/?EEGG y y VR<6T> VV4dffSkkBByyy!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\r9r:r;r<r=r>r?r@rArBrrCrDrErFrGrHrIrJrKrLrNrOrPrM __classcell__)rWs@rXrrsQD)Niy' M-1"$&&&&&&BX KKK,LLL vvv&SSS<sss>VVV8lllmmmWWWZZZBJJJ a a a b b b W W W[[[hhhQQQB>+>+>+@jjjwwwuuuiiiB>"y"y"yH#######rYr) rr exceptionsr generatorrrr`rrrrYrXrs 55555522222222   <(( C C C C C =C C C C C rY