X-Ph!2dZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZddlm Z hdZ!d5d Z"d5dZ#d6dZ$d7dZ%d5dZ&d6dZ'd8dZ(d9dZ)d:dZ*d;dZ+dd(Z.d?d,Z/d@d/Z0dAd0Z1dBd3Z2d4S)CzFVarious utilities that don't depend on other modules in mypyc.irbuild.) annotations)Any) ARG_NAMED ARG_NAMED_OPTARG_OPTARG_POSGDEFArgKind BytesExprCallExprClassDef Decorator Expression FloatExprFuncDefIntExprNameExprOverloadedFuncDefRefExprStrExpr TupleExpr UnaryExprVar)refers_to_fullname)FINAL_DECORATOR_NAMES)Errors>attr.s attr.attrsdataclasses.dataclassdrreturnboolc,t|tSN)rrr s R/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypyc/irbuild/util.pyis_final_decoratorr'%s a!6 7 77cBt|to |jdkS)Nzmypy_extensions.trait) isinstancerfullnamer%s r&is_trait_decoratorr,)s a ! ! Kaj4K&KKr(cdefr cVtd|jDp |jjS)Nc34K|]}t|VdSr$)r,.0r s r& zis_trait...s+>>!!$$>>>>>>r()any decoratorsinfo is_protocolr-s r&is_traitr8-s+ >>do>>> > > W$)BWWr( str | Nonec t|tr.|jtvr |jddSt|t rt|jtr|jjtvr}|jjdd}|dkrPd|jvrG|j|j d}t|tr |j dkrdS|SdS)N.rattr auto_attribsTruez attr-auto) r*rr+DATACLASS_DECORATORSsplitr callee arg_namesargsindexrname)r rEautos r&dataclass_decorator_typerG1s!W!*0D"D"Dz$$Q''1h qx ) ) H !5 5 5x &&s++A. 6>>n ;;6!+++N;;td|jDS)Nc34K|]}t|VdSr$)rIr0s r&r2zis_dataclass..Js+BBQ%a((BBBBBBr()r3r4r7s r& is_dataclassrLIs! BB$/BBB B BBr(cD|jD]}t|}||cSdSr$)r4rG)r-r typs r&dataclass_typerOOs8 _&q)) ?JJJ  4r(erc.t|tttfr|jSt|t r |jdkrdSt|t r |jdkrdSt|t r |jdkrdStS)zcConvert an expression from a mypyc_attr decorator to a value. Supports a pretty limited range. builtins.TrueTbuiltins.FalseF builtins.NoneN)r*rrrvaluerr+NotImplementedrPs r&get_mypyc_attr_literalrXWs!gw 233w Aw  AJ/$A$At Aw  AJ2B$B$Bu Aw  AJ/$A$At r(CallExpr | Nonect|tr,t|jtr|jjdkr|SdS)zCCheck if an expression is a call to mypyc_attr and return it if so.zmypy_extensions.mypyc_attrN)r*r rArr+r%s r&get_mypyc_attr_callr[fsG 1h qx ) ) H != = = 4r(stmtClassDef | Decoratordict[str, Any]ci}|jD]g}t|}|rTt|j|jD]9\}}| t |t r d||j<'t|||<:h|S)zJCollect all the mypyc_attr attributes on a class definition or a function.NT) r4r[ziprBrCr*rrUrX)r\attrsdecr rEargs r&get_mypyc_attrsrdqsE>>  $ $  > af55 > > c<!#w//0+/ci("8"="=E$KK Lr(pathstrerrorsrct|||}|durdSt|\}}|dur!|s|d|||j|S)NFTzEClass is marked as native_class=True but it can't be a native class. )get_explicit_native_classis_implicit_extension_classerrorline)rer-rgexplicit_native_classimplicit_extension_classreasons r&is_extension_classrps~5dD&II%%u'B4'H'H$f$$-E$ \TZ \ \  I   $#r( bool | Nonec~|jD]}t|}|st|jD]\}}|dkr |j|}t |t s |d||jdS|j dkrdS|j dkrdS|d||jdSdS)zReturn value of @mypyc_attr(native_class=True/False) decorator. Look for a @mypyc_attr decorator with native_class=True/False and return the value assigned or None if it doesn't exist. Other values are an error. native_classz1native_class must be used with True or False onlyNFalseFr>T) r4r[ enumeraterBrCr*rrkrlrE)rer-rgr mypyc_attr_callirErcs r&riris_-a00   !:;;  GAt~%%!&q)Cc8,,  PRVX\Xabbbtttx7""uuuV##ttt PRVX\Xabbbttt  4r(tuple[bool, str]c|jD]A}t|s0t|s!t|st |sdSB|jjrdS|jjrdS|jjr|jjj j dvrdSdS)zXCheck if class can be extension class and return a user-friendly reason it can't be one.)FzUClasses that have decorators other than supported decorators can't be native classes.)Fz*TypedDict classes can't be native classes.)Fz+NamedTuple classes can't be native classes.)z abc.ABCMetaztyping.TypingMetaztyping.GenericMeta)Fz_Classes with a metaclass other than ABCMeta, TypingMeta or GenericMeta can't be native classes.)T) r4r,rIr[r'r5typeddict_typeis_named_tuplemetaclass_typetyper+)r-r s r&rjrjs_  "1%% *1-- (** 'q))    yCBB yDCC y  DI$<$A$JS%%   8r(op'FuncDef | Decorator | OverloadedFuncDefrct|tr|jsJ|j}t|tr|j}|Sr$)r*rimplrfunc)rs r& get_func_defrsE"'((w W"i   W Ir(kindr cN|tkrtS|tkrtS|S)z>Find the concrete version of an arg kind that is being passed.)rrrr)rs r&concrete_arg_kindrs& w    r(ct|ttttfpt|t o+|jdko t|jttfpt|totd|j DpSt|to>|j tko.|jdvp%t|jt o |jjS)aSCheck whether we allow an expression to appear as a default value. We don't currently properly support storing the evaluated values for default arguments and default attribute values, so we restrict what expressions we allow. We allow literals of primitives types, None, and references to Final global variables. -c34K|]}t|VdSr$) is_constant)r1rPs r&r2zis_constant..s(,M,M[^^,M,M,M,M,M,Mr()rRrSrT)r*rr rrrrexprrallitemsrrr r+noderis_finalrWs r&rrs 1w 7I>?? q) $ $ a aAFWV_L`9a9a q) $ $ M,M,MQW,M,M,M)M)M  q' " " $  RRAqvs++? r(rUbytescnt|dddS)zConvert a string representing bytes into actual bytes. This is needed because the literal characters of BytesExpr (the characters inside b'') are stored in BytesExpr.value, whose type is 'str' not 'bytes'. utf8zunicode-escapezraw-unicode-escape)rdecodeencode)rUs r&bytes_from_strrs2    & &'7 8 8 ? ?@T U UUr(N)r rr!r")r-r r!r")r rr!r9)r-r r!r9)rPrr!r)r rr!rY)r\r]r!r^)rerfr-r rgrr!r")rerfr-r rgrr!rq)r-r r!rx)rrr!r)rr r!r )rPrr!r")rUrfr!r)3__doc__ __future__rtypingr mypy.nodesrrrrr r r r r rrrrrrrrrrrr mypy.semanalr mypy.typesr mypyc.errorsrr?r'r,r8rGrIrLrOrXr[rdrprirjrrrrr(r&rsLL"""""".,+++++,,,,,,HHH8888LLLLXXXX(3333CCCC          $$$$,>B0VVVVVVr(