o 7fo@s|UdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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,m-Z-m.Z.m/Z/ddl0m1Z1dd l2m3Z3m4Z4m5Z5m6Z6m7Z7dd l8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDdd lEmFZFd ZGd eHd<dZId eHd<dZJd eHd<GdddZKdS)zZSemantic analysis of named tuple definitions. This is conceptually part of mypy.semanal. ) annotations)contextmanager)FinalIteratorListMappingcast)TypeTranslationErrorexpr_to_unanalyzed_type)MessageBuilder) ARG_NAMED_OPTARG_OPTARG_POSMDEFArgumentAssignmentStmtBlockCallExprClassDefContext Decorator EllipsisExpr ExpressionExpressionStmtFuncBaseFuncDefListExprNamedTupleExprNameExprPassStmtRefExpr StatementStrExpr SymbolTableSymbolTableNodeTempNode TupleExprTypeInfo TypeVarExprVaris_StrExpr_list)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@steZdZdEd d ZdFddZdGddZdHddZdId$d%ZdJd(d)ZdKd/d0Z dLd;d<Z e dMd?d@Z dNdBdCZ dDS)ONamedTupleAnalyzeroptionsr+apir-msgr returnNonecCs||_||_||_dSN)rOrPrQ)selfrOrPrQrVP/home/ertert/spirit/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]k}t|trn|j||jtvrn|||}|dur"dS|\}}}} |r;d|jvr;|jdt |j 7_d} t|j t rG|j j } ||j||||j | } t | dd|_ |j |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 isinstancer rPacceptfullnamer1check_namedtuple_classdefnamestrlineanalyzedrinfobuild_namedtuple_typeinfocolumndefsbody) rUrXrYr[ base_exprresultitemstypes default_items statements existing_inforirVrVrWanalyze_namedtuple_classdeffs.         z.NamedTupleAnalyzer.analyze_namedtuple_classdefKtuple[list[str], list[Type], dict[str, Expression], list[Statement]] | Nonec Cst|jdkr |d|g}g}i}g}|jjD]}||t|ts[t|ts5t|t r6t|j t r6qt|t t fr>qt|t rJt|j trJq||j||t|qt|jdksjt|jdts{||j||t|q|jdj}|||jdur|ttjn|jj|j|j dd} | durdS|| |dr|d|||jdust|d r|js|t|qt|jt r|r|d |q|j||<q|j!rd } |j!D] } |j"#| | |q||||fS) aParse and validate fields in named tuple class definition. Return a four tuple: * field names * field types * field default values * valid statements or None, if any of the types are not ready. z"NamedTuple should be a single baserNNamedTuple item typeallow_placeholderprohibit_self_type_z7NamedTuple field name cannot start with an underscore: new_syntaxz:Non-default NamedTuple fields cannot follow default fieldsz( for "__init_subclass__" of "NamedTuple")$lenr`failrlrmappendrarrrexprrrrr"popremoved_statementsrKlvaluesrretyper2r7 unannotatedrP anal_typer[ startswithhasattrr}rvaluer%keywordsrQ(unexpected_keyword_argument_for_function) rUrXrYrprqrrrsstmtrerh for_functionkeyrVrVrWrdsp                   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}| rddt | 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.namedtupleFTr]z namedtuple@cSsi|]\}}||qSrVrV).0arg_namedefaultrVrVrW s z7NamedTupleAnalyzer.check_namedtuple..r^)rarcalleer rcr1parse_namedtuple_argsrfrgrjstore_namedtuple_inforPadd_symbol_skip_localzipr~rhrriset_line)rUrrr[callrrcr_rorprqdefaultstypename tvar_defsokrerirrrtrVrVrWcheck_namedtuplesf               z#NamedTupleAnalyzer.check_namedtuplerir'rerfrrr_cCs.|j|||t||d|_|j|dS)Nr^)rP add_symbolrrhr)rUrirerr_rVrVrWrFsz(NamedTupleAnalyzer.store_namedtuple_inforcWNone | 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 |dS|d j} g} g} t|dttfs|dkrt|dtr|d} | jdd} nU|d ||dS|d}|dkrt|j s|d|dSdd|j D} n,dd|j D}|j|} ||j |}|durdS|\} } }}|s ggg| gdfS| sdd| D} dd| D}|r-|d |dd||t|t| krH|d|d||dt| }| | || | dfS)aParse a namedtuple() call into data needed to construct a type. Returns a 6-tuple: - List of argument names - List of argument types - List of default values - First argument of namedtuple - All typevars found in the field definition - 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 argumentrwr, z?List or tuple literal expected as the second argument to "{}()"z.String literal expected as "namedtuple()" itemcSg|]}|jqSrV)valueritemrVrVrW z.cSs.g|]}t|trt|jdkr|jdqS)rrw)rar&r~rp)rtrVrVrWrsFcSsg|]}ttjqSrV)r2r7rrr|rVrVrWrcSsg|] }|dr|qS)r|)rrrVrVrWrz1()" field names cannot start with an underscore: z, z$Too many defaults given in call to "T)r1argsr~r enumerate arg_namesrarr&listrpformat arg_kindsrr"rreplacesplitr*rPget_and_bind_all_tvars"parse_namedtuple_fields_with_typesjoin)rUrrc type_namerrirargrrqrstr_exprrplistexpr type_exprsror|r underscorerVrVrWrMs           z(NamedTupleAnalyzer.parse_namedtuple_argsnodeslist[Expression]contextr;tuple[list[str], list[Type], list[Expression], bool] | Nonec Csg}g}|D]}t|tr}t|jdkr|d|dS|j\}}t|tr.||jn |d|dSz t||j |j j }Wnt yS|d|YdSw|j j ||j  dd} t| trkttj} | durwgggdfS|| 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 typerxryFz&Tuple expected as "NamedTuple()" fieldT)rar&r~rprr"rrr rOrPrYr rr[r;r2r7 from_error) rUrrrprqrre type_noderrhrVrVrWrs>             z5NamedTupleAnalyzer.parse_namedtuple_fields_with_typesrp list[str]rq list[Type]rrMapping[str, Expression]rgintrtTypeInfo | Nonec s&|jdttj}|jd|g}|jd|g} |jd|g} |jd|g} |jd|g} |jdfdd|D} t| |}|pR|j|| d_t|| j rqt j j rq|jj dd j kd _d |ijd <t sts|jtfd d d@dAfdd }ddt||D}|D]}||ddqddt||D}tfdd|D|}|td|dd|td| dd|td | dd|td!dd|td"| dd|td#dd|jjd$kr|td%|ddj dusJttjd&t|jjgj ttjd'  d@dBfd0d1 }|d2d3d|Dd4dCfd6d7 |d8fd9d|Ddd:|d;g| d<|d=dttd>| | dt gd?t!tjd&tgj ttj}t"t#|j$t<S)DNz builtins.strzbuiltins.tuplez builtins.dictztyping.Iterablezbuiltins.functioncsg|]}t|qSrV)r4rstrtyperVrWrrz@NamedTupleAnalyzer.build_namedtuple_typeinfo..TzNamedTuple item)force_progressfieldsrcstSrT)r.rV) tuple_baserVrWsz>NamedTupleAnalyzer.build_namedtuple_typeinfo..Fvarr)is_initialized_in_classrZ is_propertyrRrScs<|_||_||_jd|j|_tt|j|j<dS)N.) rirrrcre _fullnamer$rnames)rrr)rirVrW add_fields z?NamedTupleAnalyzer.build_namedtuple_typeinfo..add_fieldcSg|] \}}t||qSrVr)rrtyprVrVrWrr)rcSrrVrrrVrVrWrrcsg|]}qSrVrVrrrVrWrsrB)rrDrCrHrI__doc__) __match_args__r)rercidvalues upper_boundrfuncnamerfretr6rlist[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_selfcSrrV)type_annotationrrrVrVrWr;rzTNamedTupleAnalyzer.build_namedtuple_typeinfo..add_method..cSsg|]}|jjqSrV)variablererrVrVrWr<cSrrV)kindrrVrVrWr=rrT classmethod)rr)r8make_normalizedrr3rrr6 variablesrrriis_classr0rrcrrgr is_decoratedrrr$rplugin_generatedr)rrrrrfirstrqrpr signaturefuncvdecsym) function_typerirgselftypetvdrVrW add_method.s8     z@NamedTupleAnalyzer.build_namedtuple_typeinfo..add_methodrFcSsg|] }t||jttqSrV)rrrr rrrVrVrWrXs)rrrcs.|jd}|dur tnt}t||j||SrT)getrerr rr)rrr)rrrVrW make_init_arg[szCNamedTupleAnalyzer.build_namedtuple_typeinfo..make_init_argr>csg|]}|qSrVrVr)r rVrWr`r)rrrrG)rrrEiterable)rrr)FF)rr)rrZrrZrRrS) rrfrr6rrrrZrrZrRrS)rr)rRr)%rP named_typer2r7 special_formnamed_type_or_noner5basic_new_typeinfois_named_tuple special_aliasr/targetprocess_placeholder tuple_typeupdate_tuple_typergcopymetadatar<schedule_patchr,rr)rOpython_versionr:rMrc tvar_scopenew_unique_func_idfrom_omitted_genericsrrr(r$rr)rUrerprqrrrgrt implicit_anybasetuple_typedictypeordereddictypefallback iterable_typeliteralsmatch_args_typerrrvarstuple_of_stringsrself_tvar_exprrV) rrrrirgr rrrrrWrjs         '  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#rJr rrrprarrrr=) rUr(nt_names prohibitedctxrrrr_keyrVrVrWsave_namedtuple_bodyss.         z'NamedTupleAnalyzer.save_namedtuple_bodyr,cCs|j||dSrT)rPr)rUrQr,rVrVrWrszNamedTupleAnalyzer.failN)rOr+rPr-rQr rRrS)rXrrYrZr[rZrRr\)rXrrYrZrRrv)rrrrr[rZrRr) rir'rerfrrr_rZrRrS)rrrcrfrRr)rrrrrRr)rerfrprrqrrrrrgrrtrrRr')r(r'rRr))rQrfr,rrRrS)__name__ __module__ __qualname__r?rurdrrrrrjrr.rrVrVrVrWrN^s   # P m  c + )rNN)Lr __future__r contextlibrtypingrrrrrmypy.exprtotyper r mypy.messagesr mypy.nodesr r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r* mypy.optionsr+mypy.semanal_sharedr,r-r.r/r0 mypy.typesr1r2r3r4r5r6r7r8r9r:r;r< mypy.utilr=rJrIrKrMrNrVrVrVrWs     !8