o 7f$@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/J/home/ertert/spirit/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 | Nonec Cst|tsdS|}|j}t|tsdS|j}|tvrdS|||dd\}}} | sC|} |r9| dt|j 7} | | g||j } n!t t |j dj} | |ksR|r[| dt|j 7} | | |||j } | |ksj|rq|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.@r)r8r calleerfullnamerparse_enum_call_argssplitrBlinebuild_enum_call_typeinforrargsrr*add_symbol_skip_localr analyzedset_line) r.r@rAr3callrGrHitemsvaluesokrinfor/r/r0r9Rs2     z EnumCallAnalyzer.check_enum_callrrR list[str]rHrKintrc 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_enumrrU is_propertyrH _fullnamerrnames) r.rrRrHrKbaserUitemvarr/r/r0rLs   z)EnumCallAnalyzer.build_enum_call_typeinforQr class_name/tuple[list[str], list[Expression | None], bool]cCsx|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|Sg} 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]} | | qSn9|djjrt|djjtr|djj} | dd D]} | | q|n|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 )Nrr_modulequalnametypestartzUnexpected keyword argument "")NNrr_Nrr5z1() expects a string literal as the first argument, cSsg|]}|jqSr/)rreseq_itemr/r/r0 sz9EnumCallAnalyzer.parse_enum_call_args..css<|]}t|ttfot|jdkot|jdtVqdS)rhrN)r8rrr6rRrrqr/r/r0rgs  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)rMall arg_kindsfail_enum_call_argr6 arg_nameszipr8rrreplacerJappendrrrRrr rr@rrrlrrBis_final final_value)r.rQrcrM valid_namearg_namerr_argrRrSfieldsfield seq_itemsrrrkey proper_typer/r/r0rIs               "  "  z%EnumCallAnalyzer.parse_enum_call_argsmessagecontextr cCs|||ggdfS)NF)r;)r.rrr/r/r0rvs  z#EnumCallAnalyzer.fail_enum_call_argmsgctxcCs|j||dSr-)r*r;)r.rrr/r/r0r;szEnumCallAnalyzer.failN)r)rr*rr+r,)r2rr3r4r+r4)r@r rArBr3r4r+rC) rrBrRrVrHrBrKrWr+r)rQr rcrBr+rd)rrBrr r+rd)rrBrr r+r,) __name__r" __qualname__r1r?r9rLrIrvr;r/r/r/r0r(9s    -  er(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