j3zddlmZddlmZddlZddlmZmZddlmZddl m Z dZ Gd d Z d Z d ZdS) ) OrderedDict)DecimalN)JsonSchemaValueExceptionJsonSchemaDefinitionException)indent) RefResolverc6t|tr|S|gSN) isinstancelist)variables e/var/www/finuniver-perm.ru/html/student/venv/lib/python3.11/site-packages/fastjsonschema/generator.py enforce_listr s!(D!! :ceZdZdZdZddZedZedZedZ d Z d Z d Z dd Z dZdZdZedZdZddddZdZdZdZdZdZdS) CodeGeneratora7 This class is not supposed to be used directly. Anything inside of this class can be changed without noticing. This class generates code of validation function from JSON schema object as string. Example: .. code-block:: python CodeGenerator(json_schema_definition).func_code NTcg|_i|_i|_||_dg|_dt i|_t|_d|_ d|_ d|_ d|_ ||_ d|_i|_t|_|t#j|i}||_|j|j|j<t-|_dS)Nzfrom decimal import Decimalrr)store)_code_compile_regexps_custom_formats_detailed_exceptions_extra_imports_linesr_extra_imports_objectsset _variables_indent_indent_last_line _variable_variable_name_root_definition _definition_needed_validation_functions_validation_functions_doner from_schema _resolverget_scope_nameget_urir_json_keywords_to_function)self definitionresolverdetailed_exceptionss r__init__zCodeGenerator.__init__ s "!$7! *% ! w' #%% !%" *-/)*-%%'  ".zDDDH!GKnFcFcFeFe)$.*@*@*B*BC*5--'''rc^|d|jS)zP Returns generated code of whole validation function as string.  )_generate_func_codejoinrr,s r func_codezCodeGenerator.func_codeGs*   """yy$$$rcv|tdi|j|jtt dS)z Returns global variables for generating function from ``func_code``. Includes compiled regular expressions and imports, so it does not have to do it every time when validation function is called. )REGEX_PATTERNSrer)r3dictrrr9rr5s r global_statezCodeGenerator.global_statePsS   """  ) 0%=      rc ||jsd|jgdzSd|jdddddt |jzdgzS)z Returns global variables for generating function from ``func_code`` as code. Includes compiled regular expressions and imports. r2)3from fastjsonschema import JsonSchemaValueExceptionr?z import rer>r?zREGEX_PATTERNS = )r3rr4rserialize_regexesr5s rglobal_state_codezCodeGenerator.global_state_code`s   """$ 99T6:::  yy2  A   "3D4I"J"J J  6     rc@|js|dSdSr )rgenerate_func_coder5s rr3z!CodeGenerator._generate_func_codexs.z &  # # % % % % % & &rc|d|jr;|j\}}||||j9dSdS)zt Creates base code of validation function and calls helper for creating code by definition. zNoneType = type(None)N)lr%popitemgenerate_validation_function)r,urinames rrCz CodeGenerator.generate_func_code|ss &'''/ 99AACCIC  - -c4 8 8 8 / 9 9 9 9 9rc|j||d|j|5}|d|5||ddd|ddddn #1swxYwYddddS#1swxYwYdS)zL Generate validation function for given uri with given name r?z4def {}(data, custom_formats={{}}, name_prefix=None):dataT)clear_variablesz return dataN)r&addrEr( resolvinggenerate_func_code_block)r,rHrIr-s rrGz*CodeGenerator.generate_validation_functions< '++C000 r ^ % %c * * &jNPTUU & &--j&&Z^-___}%%% & & & & & & & & & & & & & & & & & & & & & & & & & & & & & & & & &s6 B4!/B B4B B4#B $B44B8;B8Fc|j|j|jf}|||c|_|_|_|r|j}t |_||}|\|_|_|_|r||_|S)z Creates validation rules for current definition. Returns the number of validation rules generated as code. )r$r!r"rr_generate_func_code_block)r,r-r variable_namerLbackupbackup_variablescounts rrOz&CodeGenerator.generate_func_code_blocks !4>43FF@JHVc=$.$*=  $# !eeDO..z::@F=$.$*=  /.DO rct|tstdd|vr|S||S)Nzdefinition must be an object$ref)r r;r generate_refrun_generate_functions)r,r-s rrQz'CodeGenerator._generate_func_code_blocksU*d++ P/0NOO O Z  $$&& &..z:: :rcnd}|jD]\}}||vr||dz }|S)z>@@  ICj    rc|j|jd5|j}|j}||jvr ||j|<|jdsJ|jdd}d |}d|vr|dz}| d|| ddddS#1swxYwYdS) aW Ref can be link to remote or local definition. .. code-block:: python {'$ref': 'http://json-schema.org/draft-04/schema#'} { 'properties': { 'foo': {'type': 'integer'}, 'bar': {'$ref': '#/properties/foo'} } } rWrKrNz(name_prefix or "data") + "{}"{z.format(**locals())z*{}({variable}, custom_formats, {name_arg}))name_arg) r(in_scoper$r)r*r&r%r" startswithformatrE)r,rIrHpathr`s rrXzCodeGenerator.generate_refsJ^ $ $T%5f%= > > Z Z>0022D.((**C$9999=1#6&11&99 9 9 9&qrr*D7>>tDDHh#&;; FF?PXF Y Y Y Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z ZsB'CC!Ccd|jz|jz}|j}|r-|dsJd|ddz}d|vr|dz}t |jr|jdur|jnif|j|d |}|j|i|}|d d d d }|j ||z|S)a Short-cut of line. Used for inserting line. It's formated with parameters ``variable``, ``variable_name`` (as ``name`` for short-cut), all keys from current JSON schema ``definition`` and also passed arguments in ``args`` and named ``kwds``. .. code-block:: python self.l('if {variable} not in {enum}: raise JsonSchemaValueException("Wrong!")') When you want to indent block, use it as context manager. For example: .. code-block:: python with self.l('if {variable} not in {enum}:'): self.l('raise JsonSchemaValueException("Wrong!")')  rKz" + (name_prefix or "data") + "rNr_z".format(**locals()) + "T)rrIr2z\n z\r) INDENTrr"rbr;r$r!rcreplacerappend)r,lineargskwdsspacesrIcontexts rrEzCodeGenerator.ls &t{"T\1"  9??6** * * *4tABBx?Dd{{88 $ 0 YT5ET5Q5QD  WY ^      t{D,G,,||D%((00u== &4-((( rcHt|ddS)z Short-cut of escape. Used for inserting user values into a string message. .. code-block:: python self.l('raise JsonSchemaValueException("Variable: {}")', self.e(variable)) "z\")strri)r,strings rezCodeGenerator.es 6{{""3...r) append_to_msgrulec|js|jd|zdzg|RdSd|zdz}|r |d|zdzz }d|zdz}||j}|t |t r||nd}|j|g|Rt|t||d dS) zG Short-cut for creating raising exception in the code. z raise JsonSchemaValueException("z")Nrqz + ()zraise JsonSchemaValueException(zH, value={variable}, name="{name}", definition={definition}, rule={rule}))r-rvdefinition_rule) rrE _expand_refsr$rtr r;getrepr)r,msgrurvrlargr-rys rexczCodeGenerator.exc s(  DF5c9$> F F F F F F#gck  0 6M)C/ /C/34~~&&t'788 &&JPT9U9U!_!5!5!5[_``siTiid:&6&6T$ZZYhiiiiiircjt|trfd|DSt|ts|Sd|vrUt|dtr:j|d5}|cdddS#1swxYwYfd|DS)Nc:g|]}|Sr:rz).0vr,s r z.CodeGenerator._expand_refs..s'===QD%%a((===rrWcBi|]\}}||Sr:r)rkrr,s r z.CodeGenerator._expand_refs.."s-GGGDAq4$$Q''GGGr)r r r;rrr(rNr[)r,r-schemas` rrzzCodeGenerator._expand_refss j$ ' ' >====*=== =*d++   Z  Jz&/A3$G$G ))*V*<==                 GGGGJ4D4D4F4FGGGGs;B  BBcd|j}||jvrdS|j||ddS)a Append code for creating variable with length of that variable (for example length of list or dictionary) with name ``{variable}_len``. It can be called several times and always it's done only when that variable still does not exists. z{}_lenNz {variable}_len = len({variable})rcr!rrMrEr,rRs rcreate_variable_with_lengthz)CodeGenerator.create_variable_with_length$sX!77 DO + + F M*** 122222rcd|j}||jvrdS|j||ddS)z Append code for creating variable with keys of that variable (dictionary) with a name ``{variable}_keys``. Similar to `create_variable_with_length`. z{}_keysNz({variable}_keys = set({variable}.keys())rrs rcreate_variable_keysz"CodeGenerator.create_variable_keys1sZ "((88 DO + + F M*** 9:::::rcd|j}||jvrdS|j||ddS)z Append code for creating variable with bool if it's instance of list with a name ``{variable}_is_list``. Similar to `create_variable_with_length`. z {}_is_listNz:{variable}_is_list = isinstance({variable}, (list, tuple))rrs rcreate_variable_is_listz%CodeGenerator.create_variable_is_list<sZ %++DN;; DO + + F M*** KLLLLLrcd|j}||jvrdS|j||ddS)z Append code for creating variable with bool if it's instance of list with a name ``{variable}_is_dict``. Similar to `create_variable_with_length`. z {}_is_dictNz1{variable}_is_dict = isinstance({variable}, dict)rrs rcreate_variable_is_dictz%CodeGenerator.create_variable_is_dictGsZ %++DN;; DO + + F M*** BCCCCCr)NT)F)__name__ __module__ __qualname____doc__rhr0propertyr6r<rAr3rCrGrOrQrYrXrrErtrrzrrrrr:rrrrs  F%8%8%8%8N%%X%   X    X .&&& 9 9 9 & & &(;;;ZZZ: %% V%N///-1tjjjjj HHH 3 3 3 ; ; ; M M M D D D D Drrctd|D}dd|zdzS)Nc3`K|])\}}t|dzt|zV*dS)z: N)r| repr_regex)rrrs r z$serialize_regexes..VsO Aq Q$A&rz{ z, z })r[r4) patterns_dictregex_patternss rr@r@SsJ!''))N y~~n55 5 ==rcd}dfd|D}|rd|znd}dj|S)N)AIDEBUGLMSXz | c3\K|]&}jtt|zd|V'dS)zre.N)flagsgetattrr9)rfregexs rrzrepr_regex.._s>TTQu{WRQR^^7STyQyyTTTTTTrz, r?zre.compile({!r}{}))r4rcpattern)r all_flagsrs` rrr]sX7I JJTTTT)TTT T TE! )D5LLrE  & &u}e < <rs###### OOOOOOOO%%%%%%DDDDDDDDD >>>=====r