mypy_plugin.pyc
1 o 2 \��c � 3 @ sp U d Z ddlmZ ddlmZ ddlmZmZmZ ddl Z 4 z1ddl Z ddl m Z ddlmZmZ ddlmZmZmZ dd lmZ eege f ZdZd 5 ed < W n eyb Z zeZW Y dZ[ndZ[ww g Z d ed < d5dd�Z d6dd�Z d7dd�Z e � Z ded< e � Z!ded< e � Z"ded< d8d d �Z#es�edu r�d9d$d%�Z$d:d,d-�Z%G d.d/� d/e�Z&d;d2d3�Z'dS d;d4d3�Z'dS )<a A mypy_ plugin for managing a number of platform-specific annotations. 6 Its functionality can be split into three distinct parts: 7 8 * Assigning the (platform-dependent) precisions of certain `~numpy.number` 9 subclasses, including the likes of `~numpy.int_`, `~numpy.intp` and 10 `~numpy.longlong`. See the documentation on 11 :ref:`scalar types <arrays.scalars.built-in>` for a comprehensive overview 12 of the affected classes. Without the plugin the precision of all relevant 13 classes will be inferred as `~typing.Any`. 14 * Removing all extended-precision `~numpy.number` subclasses that are 15 unavailable for the platform in question. Most notably this includes the 16 likes of `~numpy.float128` and `~numpy.complex256`. Without the plugin *all* 17 extended-precision types will, as far as mypy is concerned, be available 18 to all platforms. 19 * Assigning the (platform-dependent) precision of `~numpy.ctypeslib.c_intp`. 20 Without the plugin the type will default to `ctypes.c_int64`. 21 22 .. versionadded:: 1.22 23 24 Examples 25 -------- 26 To enable the plugin, one must add it to their mypy `configuration file`_: 27 28 .. code-block:: ini 29 30 [mypy] 31 plugins = numpy.typing.mypy_plugin 32 33 .. _mypy: http://mypy-lang.org/ 34 .. _configuration file: https://mypy.readthedocs.io/en/stable/config_file.html 35 36 � )� annotations)�Iterable)�Final� TYPE_CHECKING�CallableN)�Type)�Plugin�AnalyzeTypeContext)�MypyFile� 37 ImportFrom� Statement)�PRI_MEDzNone | ModuleNotFoundError�MYPY_EX� list[str]�__all__�return�dict[str, str]c C s� dt jfdt jfdt jfdt jfdt jfdt jfdt jfdt jfd t j fd 38 t j 39 fg 40 } i }| D ]\}}d |� j j }d |� d �|d|� �< q.|S )N� _NBitByte� 41 _NBitShort� _NBitIntC� _NBitIntP�_NBitInt� _NBitLongLong� _NBitHalf� _NBitSingle� _NBitDouble�_NBitLongDouble� znumpy._�Bit�numpy._typing._nbit.) �np�byte�short�intc�intp�int_�longlong�half�single�double� 42 longdouble�dtype�itemsize)�names�ret�name�typ�n� r2 ��C:\Users\Jacks.GUTTSPC\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\numpy\typing\mypy_plugin.py�_get_precision_dict8 s � r4 c s. t jt jt jt jg} h d�� � fdd�| D �S )N> �uint128�float80�float256� 43 complex512� 44 complex256� 45 complex160�float96�int128�float128� 46 complex192�int256�uint256c s g | ] 47 }|j � v r|j �qS r2 )�__name__)�.0�i�Zextended_namesr2 r3 � 48 <listcomp>] s z0_get_extended_precision_list.<locals>.<listcomp>)r � ulonglongr& r* � clongdouble)Zextended_typesr2 rD r3 � _get_extended_precision_listM s rH �strc C s4 t �d�j} | dkr dS | dkrdS | dkrdS dS )N�prC �c_int�l�c_long�q� 49 c_longlong)r r+ �char)rP r2 r2 r3 �_get_c_intp_name` s rQ r �_PRECISION_DICT�_EXTENDED_PRECISION_LIST�_C_INTP�ctxr r c C s2 | \}}}|j �d�d }td|� � }|�|�S )z;Replace a type-alias with a concrete ``NBitBase`` subclass.�.�����r )r/ �splitrR Z 50 named_type)rU r0 �_�apir/ Zname_newr2 r2 r3 �_hookx s 51 52 r[ �iterable�Iterable[Statement]�id�intc C s8 t | �D ]\}}t|dd�|kr| S qtd|����)z>Identify the first ``ImportFrom`` instance the specified `id`.r^ NzBFailed to identify a `ImportFrom` instance with the following id: )� enumerate�getattr� 53 ValueError)r\ r^ rC �valuer2 r2 r3 �_index� s ��rd �filer 54 �module�imports� list[tuple[str, None | str]]�Nonec C s< t |d|d�}d|_| j| jfD ] }t||�}|||< qdS )z<Override the first `module`-based import with new `imports`.r )r- TN)r Z is_top_level�defsrg rd )re rf rg Z 55 import_obj�lstrC r2 r2 r3 �_override_imports� s 56 57 �rl c @ s$ e Zd ZdZddd�Zdd d �Zd S )� _NumpyPluginz>A mypy plugin for handling versus numpy-specific typing tasks.�fullnamerI r �None | _HookFuncc C s |t v rtS dS )z�Set the precision of platform-specific `numpy.number` 58 subclasses. 59 60 For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`. 61 N)rR r[ )�selfrn r2 r2 r3 �get_type_analyze_hook� s z"_NumpyPlugin.get_type_analyze_hookre r 62 �list[tuple[int, str, int]]c C sV t |jdfg}|jdkrt|ddd� tD �d� |S |jdkr)t|dtd fgd� |S ) 63 a. Handle all import-based overrides. 64 65 * Import platform-specific extended-precision `numpy.number` 66 subclasses (*e.g.* `numpy.float96`, `numpy.float128` and 67 `numpy.complex256`). 68 * Import the appropriate `ctypes` equivalent to `numpy.intp`. 69 70 rW �numpyz!numpy._typing._extended_precisionc S s g | ]}||f�qS r2 r2 )rB �vr2 r2 r3 rE � s z4_NumpyPlugin.get_additional_deps.<locals>.<listcomp>)rg znumpy.ctypeslib�ctypesZ_c_intp)r rn rl rS rT )rp re r. r2 r2 r3 �get_additional_deps� s 71 � 72 ��z _NumpyPlugin.get_additional_depsN)rn rI r ro )re r 73 r rr )rA � 74 __module__� __qualname__�__doc__rq rv r2 r2 r2 r3 rm � s 75 76 rm �version�type[_NumpyPlugin]c C s t S �zAn entry-point for mypy.)rm �rz r2 r2 r3 �plugin� � r~ c C s t �r| )r r} r2 r2 r3 r~ � r )r r )r r )r rI )rU r r r )r\ r] r^ rI r r_ )re r 77 rf rI rg rh r ri )rz rI r r{ )(ry � 78 __future__r �collections.abcr �typingr r r rs r � 79 mypy.types�mypyr � mypy.pluginr r � 80 mypy.nodesr 81 r r � 82 mypy.buildr � _HookFuncr �__annotations__�ModuleNotFoundError�exr r4 rH rQ rR rS rT r[ rd rl rm r~ r2 r2 r2 r3 �<module> s> ! �� 83 84 85 86 87 88 &