U
|
P ±d ã
|
@ s U d Z ddlmZ ddlmZ ddlmZmZmZ ddl Z
|
zbddl Z ddl m Z ddlmZmZ ddlmZmZmZ dd lmZ eege f ZdZd
|
ed < W n& ek
|
r Z zeZW 5 dZ[X Y nX g Z d ed < ddddZ d dddZ ddddZ e Z ded< e Z!ded< e Z"ded< dd d d d Z#esDedkrd dd!d"d#d$Z$d%dd&d'd(d)d*Z%G d+d, d,eZ&dd-d.d/d0Z'ndd-d.d1d0Z'dS )2a 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 <arrays.scalars.built-in>` 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_CHECKINGÚCallableN)ÚType)ÚPluginÚAnalyzeTypeContext)ÚMypyFileÚ
|
ImportFromÚ Statement)ÚPRI_MEDzNone | ModuleNotFoundErrorÚMYPY_EXz list[str]Ú__all__zdict[str, str])Úreturnc C s dt jfdt jfdt jfdt jfdt jfdt jfdt jfdt jfd t j fd
|
t j
|
fg
|
} i }| D ],\}}d | j j }d | d |d| < q\|S )NZ _NBitByteZ
|
_NBitShortZ _NBitIntCZ _NBitIntPZ_NBitIntZ _NBitLongLongZ _NBitHalfZ _NBitSingleZ _NBitDoubleZ_NBitLongDoubleé znumpy._ZBitúnumpy._typing._nbit.) ÚnpÚbyteÚshortZintcZintpÚint_ÚlonglongZhalfZsingleÚdoubleÚ
|
longdoubleÚdtypeÚitemsize)ÚnamesÚretÚnameÚtypÚn© r! úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/typing/mypy_plugin.pyÚ_get_precision_dict8 s õ r# c sB t jt jt jt jg} ddddddddd d
|
d d h fd d| D S )NZuint128Zuint256Zint128Zint256Zfloat80Zfloat96Zfloat128Zfloat256Z
|
complex160Z
|
complex192Z
|
complex256Z
|
complex512c s g | ]}|j kr|j qS r! )Ú__name__)Ú.0Úi©Zextended_namesr! r" Ú
|
<listcomp>] s
|
z0_get_extended_precision_list.<locals>.<listcomp>)r Z ulonglongr r Z clongdouble)Zextended_typesr! r' r" Ú _get_extended_precision_listM s ôr) Ústrc C s8 t d¡j} | dkrdS | dkr$dS | dkr0dS dS d S )NÚpr& Úc_intÚlÚc_longÚqÚ
|
c_longlong)r r Úchar)r1 r! r! r" Ú_get_c_intp_name` s r2 r Ú_PRECISION_DICTÚ_EXTENDED_PRECISION_LISTÚ_C_INTPr r )Úctxr c C s2 | \}}}|j d¡d }td| }| |¡S )z;Replace a type-alias with a concrete ``NBitBase`` subclass.Ú.éÿÿÿÿr )r Úsplitr3 Z
|
named_type)r6 r Ú_Úapir Zname_newr! r! r" Ú_hookx s
|
r< zIterable[Statement]Úint)ÚiterableÚidr c C s< t | D ] \}}t|dd|kr| S qtd|dS )z>Identify the first ``ImportFrom`` instance the specified `id`.r? NzBFailed to identify a `ImportFrom` instance with the following id: )Ú enumerateÚgetattrÚ
|
ValueError)r> r? r& Úvaluer! r! r" Ú_index s
|
rD r
|
z list[tuple[str, None | str]]ÚNone)ÚfileÚmoduleÚimportsr c C s< t |d|d}d|_| j| jfD ]}t||}|||< q dS )z<Override the first `module`-based import with new `imports`.r )r TN)r Z is_top_levelZdefsrH rD )rF rG rH Z
|
import_objÚlstr& r! r! r" Ú_override_imports s
|
|
rJ c @ s0 e Zd ZdZdddddZddd d
|
d Zd S ) Ú _NumpyPluginz>A mypy plugin for handling versus numpy-specific typing tasks.r* zNone | _HookFunc)Úfullnamer c C s |t kr tS dS )z«Set the precision of platform-specific `numpy.number`
|
subclasses.
|
|
For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`.
|
N)r3 r< )ÚselfrL r! r! r" Úget_type_analyze_hook s z"_NumpyPlugin.get_type_analyze_hookr
|
zlist[tuple[int, str, int]])rF r c C sT t |jdfg}|jdkr2t|ddd tD d n |jdkrPt|dtd fgd |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`.
|
|
r8 Únumpyz!numpy._typing._extended_precisionc S s g | ] }||fqS r! r! )r% Úvr! r! r" r( µ s z4_NumpyPlugin.get_additional_deps.<locals>.<listcomp>)rH znumpy.ctypeslibÚctypesZ_c_intp)r rL rJ r4 r5 )rM rF r r! r! r" Úget_additional_deps¥ s
|
þ
|
þz _NumpyPlugin.get_additional_depsN)r$ Ú
|
__module__Ú __qualname__Ú__doc__rN rR r! r! r! r" rK s
|
rK ztype[_NumpyPlugin])Úversionr c C s t S )úAn entry-point for mypy.)rK ©rV r! r! r" Úplugin¾ s rY c C s t dS )rW N)r rX r! r! r" rY Ã s )(rU Ú
|
__future__r Úcollections.abcr Útypingr r r rO r Z
|
mypy.typesZmypyr Z mypy.pluginr r Z
|
mypy.nodesr
|
r r Z
|
mypy.buildr Z _HookFuncr Ú__annotations__ÚModuleNotFoundErrorÚexr r# r) r2 r3 r4 r5 r< rD rJ rK rY r! r! r! r" Ú<module> s8 ! &
|
