o l hXx@s^ddlZddlZddlmZddlmZmZddddd d d d Zed Z GdddeZ dS)N)JsonSchemaDefinitionException) CodeGenerator enforce_listNoneTypeboolzint, float, Decimalintstrz list, tupledict)nullbooleannumberintegerstringarrayobjectz (?d?ZZ S)@CodeGeneratorDraft04zk^\d{4}-[01]\d-[0-3]\d(t|T)[0-2]\d:[0-5]\d:[0-5]\d(?:\.\d+)?(?:[+-][0-2]\d:[0-5]\d|[+-][0-2]\d[0-5]\d|z|Z)\Zz.^(?!.*\.\..*@)[^@.][^@]*(? zCodeGeneratorDraft04.__init__cstj}|j|d<|S)Ncustom_formats)r0 global_stater2)rPresrWrYrZr\Gs z!CodeGeneratorDraft04.global_statec Cst|jd}z ddd|D}Wnty&}ztd|d}~wwd}d|vs1d |vrXsz5CodeGeneratorDraft04.generate_type..zUnknown type: {}Nr rr z or isinstance({variable}, bool))variablez&if not isinstance({variable}, ({})){}:{name} must be {}z or rule) r _definitionjoinKeyErrorrr! _variablelexc)rPtypes python_typesrmextrarYrYrZr7Ms "z"CodeGeneratorDraft04.generate_typecCsf|jd}t|ttfstd|d|jd||ddWddS1s,wYdS)z Means that only value specified in the enum is valid. .. code-block:: python { 'enum': ['a', 'b'], } rzenum must be an arrayzif {variable} not in {enum}:z{name} must be one of {}rfN)rh isinstancelisttuplerrlrme)rPrrYrYrZr8cs  "z"CodeGeneratorDraft04.generate_enumcCs*|jdD] }|j||j|jddqdS)a Means that value have to be valid by all of those definitions. It's like put it in one big definition. .. code-block:: python { 'allOf': [ {'type': 'number'}, {'minimum': 5}, ], } Valid values for this definition are 5, 6, 7, ... but not 4 or 'abc' for example. rTclear_variablesN)rhgenerate_func_code_blockrk_variable_name)rPdefinition_itemrYrYrZr9ssz$CodeGeneratorDraft04.generate_all_ofc Cs|jd7_|j}|jd|d|jdD]H}|jd|dd6|jddd |j||j|jd d |jd |dWd n1sEwY|dWd n1sYwYq|jd|dd|jdddWd d S1szwYd 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}:Fr{optimizetry:r}Truz#{variable}_any_of_count{count} += 1N%except JsonSchemaValueException: passz,{name} cannot be validated by any definitionrfrOrlrhrwrkrxrm)rPr{ryrYrYrZr:s "z$CodeGeneratorDraft04.generate_any_ofc Cs|jd7_|j}|jd|d|jdD]H}|jd|dd6|jddd |j||j|jd d |jd |dWd n1sEwY|dWd n1sYwYq|jd|dd}|jd||ddWd d S1s}wYd S)a Means that value have to be valid by only one of those definitions. It can't be valid by two or more of them. .. code-block:: python { 'oneOf': [ {'type': 'number', 'multipleOf': 3}, {'type': 'number', 'multipleOf': 5}, ], } Valid values for this definition are 3, 5, 6, ... but not 15 for example. rz"{variable}_one_of_count{count} = 0rzrz&if {variable}_one_of_count{count} < 2:Fr|r~rTruz#{variable}_one_of_count{count} += 1Nrz'if {variable}_one_of_count{count} != 1:z9" (" + str({variable}_one_of_count{}) + " matches found)"z.{name} must be valid exactly by one definition) append_to_msgrgr)rPr{rydynamicrYrYrZr;s  "z$CodeGeneratorDraft04.generate_one_ofcCs|jd}|dur|jddddS|durdS|s#|jddddS|jddd |||j|jWdn1s>wY|d |d |jdddWddS1s`wYdS) 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. rT{name} must not be thererfFNz-{name} must NOT match a disallowed definitionr~rrzelse:)rhrmrlrwrkrx)rPnot_definitionrYrYrZr<s   "z!CodeGeneratorDraft04.generate_notc C|d?|t|jdttfstd|d|jdddWdn1s/wYWddSWddS1sGwYdS)Nif isinstance({variable}, str):rzminLength must be a numberz if {variable}_len < {minLength}:z={name} must be longer than or equal to {minLength} charactersrfrlcreate_variable_with_lengthrqrhrfloatrrmrPrYrYrZr=  "z(CodeGeneratorDraft04.generate_min_lengthc Cr)NrrzmaxLength must be a numberz if {variable}_len > {maxLength}:z>{name} must be shorter than or equal to {maxLength} charactersrfrrrYrYrZr>rz(CodeGeneratorDraft04.generate_max_lengthc Cs|dN|jd}|dddd}td|}t||j|<|dt||j d |dd Wdn1s>wYWddSWddS1sVwYdS) Nrr \z\\"z\"z\\Zz-if not REGEX_PATTERNS[{}].search({variable}):z{name} must match pattern {}rf) rlrhreplace DOLLAR_FINDERsubrecompile_compile_regexpsreprrm)rPr safe_patternend_of_string_fixed_patternrYrYrZr?s   "z%CodeGeneratorDraft04.generate_patternc Cs|jsdS|d|jd}||jvrH|j|}t|tr)|||d|n|d||jd|ddWdn1sBwYn\||jvr\|j|}|||d|nX|dkrd g|_ |jd d d  |d Wdn1s{wY|d|jdddWdn1swYnt d |WddSWddSWddSWddS1swYdS)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}):rerfregexz import rer~Frzre.compile({variable})zexcept Exception:z{name} must be a valid regexzUnknown format: {}) r3rlrhr2rqr _generate_formatrm FORMAT_REGEXS_extra_imports_linesrr!)rPformat_ custom_format format_regexrYrYrZr@sB           "z$CodeGeneratorDraft04.generate_formatcCsn|jd|kr5||jvrt||j|<|d||jd|ddWddS1s.wYdSdS)Nr!z.if not REGEX_PATTERNS["{}"].match({variable}):rerf)rhrrrrlrm)rP format_name regexp_nameregexprYrYrZrs "z%CodeGeneratorDraft04._generate_formatc C|dit|jdtttjfstd|jddr:|d|j dddWdn1s4wYn$|d |j d ddWdn1sQwYWddSWddSWddS1sqwYdS) N1if isinstance({variable}, (int, float, Decimal)):r"zminimum must be a numberexclusiveMinimumFzif {variable} <= {minimum}:z${name} must be bigger than {minimum}rfzif {variable} < {minimum}:z0{name} must be bigger than or equal to {minimum} rlrqrhrrdecimalDecimalrgetrmrrYrYrZrA"   "z%CodeGeneratorDraft04.generate_minimumc Cr) Nrr#zmaximum must be a numberexclusiveMaximumFzif {variable} >= {maximum}:z%{name} must be smaller than {maximum}rfzif {variable} > {maximum}:z1{name} must be smaller than or equal to {maximum}rrrYrYrZrB-rz%CodeGeneratorDraft04.generate_maximumc Cs|dlt|jdtttjfstdt|jdtr$|dn|d|d|jdddWdn1s@wY|d |jd ddWdn1s\wYWddSWddS1stwYdS) 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}rfz-if {variable} / {multipleOf} == float("inf"):zinifinity reached) rlrqrhrrrrrrmrrYrYrZrC8s      "z)CodeGeneratorDraft04.generate_multiple_ofc C||d?t|jdttfstd||d|jdddWdn1s3wYWddSWddS1sKwYdS)Nif {variable}_is_list:r%zminItems must be a numberzif {variable}_len < {minItems}:z-{name} must contain at least {minItems} itemsrf create_variable_is_listrlrqrhrrrrrmrrYrYrZrDH  "z'CodeGeneratorDraft04.generate_min_itemsc Cr)Nrr&zmaxItems must be a numberzif {variable}_len > {maxItems}:z:{name} must contain less than or equal to {maxItems} itemsrfrrrYrYrZrEQrz'CodeGeneratorDraft04.generate_max_itemsc Cs|jd}|s dS||d6|d||d|jdddWdn1s3wYWddSWddS1sKwYdS)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 itemsrf)rhrrlrrm)rPunique_definitionrYrYrZrFZs    "z*CodeGeneratorDraft04.generate_unique_itemsc Cs|jd}|dur dS||d=||dur;|d|jdddWdn1s5wYnt|trt|D]G\}}|d ||d |||d |j |d |j |Wdn1spwY|j rt|t rd |vr|dt|d qDd|jvr|jddur|d t||jdddWdn1swYn|dt|%||jdd |j d |j |j }|dkr|dWdn1swYnL|r2|d#||d |j d |j |j }|dkr|dWdn11swYWddSWddSWddSWddSWddSWddS1sVwYdS)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}:rrfzif {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}):)rhrrlrrmrqrr enumeraterwr!rkrxr4r rlen)rPitems_definitionidxitem_definitionr{rYrYrZrG{s~               $z#CodeGeneratorDraft04.generate_itemsc Cr)Nif {variable}_is_dict:r)zminProperties must be a numberz$if {variable}_len < {minProperties}:z7{name} must contain at least {minProperties} propertiesrf create_variable_is_dictrlrqrhrrrrrmrrYrYrZrHrz,CodeGeneratorDraft04.generate_min_propertiesc Cr)Nrr*zmaxProperties must be a numberz$if {variable}_len > {maxProperties}:zD{name} must contain less than or equal to {maxProperties} propertiesrfrrrYrYrZrIrz,CodeGeneratorDraft04.generate_max_propertiesc sd|tjdttfstdtjdttjdkr,tdj ddsJfddjdD}|rJtd j |d d d }j d jdd|dWdn1spwYWddSWddS1swYdS)Nrr+zrequired must be an arrayz%required must contain unique elementsr/TcsBg|]jdivrtfddjdiDsqS)r-c3s|] }t|VqdSr^)rsearch)r`rproprYrZrbszDCodeGeneratorDraft04.generate_required...r.)rhrany)r`rrrZ sz:CodeGeneratorDraft04.generate_required..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 )rgr)rrlrqrhrrrsrrsetrr!rkrmrt)rP not_possiblerrYrrZrJs(      "z&CodeGeneratorDraft04.generate_requiredc Cs>||d||jdD]w\}}tdd|}t|tt fs0t d |j ||d| |1|d| ||d|| ||j|d |j |d |j| |d d Wd n1smwY|jrt|trd|vr|d| |t|dqWd d S1swYd 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. rr-z($[^a-zA-Z]|[^a-zA-Z0-9])rcz{}[{}] must be objectzif "{}" in {variable}_keys:z{variable}_keys.remove("{}")z#{variable}__{0} = {variable}["{1}"]rz{}.{}TruNrzelse: {variable}["{}"] = {})rrlcreate_variable_keysrhr(rrrqr rrr!rkrtrwrxr4r)rPkeyprop_definitionkey_namerYrYrZrLs,   "z(CodeGeneratorDraft04.generate_propertiesc CsD||d||jdD] \}}t||j|<q|dY|jdD]I\}}|dt|5|d |dWdn1sQwY|j |d |j d |j |j d d Wdn1sswYq/Wdn1swYWddSWddS1swYdS) 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. rr.z9for {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}}Tru) rrlrrhr(rrrrrwr!rkrx)rPr rQrYrYrZrMs0     "z0CodeGeneratorDraft04.generate_pattern_propertiesc Csb||d||jd}|dus|ikr$ WddS|rst|jdi}|d4|d||d||d |j d |j |j Wdn1s^wYWdn1smwYn$|d |j d dd Wdn1swYWddSWddSWddS1swYdS)a Means object with keys with values defined by definition. .. code-block:: python { 'properties': { 'key': {'type': 'number'}, } 'additionalProperties': {'type': 'string'}, } Valid object is containing key called 'key' and it's value any number and any other key with any string. rr/TNr-z&for {variable}_key in {variable}_keys:zif {variable}_key not in {}:z1{variable}_value = {variable}.get({variable}_key)z{}_valuerzif {variable}_keys:z;{name} must not contain "+str({variable}_keys)+" propertiesrf) rrlrrhrrrkeysrwr!rkrxrm)rPadd_prop_definitionproperties_keysrYrYrZrN#s:      "z3CodeGeneratorDraft04.generate_additional_propertiesc CsP||dd}|jdD]s\}}|gks|dur qd}|d||S|dur9|jd|ddn>t|trl|D]*}|d|||jd ||||ddWd n1sewYq@n |j||j |j dd Wd n1swYq|r|d Wd d SWd d S1swYd 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 thererfzif "{}" not in {variable}:z#{name} missing dependency {} for {}Nrur) rrlrhr(rtrmrqrrrwrkrx)rPis_emptyrvaluesvaluerYrYrZrKGs4    "z*CodeGeneratorDraft04.generate_dependencies)!__name__ __module__ __qualname__rr1propertyr\r7r8r9r:r;r<r=r>r?r@rrArBrCrDrErFrGrHrIrJrLrMrNrK __classcell__rYrYrWrZrsH " $    !@  !$r) rr exceptionsr generatorrrr_rrrrYrYrYrZs