o 2g%@sUdZddlmZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZddlmZm Z m!Z!e"dZ#d e$d <e"d d d dgedddddRZ%d e$d<GdddZ&dS)zySemantic analysis of call-based Enum definitions. This is conceptually part of mypy.semanal (semantic analyzer pass 2). ) annotations)Finalcast) ARG_NAMEDARG_POSMDEFAssignmentStmtCallExprContextDictExpr EnumCallExpr ExpressionListExpr MemberExprNameExprRefExprStrExprSymbolTableNode TupleExprTypeInfoVaris_StrExpr_list)Options)SemanticAnalyzerInterface)ENUM_REMOVED_PROPS LiteralTypeget_proper_type)z enum.Enumz enum.IntEnumz enum.Flagz enum.IntFlagz enum.StrEnumr ENUM_BASESnamevalue_name__value_ __module____annotations____doc__ __slots____dict__ENUM_SPECIAL_PROPSc@sReZdZd/ddZd0d dZd1ddZd2ddZd3d#d$Zd4d(d)Zd5d,d-Z d.S)6EnumCallAnalyzeroptionsrapirreturnNonecCs||_||_dSN)r)r*)selfr)r*r/I/home/garg/my-data/venv/lib/python3.10/site-packages/mypy/semanal_enum.py__init__:s zEnumCallAnalyzer.__init__sr is_func_scopeboolcCst|jdkst|jdttfsdS|jd}|j}||j||}|dur)dSt|tr6|d|dS|j |||dS)zCheck if s defines an Enum; if yes, store the definition in symbol table. Return True if this looks like an Enum definition (but maybe with errors), otherwise return False. rFNz'Enum type as attribute is not supportedT) lenlvalues isinstancerrrcheck_enum_callrvaluefailr* add_symbol)r.r2r3lvaluer enum_callr/r/r0process_enum_call>s"   z"EnumCallAnalyzer.process_enum_callnoder var_namestrTypeInfo | NonecCs6t|tsdS|}|j}t|tsdS|j}|tvrdS|||dd\}}} } | sD|} |r:| dt|j 7} | | g||j } n7||krZd|d|d|d} | | |t t |jd j} | |ksi|rr| dt|j 7} | | |||j } | |ks|r|j| | t| || |_|j||j | _ | S) zCheck if a call defines an Enum. Example: A = enum.Enum('A', 'foo bar') is equivalent to: class A(enum.Enum): foo = 1 bar = 2 N.@zString argument 1 "z" to z$(...) does not match variable name ""r)r8r calleerfullnamerparse_enum_call_argssplitrBlinebuild_enum_call_typeinfor;rrargsrr*add_symbol_skip_localr analyzedset_line)r.r@rAr3callrHrInew_class_nameitemsvaluesokrinfomsgr/r/r0r9Rs<       z EnumCallAnalyzer.check_enum_callrrT list[str]rIrLintrc Cs||j|}|dus J|j|||}||_d|_|D]}t|}||_d|_|j d||_ t t ||j |<q|S)NTrD)r*named_type_or_nonebasic_new_typeinfocalculate_metaclass_typemetaclass_typeis_enumrrW is_propertyrI _fullnamerrnames) r.rrTrIrLbaserWitemvarr/r/r0rMs   z)EnumCallAnalyzer.build_enum_call_typeinforRr class_name4tuple[str, list[str], list[Expression | None], bool]cCs|j}tdd|jDs|d|d|St|dkr'|d|d|St|dkr7|d|d|Sgd }|jD]}||vrN|d |d |q>d \}}t|j|D]\}}|d krc|}|dkri|}qY|durr|d}|durz|d}t|ts||d|S|j } g} g} t|tr|j } | dd D]} | | qnt|t tfr|j}t|rdd|D} ntdd|Dr|D]"}t|t tfsJ|j\}}t|tsJ| |j | |qn|d||St|tr |jD] \}}t|ts||d|S| |j | |qnyt|dtrt|djtrt|djj}|dur`t|tr`t|j tr`|j } | dd D]} | | qVn9|djjrt|djjtr|djj} | dd D]} | | qn|d||S|d||S| s||d|S| sdgt| } t| t| ksJ| | | dfS)zhParse arguments of an Enum call. Return a tuple of fields, values, was there an error. css|] }|ttfvVqdSr-)rr).0arg_kindr/r/r0 sz8EnumCallAnalyzer.parse_enum_call_args..zUnexpected arguments to z()zToo few arguments for zToo many arguments for )NrrbmodulequalnametypestartzUnexpected keyword argument "rG)NNrrbNrr5z1() expects a string literal as the first argument, cSsg|]}|jqSr/)rrhseq_itemr/r/r0 sz9EnumCallAnalyzer.parse_enum_call_args..css<|]}t|ttfot|jdkot|jdtVqdS)rkrN)r8rrr6rTrrsr/r/r0rjs  z>%s() with tuple or list expects strings or (name, value) pairsz-() with dict literal requires string literalszfSecond argument of %s() must be string, tuple, list or dict literal for mypy to determine Enum membersz() needs at least one itemT)rNall arg_kindsfail_enum_call_argr6 arg_nameszipr8rrreplacerKappendrrrTrr rr@rrrorrBis_final final_value)r.rRrfrN valid_namearg_namerrbargrSrTrUfieldsfield seq_itemsrtrkey proper_typer/r/r0rJs               "  "  z%EnumCallAnalyzer.parse_enum_call_argsmessagecontextr cCs|||dggdfS)NF)r;)r.rrr/r/r0rxs  z#EnumCallAnalyzer.fail_enum_call_argrXctxcCs|j||dSr-)r*r;)r.rXrr/r/r0r;szEnumCallAnalyzer.failN)r)rr*rr+r,)r2rr3r4r+r4)r@r rArBr3r4r+rC) rrBrTrYrIrBrLrZr+r)rRr rfrBr+rg)rrBrr r+rg)rXrBrr r+r,) __name__r" __qualname__r1r?r9rMrJrxr;r/r/r/r0r(9s    4  gr(N)'r$ __future__rtypingrr mypy.nodesrrrrr r r r r rrrrrrrrrr mypy.optionsrmypy.semanal_sharedr mypy.typesrrr frozensetrr#r'r(r/r/r/r0s< T