sCK) U d Z ddlmZ ddlmZ ddlmZmZmZ ddl Z
ddl Z ddl m
Z
ddlmZmZ ddlmZmZmZ dd lmZ eege
f ZdZd
ed < n# e$ r ZeZY dZ[ndZ[ww xY wg Z d ed
< d.dZ d/dZ d0dZ e Z ded< e Z!ded< e Z"ded< d1dZ#ese d2d Z$d3d'Z% G d( d)e Z&d4d,Z'dS d4d-Z'dS )5a A mypy_ plugin for managing a number of platform-specific annotations.
Its functionality can be split into three distinct parts:
* Assigning the (platform-dependent) precisions of certain `~numpy.number`
subclasses, including the likes of `~numpy.int_`, `~numpy.intp` and
`~numpy.longlong`. See the documentation on
:ref:`scalar types ` for a comprehensive overview
of the affected classes. Without the plugin the precision of all relevant
classes will be inferred as `~typing.Any`.
* Removing all extended-precision `~numpy.number` subclasses that are
unavailable for the platform in question. Most notably this includes the
likes of `~numpy.float128` and `~numpy.complex256`. Without the plugin *all*
extended-precision types will, as far as mypy is concerned, be available
to all platforms.
* Assigning the (platform-dependent) precision of `~numpy.ctypeslib.c_intp`.
Without the plugin the type will default to `ctypes.c_int64`.
.. versionadded:: 1.22
Examples
--------
To enable the plugin, one must add it to their mypy `configuration file`_:
.. code-block:: ini
[mypy]
plugins = numpy.typing.mypy_plugin
.. _mypy: http://mypy-lang.org/
.. _configuration file: https://mypy.readthedocs.io/en/stable/config_file.html
) annotations)Iterable)Final
TYPE_CHECKINGCallableN)Type)PluginAnalyzeTypeContext)MypyFile
ImportFrom Statement)PRI_MEDzNone | ModuleNotFoundErrorMYPY_EX list[str]__all__returndict[str, str]c h dt j fdt j fdt j fdt j fdt j fdt j fdt j fdt j fd t j fd
t j
fg
} i }| D ](\ }}d | j j z }d | d
|d| < )|S )N _NBitByte
_NBitShort _NBitIntC _NBitIntP_NBitInt
_NBitLongLong _NBitHalf _NBitSingle _NBitDouble_NBitLongDouble znumpy._Bitnumpy._typing._nbit.)
npbyteshortintcintpint_longlonghalfsingledouble
longdoubledtypeitemsize)namesretnametypns l/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/typing/mypy_plugin.py_get_precision_dictr5 8 s bg rx bg bg RW "+& bg " " BM*
E
C > > cSSUU[))-=q-=-=-= )4 ) )** J c t j t j t j t j g} h dfd| D S )N> int128int256float80float96uint128uint256float128float256
complex160
complex192
complex256
complex512c 0 g | ]}|j v |j S )__name__).0iextended_namess r4
z0_get_extended_precision_list..] s' O O O1!*2N2NAJ2N2N2Nr6 )r" ulonglongr( r, clongdouble)extended_typesrI s @r4 _get_extended_precision_listrN M sG lBK ON
N P O O O O O OOr6 strc h t j d j } | dk rdS | dk rdS | dk rdS dS )NprH c_intlc_longq
c_longlong)r" r- char)rW s r4 _get_c_intp_namerX ` sA
8C== D s{{w
x
|xr6 r _PRECISION_DICT_EXTENDED_PRECISION_LIST_C_INTPctxr
r c | \ }}}|j d d }t d| }| | S )z;Replace a type-alias with a concrete ``NBitBase`` subclass..r! )r1 splitrY
named_type)r\ r2 _apir1 name_news r4 _hookre x sJ KCC 8>># r "D >( # ##r6 iterableIterable[Statement]idintc t | D ] \ }}t |dd |k r|c S t d| )z>Identify the first ``ImportFrom`` instance the specified `id`.rh NzBFailed to identify a `ImportFrom` instance with the following id: ) enumerategetattr
ValueError)rf rh rH values r4 _indexro sf !(++ HAuudD))R// 0 :35: : ; ; ;r6 filer moduleimports list[tuple[str, None | str]]Nonec t |d| }d|_ | j | j fD ]}t || }|||< dS )zA mypy plugin for handling versus numpy-specific typing tasks.fullnamerO r None | _HookFuncc & |t v rt S dS )zSet the precision of platform-specific `numpy.number`
subclasses.
For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`.
N)rY re )selfr} s r4 get_type_analyze_hookz"_NumpyPlugin.get_type_analyze_hook s ?** 4r6 rp r list[tuple[int, str, int]]c t |j dfg}|j dk r"t |dd t D n%|j dk rt |dt dfg |S ) a. Handle all import-based overrides.
* Import platform-specific extended-precision `numpy.number`
subclasses (*e.g.* `numpy.float96`, `numpy.float128` and
`numpy.complex256`).
* Import the appropriate `ctypes` equivalent to `numpy.intp`.
r_ numpyz!numpy._typing._extended_precisionc g | ]}||fS rE rE )rG vs r4 rJ z4_NumpyPlugin.get_additional_deps.. s F F FaV F F Fr6 )rr znumpy.ctypeslibctypes_c_intp)r r} rz rZ r[ )r rp r0 s r4 get_additional_depsz _NumpyPlugin.get_additional_deps s T]B/0C}''!= F F-E F F F "333!( %y 1 2 Jr6 N)r} rO r r~ )rp r r r )rF
__module__ __qualname____doc__r r rE r6 r4 r| r| sB LL r6 r| versiontype[_NumpyPlugin]c t S zAn entry-point for mypy.)r| r s r4 pluginr s r6 c t r )r r s r4 r r s
r6 )r r )r r )r rO )r\ r
r r )rf rg rh rO r ri )rp r rq rO rr rs r rt )r rO r r )(r
__future__r collections.abcr typingr r r r r"
mypy.typesmypyr mypy.pluginr r
mypy.nodesr r r
mypy.buildr _HookFuncr __annotations__ModuleNotFoundErrorexr r5 rN rX rY rZ r[ re ro rz r| r rE r6 r4 r s B # " " " " " $ $ $ $ $ $ 1 1 1 1 1 1 1 1 1 1
66666666::::::::::"""""",-t34I*.G..... GGGGGG *P P P P&
-,.. . . . . #?">"@"@ @ @ @ @ "!## # # # #$ $ $ $ EGO; ; ; ;
$ $ $ $ $v $ $ $L
s 4A A%A A%