pPjY& UdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZddlmZm Z e!dZ"d e#d <e!d d d dgedddddZ$d e#d<GddZ%y)zySemantic analysis of call-based Enum definitions. This is conceptually part of mypy.semanal (semantic analyzer pass 2). ) annotations)Final) ARG_NAMEDARG_POSEXCLUDED_ENUM_ATTRIBUTESMDEFAssignmentStmtCallExprContextDictExpr EnumCallExpr ExpressionListExpr MemberExprNameExprRefExprStrExprSymbolTableNode TupleExprTypeInfoVaris_StrExpr_list)Options)SemanticAnalyzerInterface) 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_PROPScpeZdZd dZd dZd dZ d dZ d dZ ddZddZ y)EnumCallAnalyzerc ||_||_yN)optionsapi)selfr,r-s V/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/mypy/semanal_enum.py__init__zEnumCallAnalyzer.__init__;s cpt|jdk7s#t|jdttfsy|jd}|j }|j |j|}|yt|tr|jd|y|jj|||y)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. rFz'Enum type as attribute is not supportedT) lenlvalues isinstancerrrcheck_enum_callrvaluefailr- add_symbol)r.slvaluer enum_calls r/process_enum_callz"EnumCallAnalyzer.process_enum_call?s qyy>Q j1*?U&V1{{((48   fj ) II? H D)Q/r1ct|tsy|}|j}t|tsy|j}|t vry|j ||jdd\}}}} | s|j|g||j} nA||k7rd|d|d|d} |j| ||j||||j} |jjr|jj||| t| |||_|jj!||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 "")r6r calleerfullnamerparse_enum_call_argssplitbuild_enum_call_typeinfoliner9r-is_nested_within_func_scopeadd_global_symbolr analyzedset_line) r.nodevar_namecallrCrDnew_class_nameitemsvaluesokinfomsgs r/r7z EnumCallAnalyzer.check_enum_callSs2$)&'*?? : %,0,E,E (..%b)- )vr002xSD)+N+;5 Jnownxxyz #t$005(DIIVD 88 / / 1 HH & &xt <$T5&9  t$II  r1cj|jj|}|J|jj|||}|j|_d|_|D]U}t |}||_d|_d|_ |jd||_ tt||j|<W|S)NTr@)r-named_type_or_nonebasic_new_typeinfocalculate_metaclass_typemetaclass_typeis_enumrrT is_propertyhas_explicit_valuerD _fullnamerrnames) r.rrQrDrHbaserTitemvars r/rGz)EnumCallAnalyzer.build_enum_call_typeinfo}sxx**84xx**4t<";;=  :Dd)CCH"CO &*C "#}}oQtf5CM.tS9DJJt  : r1c> |j}td|jDs|jd|d|St |dkr|jd|d|St |dkDr|jd|d|Sgd}|j D]}||vs|jd |d |d \}}t |j |D]\}}|d k(r|}|d k(s|}||d}||d}t|ts|j|d|S|j} g} g} t|trD|j} | jddjD]} | j| nt|ttfr|j}t!|r|Dcgc]}|j} }ntd|Drn|D]g}t|ttfsJ|j\}}t|tsJ| j|j| j|in |jd|z|St|t"ri|jD]X\}}t|ts|j|d|cS| j|j| j|Zn}t|dt$rTt|dj&t(r6t+|dj&j,}|mt|t.r]t|jt0rC|j} | jddjD]} | j| n|dj&j2rwt|dj&j4t0rP|dj&j4} | jddjD]} | j| n*|jd|z|S|jd|z|S| s|j|d|S| sdgt | z} t | t | k(sJ| | | dfScc}w)zhParse arguments of an Enum call. Return a tuple of fields, values, was there an error. c38K|]}|ttfvywr+)rr).0arg_kinds r/ z8EnumCallAnalyzer.parse_enum_call_args..sS833SszUnexpected arguments to z()zToo few arguments for zToo many arguments for )Nrr_modulequalnametypestartzUnexpected keyword argument "rB)NNrr_Nrr3z1() expects a string literal as the first argument, c3K|]S}t|ttfxr7t|jdk(xrt|jdt Uyw)rhrN)r6rrr4rQr)reseq_items r/rgz8EnumCallAnalyzer.parse_enum_call_args..sY8i%:;;'1,;x~~a0':;sAAz>%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)argsall arg_kindsfail_enum_call_argr4 arg_nameszipr6rrreplacerFappendrrrQrr rrMrrrlrstris_final final_value)r.rO class_namerr valid_namearg_namerr_argrPrQrRfieldsfield seq_itemsrqrkey proper_types r/rEz%EnumCallAnalyzer.parse_enum_call_argssyySDNNSS**-Ej\QS+TVZ[ [ t9q=**-CJr7rGrErur9r1r/r)r):s((T )58@C ,e3e3*-e3 =e3N!!%,! =! r1r)N)&r$ __future__rtypingr mypy.nodesrrrrr r r r r rrrrrrrrrrr mypy.optionsrmypy.semanal_sharedr mypy.typesrr frozensetrr#r'r)rr1r/rs #,!93N E&  "           E"I I r1