o l h3@srddlmZddlmZddlZddlmZmZddlmZddl m Z dd Z Gd d d Z d d Z ddZdS)) OrderedDict)DecimalN)JsonSchemaValueExceptionJsonSchemaDefinitionException)indent) RefResolvercCst|tr|S|gSN) isinstancelist)variabler l/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/fastjsonschema/generator.py enforce_list s rc@seZdZdZdZd-ddZeddZed d Zed d Z d dZ ddZ ddZ d.ddZ ddZddZddZeddZddZddd d!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,ZdS)/ 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 NTcCsg|_i|_i|_||_dg|_dti|_t|_d|_ d|_ d|_ d|_ ||_ d|_i|_t|_|dur=tj|id}||_|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_exceptionsr r r__init__ s, zCodeGenerator.__init__cCs|d|jS)zP Returns generated code of whole validation function as string.  )_generate_func_codejoinrr(r r r func_codeGs zCodeGenerator.func_codecCs&|tdi|j|jttdS)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_PATTERNSrerNr )r.dictrrr3rr0r r r global_statePs zCodeGenerator.global_statec CsH||jsd|jgdSd|jdddddt|jdgS)z Returns global variables for generating function from ``func_code`` as code. Includes compiled regular expressions and imports. r-)3from fastjsonschema import JsonSchemaValueExceptionr7z import rer6r7zREGEX_PATTERNS = )r.rr/rserialize_regexesr0r r rglobal_state_code`s zCodeGenerator.global_state_codecCs|js |dSdSr )rgenerate_func_coder0r r rr.xs z!CodeGenerator._generate_func_codecCs8|d|jr|j\}}||||jsdSdS)zt Creates base code of validation function and calls helper for creating code by definition. zNoneType = type(None)N)lr!popitemgenerate_validation_function)r(urinamer r rr:|s  z CodeGenerator.generate_func_codec Cs|j||d|j|5}|d||j|dddd|dWdn1s1wYWddSWddS1sIwYdS)zL Generate validation function for given uri with given name r7z4def {}(data, custom_formats={{}}, name_prefix=None):dataT)clear_variablesz return dataN)r"addr;r$ resolvinggenerate_func_code_block)r(r>r?r)r r rr=s   "z*CodeGenerator.generate_validation_functionFcCs`|j|j|jf}||||_|_|_|r|j}t|_||}|\|_|_|_|r.||_|S)z Creates validation rules for current definition. Returns the number of validation rules generated as code. )r rrrr_generate_func_code_block)r(r)r variable_namerAbackupbackup_variablescountr r rrDs z&CodeGenerator.generate_func_code_blockcCs,t|ts tdd|vr|S||S)Nzdefinition must be an object$ref)r r4r generate_refrun_generate_functions)r(r)r r rrEs  z'CodeGenerator._generate_func_code_blockcCs2d}|jD]\}}||vr||d7}q|S)zpathrQr r rrKs     "zCodeGenerator.generate_refcOsd|j|j}|j}|r$|dsJd|dd}d|vr$|d}t|jr0|jdur0|jnif|j|d |}|j|i|}|d d d d }|j |||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!")')  r@z" + (name_prefix or "data") + "rNrPz".format(**locals()) + "T)r r?r-z\n z\r) INDENTrrrSr4r rrTreplacerappend)r(lineargskwdsspacesr?contextr r rr;s&zCodeGenerator.lcCst|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\")strrY)r(stringr r reszCodeGenerator.e) append_to_msgrulecGs|js|jd|dg|RdSd|d}|r"|d|d7}d|d}||j}|t|tr:||nd}|j|g|Rt|t||d dS) zG Short-cut for creating raising exception in the code. z raise JsonSchemaValueException("z")Nr`z + ()zraise JsonSchemaValueException(zH, value={variable}, name="{name}", definition={definition}, rule={rule}))r)redefinition_rule) rr; _expand_refsr rcr r4getrepr)r(msgrdrer\argr)rgr r rexc s   (zCodeGenerator.exccst|trfdd|DSt|ts|Sd|vr:t|dtr:j|d }|WdS1s5wYfdd|DS)Ncsg|]}|qSr rh).0vr0r r sz.CodeGenerator._expand_refs..rJcsi|] \}}||qSr rnrokrpr0r r "sz.CodeGenerator._expand_refs..)r r r4rar$rCrM)r(r)schemar r0rrhs   zCodeGenerator._expand_refscC4d|j}||jvr dS|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})rTrrrBr;r(rFr r rcreate_variable_with_length$s   z)CodeGenerator.create_variable_with_lengthcCrv)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())rwrxr r rcreate_variable_keys1   z"CodeGenerator.create_variable_keyscCrv)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))rwrxr r rcreate_variable_is_list<r{z%CodeGenerator.create_variable_is_listcCrv)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)rwrxr r rcreate_variable_is_dictGr{z%CodeGenerator.create_variable_is_dict)NT)F)__name__ __module__ __qualname____doc__rXr,propertyr1r5r9r.r:r=rDrErLrKrr;rcrmrhryrzr|r}r r r rrs4 '       '    rcCs$dd|D}dd|dS)Ncss(|]\}}t|dt|VqdS)z: N)rj repr_regexrrr r r Vs  z$serialize_regexes..z{ z, z })rMr/) patterns_dictregex_patternsr r rr8Ssr8cs:d}dfdd|D}|rd|nd}dj|S)N)AIDEBUGLMSXz | c3s*|]}jtt|@rd|VqdS)zre.N)flagsgetattrr3)rofregexr rr_s(zrepr_regex..z, r7zre.compile({!r}{}))r/rTpattern)r all_flagsrr rrr]sr) collectionsrdecimalrr3 exceptionsrrr ref_resolverrrrr8rr r r rs   D