s㱬 8 U d Z ddlmZ ddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZmZ dd lmZmZmZ dd
lmZ dd lmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z" dd l#m$Z$m%Z% ej& d
k rddlm'Z' nddl(m'Z' ddhZ)h dZ*de+d< ddhZ,de+d< dZ- e. e/edg Z0 e1d Z2dZ3d6d Z4d7d Z5d6d Z6 d7d8d%Z7d9d'Z8 d7d:d)Z9 d7d;d+Z:dd0Z=d d7d>d2Z?d dS )@z$Astroid hooks for typing.py support. ) annotationsN)Iterator)partial)context extract_node
inference_tip)_extract_single_node) PY38_PLUS PY39_PLUS)AttributeInferenceErrorInferenceErrorUseInferenceDefault)AstroidManager)
Assign
AssignName AttributeCallConst JoinedStrNameNodeNG SubscriptTuple)ClassDef FunctionDef) )FinalTypeVarNewType> typing.NewTypetyping.TypeVartyping_extensions.TypeVarr TYPING_TYPEVARS_QUALIFIEDztyping.TypedDictztyping_extensions.TypedDictTYPING_TYPEDDICT_QUALIFIEDz
class Meta(type):
def __getitem__(self, item):
return self
@property
def __args__(self):
return ()
class {0}(metaclass=Meta):
pass
__all__)(ztyping.Hashableztyping.Awaitableztyping.Coroutineztyping.AsyncIterableztyping.AsyncIteratorztyping.Iterableztyping.Iteratorztyping.Reversiblez typing.Sizedztyping.Containerztyping.Collectionztyping.Callableztyping.AbstractSetztyping.MutableSetztyping.Mappingztyping.MutableMappingztyping.Sequenceztyping.MutableSequenceztyping.ByteStringz typing.Tuplez typing.Listz typing.Dequez
typing.Setztyping.FrozenSetztyping.MappingViewztyping.KeysViewztyping.ItemsViewztyping.ValuesViewztyping.ContextManagerztyping.AsyncContextManagerz typing.Dictztyping.DefaultDictztyping.OrderedDictztyping.Counterztyping.ChainMapztyping.Generatorztyping.AsyncGeneratorz typing.Typeztyping.Patternz typing.Matchz?
@classmethod
def __class_getitem__(cls, item):
return cls
returnboolc | j }t |t r|j t v S t |t
r|j t v S dS )NF)func
isinstancer attrnameTYPING_TYPEVARSr name)noder* s l/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/astroid/brain/brain_typing.py$looks_like_typing_typevar_or_newtyper1 s sL 9D$ "" 0}//$ ,yO++ 5 c $ t | j | }n!# t t f$ r
}t
|d}~ww xY w| t vrt
| j st
t | j d t rt
| j d d }t t | } | | S )z8Infer a typing.TypeVar(...) or typing.NewType(...) call.r Nr ')nextr* inferr
StopIterationr qnamer$ argsr+ r as_stringstripr TYPING_TYPE_TEMPLATEformat)r/
context_ittonr* exctypenames r0 infer_typing_typevar_or_newtyperB | s +DIOOMO::;; M * + + +!s*+ zz||444!! 9 "!!$)A, ** "!!y|%%''--c22H ,33H== > >D ::m: , ,,s (+ A AA c t | t r| j t v S t | t r| j t v S t | t rt | j S dS )zLTry to figure out if a Subscript node *might* be a typing-related subscript.F) r+ r r. TYPING_MEMBERSr r, r _looks_like_typing_subscriptvaluer/ s r0 rE rE sg $ +yN**$ "" /}..$ "" 8+DJ777 5r2 r/ r ctx context.InferenceContext | NoneIterator[ClassDef]c t | j }n!# t t f$ r
}t
|d}~ww xY w| d r| t v rt
t |t r| dv rt t }|g|j
d< t | j t r\| | j j v rNt! | j dd r8| j j }| | j j
|| j j = t) |g S t+ t, | d d } | | S ) z Infer a typing.X[...] subscript.Nztyping.> typing.Generictyping.Annotatedtyping_extensions.Annotated__class_getitem____cache.r4 )r6 rF r7 r
r8 r r9
startswith TYPING_ALIASr+ r r CLASS_GETITEM_TEMPLATElocalsparentbasesgetattrrP getslotsiterr r= r> split)r/ rH rF r@ func_to_addcaches r0 infer_typing_attrr` s +TZ%%''(( M * + + +!s*+ ;;== # #I . . "%++--<2O2O!!%"" u{{}} 9 ( ( ++ABB -8M () t{H - - - ))) Y55 *
K'Eyy*++7$++,UG}} ,33EKKMM4G4G4L4LR4PQQ R RD ::c: " ""s &) AAAFunctionDef | ClassDefc 8 | t v S )z'Check if node is TypedDict FunctionDef.)r9 r% rG s r0 _looks_like_typedDictrc s ::<<5 55r2 r c V t d }|g| j d< t | g S )Ndict__call__)r rV r\ )r/ rH r^ s r0 infer_old_typedDictrg s- 'v..K *mDK
<<r2 r c t d| j | j | j }| t d gg d t
d }|g|j d< t |g S )z,Replace TypedDict FunctionDef with ClassDef. TypedDictr. lineno
col_offsetrW re NrX body
decoratorsrf ) r rk rl rW postinitr r rV r\ )r/ rH class_defr^ s r0 infer_typedDictrr s~
{?{ I l6 2 2 3"NNN&v..K$/=IZ r2 r c t | j t o6| j j dk o&t | j d t
t f S )z
Returns True if the node corresponds to a call to _alias function.
For example :
MutableSet = _alias(collections.abc.MutableSet, T)
:param node: call node
_aliasr )r+ r* r r. r: r rG s r0 _looks_like_typing_aliasru sH 49d##
INh &
ty|i%6 7 7 r2 Nonec d } | d t || j }|| _ dS # t $ r Y dS w xY w)zJDisable the access to __class_getitem__ method for the node in parameters.c D |dk rt d | |g|R i |S )z
Raises an AttributeInferenceError in case of access to __class_getitem__ method.
Otherwise, just call origin_func.
rO z'__class_getitem__ access is not allowed)r ) origin_funcattrr: kwargss r0 full_raiserz1_forbid_class_getitem_access..full_raiser s?
& & &)*STT T{41$111&111r2 rO N)rY r r )r/ r| partial_raisers r0 _forbid_class_getitem_accessr~ si 2 2 2
()))
!dl;;% "
s 18
AAc t | j t rBt | j j dk r%t | j j d t
st t | j d | }n/# t $ r"}t | j d | |d}~ww xY w| j j d }t |j
|j |j | j }t |t r| |gg d | j d }t" s t |t$ r.|j s't" r@t |t( r+|j dk r t- t. }|g|j d< nt3 | t5 |g S ) z
Infers the call to _alias function
Insert ClassDef, with same name as aliased class,
in mro to simulate _GenericAlias.
:param node: call node
:param context: inference context
r r4 r/ r Nrj rm rO )r+ rW r lentargetsr r r6 r: r7 r8 r
r r. rk rl rp r r eltsr rF r rU rV r~ r\ )r/ rH rese assign_namerq maybe_type_varr^ s r0 infer_typing_aliasr
s
t{F + +"4;&''1,,$+-a0*== - "!D49Q<%%c%2233 D D D $)A,<<>K-8MI() r c t | t oSt | j t r| j j dk p)t | j t
o| j j dk S )Ncast)r+ r r* r r. r r, rG s r0 _looks_like_typing_castr ~ s_ dD ! ! 49d## % INf $ ) di + + ) I & ( r2 Iterator[NodeNG]c t | j t t f st t | j | }n!# t t f$ r
}t |d}~ww xY wt |t r0|
dk st | j dk rt | j d | S )z>Q !! 9Q< c * **s (A A1$A,,A1)r' r( )N)r/ r rH rI r' rJ )r/ ra r' r( )r/ r rH rI r' rJ )r/ r rH rI r' rJ )r/ r r' r( )r/ r r' rv )r/ r rH rI r' rJ )r/ r rH rI r' r )C__doc__
__future__r systypingcollections.abcr functoolsr astroidr r r astroid.builderr
astroid.constr
r astroid.exceptionsr r
r astroid.managerr astroid.nodes.node_classesr r r r r r r r r r astroid.nodes.scoped_nodesr r version_infor typing_extensionsr- r$ __annotations__r% r= setrY rD frozensetrT rU r1 rB rE r` rc rg rr ru r~ r r r r r register_transform r2 r0 r s)
+ * * " " " " " "
$ $ $ $ $ $ 8 8 8 8 8 8 8 8 8 8 0 0 0 0 0 0 . . . . . . . .
+ * * * * * = < < < < < < <v '''''' i($ $ $
&89V$W W W W W WWVY3344y)+ + Z - - - -( =A$# $# $# $# $#N6 6 6 6 <@ ?C
(
2 8<3 3 3 3 3 l D 8< 4 8<+ + + + +* # #M122(
# #
}} .//1M # #--)
*
*,C
N'']]?335J N''-- 3446K # #--*
+
+-E # #--+
,
,.G r2