U mdO @sUdZddlmZddlmZddlmZmZmZddl Z zbddl Z ddl m Z ddlmZmZddlmZmZmZdd lmZeege fZdZd ed <Wn&ek rZzeZW5dZ[XYnXgZd ed <ddddZd dddZddddZeZ ded<eZ!ded<eZ"ded<dddddZ#esDedkrd dd!d"d#d$Z$d%dd&d'd(d)d*Z%Gd+d,d,eZ&dd-d.d/d0Z'ndd-d.d1d0Z'dS)2aA 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_EXz list[str]__all__zdict[str, str])returnc Csdtjfdtjfdtjfdtjfdtjfdtjfdtjfdtjfd tj fd tj fg }i}|D],\}}d |j j }d |d |d|<q\|S)NZ _NBitByteZ _NBitShortZ _NBitIntCZ _NBitIntPZ_NBitIntZ _NBitLongLongZ _NBitHalfZ _NBitSingleZ _NBitDoubleZ_NBitLongDoubleznumpy._ZBitnumpy._typing._nbit.) npbyteshortZintcZintpint_longlongZhalfZsingledouble longdoubledtypeitemsize)namesretnametypnr!Q/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numpy/typing/mypy_plugin.py_get_precision_dict8s   r#c sBtjtjtjtjg}ddddddddd d d d h fd d|DS)NZuint128Zuint256Zint128Zint256Zfloat80Zfloat96Zfloat128Zfloat256Z complex160Z complex192Z complex256Z complex512csg|]}|jkr|jqSr!)__name__).0iZextended_namesr!r" ]s z0_get_extended_precision_list..)rZ ulonglongrrZ clongdouble)Zextended_typesr!r'r"_get_extended_precision_listMsr)strcCs8tdj}|dkrdS|dkr$dS|dkr0dSdSdS)Npr&c_intlZc_longqZ c_longlong)rrchar)r/r!r!r"_get_c_intp_name`s r0r_PRECISION_DICT_EXTENDED_PRECISION_LIST_C_INTPr r)ctxrcCs2|\}}}|jdd}td|}||S)z;Replace a type-alias with a concrete ``NBitBase`` subclass..r)rsplitr1Z named_type)r4r_apirZname_newr!r!r"_hookxs r:zIterable[Statement]int)iterableidrcCs<t|D] \}}t|dd|kr|Sqtd|dS)z>Identify the first ``ImportFrom`` instance the specified `id`.r=NzBFailed to identify a `ImportFrom` instance with the following id: ) enumerategetattr ValueError)r<r=r&valuer!r!r"_indexs rBr zlist[tuple[str, None | str]]None)filemoduleimportsrcCs<t|d|d}d|_|j|jfD]}t||}|||<q dS)zA mypy plugin for handling versus numpy-specific typing tasks.r*zNone | _HookFunc)fullnamercCs|tkr tSdS)zSet the precision of platform-specific `numpy.number` subclasses. For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`. N)r1r:)selfrJr!r!r"get_type_analyze_hooksz"_NumpyPlugin.get_type_analyze_hookr zlist[tuple[int, str, int]])rDrcCsTt|jdfg}|jdkr2t|dddtDdn|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`. r6numpyz!numpy._typing._extended_precisioncSsg|] }||fqSr!r!)r%vr!r!r"r(sz4_NumpyPlugin.get_additional_deps..)rFznumpy.ctypeslibctypesZ_c_intp)r rJrHr2r3)rKrDrr!r!r"get_additional_depss    z _NumpyPlugin.get_additional_depsN)r$ __module__ __qualname____doc__rLrPr!r!r!r"rIs rIztype[_NumpyPlugin])versionrcCstS)An entry-point for mypy.)rIrTr!r!r"pluginsrWcCstdS)rUN)rrVr!r!r"rWs)(rS __future__rcollections.abcrtypingrrrrMrZ mypy.typesZmypyrZ mypy.pluginrr Z mypy.nodesr r r Z mypy.buildr Z _HookFuncr__annotations__ModuleNotFoundErrorexrr#r)r0r1r2r3r:rBrHrIrWr!r!r!r"s8!     &