o tBh[n@spUdZddlmZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.dd l/m0Z0m1Z1m2Z2m3Z3m4Z4dd l5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAdd lBmCZCd ZDd eEd<dZFd eEd<dZGd eEd<GdddZHdS)zZSemantic analysis of named tuple definitions. This is conceptually part of mypy.semanal. ) annotations)contextmanager)IteratorListMappingcast)Final)TypeTranslationErrorexpr_to_unanalyzed_type) ARG_NAMED_OPTARG_OPTARG_POSMDEFArgumentAssignmentStmtBlockCallExprClassDefContext Decorator EllipsisExpr ExpressionExpressionStmtFuncBaseFuncDefListExprNamedTupleExprNameExprPassStmtRefExprStrExpr SymbolTableSymbolTableNodeTempNode TupleExprTypeInfo TypeVarExprVar)Options)PRIORITY_FALLBACKSSemanticAnalyzerInterfacecalculate_tuple_fallbackhas_placeholderset_callable_name) TYPED_NAMEDTUPLE_NAMESAnyType CallableType LiteralType TupleTypeType TypeOfAnyTypeTypeTypeVarLikeType TypeVarType UnboundType has_type_vars)get_unique_redefinition_name) __new____init__ __slots____getnewargs___fields_field_defaults _field_types_make_replace_asdict_source__annotations__rNAMEDTUPLE_PROHIBITED_NAMESzYInvalid statement in NamedTuple definition; expected "field_name: field_type [= default]"NAMEDTUP_CLASS_ERROR_NTSELF_TVAR_NAMEc@steZdZdDddZdEddZdFddZdGddZdHd"d#ZdId&d'ZdJd-d.Z dKd9d:Z e dLd=d>Z dMdAdBZ dCS)NNamedTupleAnalyzeroptionsr(apir*returnNonecCs||_||_dSN)rLrM)selfrLrMrRn/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mypy/semanal_namedtuple.pyr<]s zNamedTupleAnalyzer.__init__defnr is_stub_filebool is_func_scopetuple[bool, TypeInfo | None]c Cs|jD]f}t|tri|j||jtvri|||}|dur"dS|\}}}|r:d|jvr:|jdt |j 7_d} t|j t rF|j j } ||j||||j | } t | dd|_ |j |j _ |j|j _d| fSqdS)aAnalyze if given class definition can be a named tuple definition. Return a tuple where first item indicates whether this can possibly be a named tuple, and the second item is the corresponding TypeInfo (may be None if not ready and should be deferred). N)TN@Tis_typed)FN)base_type_exprs isinstancerrMacceptfullnamer.check_namedtuple_classdefnamestrlineanalyzedrinfobuild_namedtuple_typeinfocolumn) rQrTrUrW base_exprresultitemstypes default_items existing_inforerRrRrSanalyze_namedtuple_classdefas,         z.NamedTupleAnalyzer.analyze_namedtuple_classdef:tuple[list[str], list[Type], dict[str, Expression]] | Nonec Cs|jjdkr|s|d|ggifSt|jdkr |d|g}g}i}|jjD]}t|ts]t|t sAt|t rBt|j t rBq*t|t tfrJq*t|t rVt|j trVq*|t|q*t|jdkslt|jdtss|t|q*|jdj}|||jdur|ttjn|jj|j|jjo|j d}|durdS|||dr|d |||jdust|d r|js|t|q*t|j t!r|r|d |q*|j ||<q*|||fS) zParse and validate fields in named tuple class definition. Return a three tuple: * field names * field types * field default values or None, if any of the types are not ready. )z7NamedTuple class syntax is only supported in Python 3.6z"NamedTuple should be a single baserNallow_placeholder_z7NamedTuple field name cannot start with an underscore: new_syntaxz:Non-default NamedTuple fields cannot follow default fields)"rLpython_versionfaillenr\defsbodyr]rrrexprrrrr rHlvaluesrraappendtyper/r4 unannotatedrM anal_typeenable_recursive_aliasesrW startswithhasattrrvrvaluer#) rQrTrUrjrkrlstmtrardrRrRrSr`sb                 z,NamedTupleAnalyzer.check_namedtuple_classdefnodervar_name str | None9tuple[str | None, TypeInfo | None, list[TypeVarLikeType]]cCst|ts ddgfS|}|j}t|tsddgfS|j}|dkr#d}n |tvr*d}nddgfS|||}|r@|\} } } } } }n>|rP|}|rO|dt|j7}n dt|j}}| |ggi|jd}| ||||||ksr|ry|j ||||gfS|s| dgfS|r|}n| }|dus|r|dt|j7}t | dkrdd t| t |  d| D}ni}d}t|jtr|jj}| || | ||j|}|r| ||||n t||d |_|j|||ks|r|j ||| || fS) aCheck if a call defines a namedtuple. The optional var_name argument is the name of the variable to which this is assigned, if any. Return a tuple of two items: * Internal name of the named tuple (e.g. the name passed as an argument to namedtuple) or None if it is not a valid named tuple * Corresponding TypeInfo, or None if not ready. If the definition is invalid but looks like a namedtuple, report errors but return (some) TypeInfo. Ncollections.namedtupleFTrYz namedtuple@rcSsi|]\}}||qSrRrR).0arg_namedefaultrRrRrS s z7NamedTupleAnalyzer.check_namedtuple..rZ)r]rcalleerr_r.parse_namedtuple_argsrbrcrfstore_namedtuple_inforMadd_symbol_skip_localryziprdrreset_line)rQrrrWcallrr_r[rirjrkdefaultstypename tvar_defsokrarerlrmrRrRrScheck_namedtuplesf               z#NamedTupleAnalyzer.check_namedtuplerer%rarbrrr[cCs.|j|||t||d|_|j|dS)NrZ)rM add_symbolrrdr)rQrerarr[rRrRrSr8sz(NamedTupleAnalyzer.store_namedtuple_infor_WNone | tuple[list[str], list[Type], list[Expression], str, list[TypeVarLikeType], bool]cCs|tvrdnd}|j}t|dkr|d|d|dSg}t|dkr`|tvr1|d|dSt|jdddD]$\}}|dkr_||}t|ttfrTt |j }n |d ||nq;|j ddt t gkrw|d |d|dSt|d ts|d |d |dStt|jd j} g} g} t|dttfs|dkrt|dtr|d} | jdd} n[|d ||dS|d}|dkrtdd|j Dr|d|dSdd|j D} n-dd|j D}|j|} ||j |}|durdS|\} } }}|sggg| gdfS| sdd| D} dd| D}|r7|d |dd||t|t| krR|d|d||dt| }| | || | dfS) aNParse a namedtuple() call into data needed to construct a type. Returns a 5-tuple: - List of argument names - List of argument types - List of default values - First argument of namedtuple - Whether all types are ready. Return None if the definition didn't typecheck. NamedTuple namedtuplezToo few arguments for "z()"Nz%Too many arguments for "NamedTuple()"rz?List or tuple literal expected as the defaults argument to {}()zUnexpected arguments to "r"z2()" expects a string literal as the first argumentrrr, z?List or tuple literal expected as the second argument to "{}()"css|] }t|t VqdSrP)r]r ritemrRrRrS sz;NamedTupleAnalyzer.parse_namedtuple_args..z.String literal expected as "namedtuple()" itemcSsg|]}tt|jqSrR)rr valuerrRrRrS sz.cSs.g|]}t|trt|jdkr|jdqS)rrr)r]r$ryrj)rtrRrRrSrsFcSsg|]}ttjqSrR)r/r4rrrurRrRrSrcSsg|] }|dr|qS)ru)rrrRrRrSrz1()" field names cannot start with an underscore: z, z$Too many defaults given in call to "T)r.argsryrx enumerate arg_namesr]rr$listrjformat arg_kindsr r rrreplacesplitanyrMget_and_bind_all_tvars"parse_namedtuple_fields_with_typesjoin)rQrr_ type_namerrirargrrkrstr_exprrjlistexpr type_exprsrirur underscorerRrRrSr?s          z(NamedTupleAnalyzer.parse_namedtuple_argsnodeslist[Expression]contextr;tuple[list[str], list[Type], list[Expression], bool] | Nonec Csg}g}|D]}t|trt|jdkr|d|dS|j\}}t|tr.||jn |d|dSz t||j |j j }Wnt yS|d|YdSw|j j ||j joa|j  d} t| trnttj} | durzgggdfS|| q|d|dS||gd fS) zParse typed named tuple fields. Return (names, types, defaults, whether types are all ready), or None if error occurred. rz'Invalid "NamedTuple()" field definitionNz!Invalid "NamedTuple()" field namezInvalid field typersFz&Tuple expected as "NamedTuple()" fieldT)r]r$ryrjrxr r~rr rLrMrUr rrrWr8r/r4 from_error) rQrrrjrkrra type_noderrdrRrRrSrs>             z5NamedTupleAnalyzer.parse_namedtuple_fields_with_typesrj list[str]rk list[Type]rlMapping[str, Expression]rcintrmTypeInfo | Nonec sN|jdttj}|jd|g}|jd|gp!|jd} |jd|gp0|jd} |jd|g} |jd|g} |jdfdd|D} t| |}|p^|j|| d _t|| j rwt j j rw|jj d d  _d |ijd <t sts|jtfd d dAdBfdd }ddt||D}|D]}||d dqddt||D}tfdd|D|}|td|d d|td| d d|td | d d|td!d d|td"| d d|td#d d|jjd$kr|td%|d djdusJttjd&t|jjgj  dAdCfd/d0 }|d1d2d|Dd3dDfd5d6 |d7fd8d|Dd d9|d:g| d;ttj}|d<d ttd=| | dtttd>|t t!ttd?|t t!gd@t"tjd&tgj}t#t$|j%t<S)ENz builtins.strzbuiltins.tuplez builtins.dictzbuiltins.objectztyping.Iterablezbuiltins.functioncsg|]}t|qSrR)r1rstrtyperRrSrrz@NamedTupleAnalyzer.build_namedtuple_typeinfo..T)force_progressfieldsrcstSrP)r+rR) tuple_baserRrSsz>NamedTupleAnalyzer.build_namedtuple_typeinfo..Fvarr'is_initialized_in_classrV is_propertyrNrOcs<|_||_||_jd|j|_tt|j|j<dS)N.) rerrr_ra _fullnamer"rnames)rrr)rerRrS add_fields z?NamedTupleAnalyzer.build_namedtuple_typeinfo..add_fieldcSg|] \}}t||qSrRr'rrtyprRrRrSrr)rcSrrRrrrRrRrSr rcsg|]}qSrRrRrrrRrSr sr?)rrAr@rErF__doc__)rp __match_args__rfuncnamerbretr3rlist[Argument]is_classmethodis_newcsB|s|rttdtdtg}n ttddtg}||}dd|D}dd|D}dd|D}d|vs;Jtttt||||} g| _ t ||t g} | _ || _ t| | | _jd|| _| _|rt|| j} d| _| _ | j| _d| _t| td g| } | _tt| } ntt| } d| _| j|<dS) N_cls_selfcSg|]}|jqSrR)type_annotationrrrRrRrSr-zTNamedTupleAnalyzer.build_namedtuple_typeinfo..add_method..cSsg|]}|jjqSrR)variablerarrRrRrSr.cSrrR)kindrrRrRrSr/rrT classmethod)rr'r5make_normalizedr r0rrr3 variablesrrreis_classr-rr_rrcr is_decoratedrrr"rplugin_generatedr)rrrrrfirstrkrjr signaturefuncvdecsym) function_typerercselftypetvdrRrS add_method s8     z@NamedTupleAnalyzer.build_namedtuple_typeinfo..add_methodrCcSsg|] }t||jttqSrR)rrrr rrrRrRrSrJs)rrrcs.|jd}|dur tnt}t||j||SrP)getrar r rr)rrr)rlrRrS make_init_argMszCNamedTupleAnalyzer.build_namedtuple_typeinfo..make_init_argr;csg|]}|qSrRrRr)rrRrSrRr)rrrrD)rrrBiterablenewry)rrr)FF)rr'rrVrrVrNrO) rrbrr3rrrrVrrVrNrO)rr'rNr)&rM named_typer/r4 special_formnamed_type_or_noner2basic_new_typeinfois_named_tuple special_aliasr,targetdeferupdate_tuple_typerccopymetadatar9schedule_patchr)rr'rLrw tuple_typer7rJr_ tvar_scopenew_unique_func_idrr rr r&r"rr)rQrarjrkrlrcrm implicit_anybasetuple_typedictypeordereddictypefallback iterable_typeliteralsmatch_args_typerrrvarstuple_of_stringsrspecial_form_anyself_tvar_exprrR) rlrrercrrrrrrSrfs           '   z,NamedTupleAnalyzer.build_namedtuple_typeinfonamed_tuple_infoIterator[None]c cs|j}t|_dVtD](}||jvr5|||j|urq |j|j}|dus+J|d|d|q |D]/\}}||jvrd|dkrHq:|j|}t|jtt frY|j sYq:t ||j}||j|<||j|<q:dS)ajPreserve the generated body of class-based named tuple and then restore it. Temporarily clear the names dict so we don't get errors about duplicate names that were already set in build_namedtuple_typeinfo (we already added the tuple field names while generating the TypeInfo, and actual duplicates are already reported). Nz'Cannot overwrite NamedTuple attribute "rr) rr!rGrrrxrjr]rrrr:) rQrnt_names prohibitedctxkeyrrr_keyrRrRrSsave_namedtuple_bodyfs.         z'NamedTupleAnalyzer.save_namedtuple_bodymsgr!cCs|j||dSrP)rMrx)rQr%r!rRrRrSrxszNamedTupleAnalyzer.failN)rLr(rMr*rNrO)rTrrUrVrWrVrNrX)rTrrUrVrNro)rrrrrWrVrNr) rer%rarbrrr[rVrNrO)rrr_rbrNr)rrrrrNr)rarbrjrrkrrlrrcrrmrrNr%)rr%rNr)r%rbr!rrNrO)__name__ __module__ __qualname__r<rnr`rrrrrfrr$rxrRrRrRrSrK\s   " H m  b + )rKN)Ir __future__r contextlibrtypingrrrrtyping_extensionsrmypy.exprtotyper r mypy.nodesr r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r' mypy.optionsr(mypy.semanal_sharedr)r*r+r,r- mypy.typesr.r/r0r1r2r3r4r5r6r7r8r9 mypy.utilr:rGrFrHrJrKrRrRrRrSs    | 8