o ZZhF @stUdZddlmZddlmZmZmZddlZerddl m Z z1ddl Z ddl m Z ddlmZmZddlmZmZmZdd lmZeege fZdZd ed <WneydZzeZWYdZ[ndZ[wwgZd ed <d5ddZd6ddZd7ddZeZ ded<eZ!ded<eZ"ded<d8ddZ#esedurd9d$d%Z$d:d,d-Z%Gd.d/d/eZ&d;d2d3Z'dSd;d4d3Z'dS)` 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: https://mypy-lang.org/ .. _configuration file: https://mypy.readthedocs.io/en/stable/config_file.html ) annotations)Final TYPE_CHECKINGCallableN)Iterable)Type)PluginAnalyzeTypeContext)MypyFile ImportFrom Statement)PRI_MEDzNone | ModuleNotFoundErrorMYPY_EX list[str]__all__returndict[str, str]c Csdtjfdtjfdtjfdtjfdtjfdtjfdtjfdtjfd tj fd tj fd tj fg }i}d }|D]\}}d |j j }|d|d||d|<q4|S)NZ _NBitByteZ _NBitShortZ _NBitIntCZ _NBitIntPZ_NBitIntZ _NBitLongZ _NBitLongLongZ _NBitHalfZ _NBitSingleZ _NBitDoubleZ_NBitLongDoublez numpy._typingz ._nbit_base._ZBitz._nbit.)npbyteZshortZintcZintpint_longZlonglongZhalfsingledoubleZ longdoubledtypeitemsize)namesretmodulenametypnr"G/var/www/auris/lib/python3.10/site-packages/numpy/typing/mypy_plugin.py_get_precision_dict:s$  r$cCsgd}dd|DS)N) Zuint128Zuint256Zint128Zint256Zfloat80Zfloat96Zfloat128Zfloat256Z complex160Z complex192Z complex256Z complex512cSsg|] }tt|r|qSr")hasattrr).0ir"r"r# `sz0_get_extended_precision_list..r")Zextended_namesr"r"r#_get_extended_precision_listQsr)strcCs4tdj}|dkr dS|dkrdS|dkrdSdS)Nr!r'c_intlZc_longqZ c_longlong)rrchar)r.r"r"r#_get_c_intp_namebs r/r_PRECISION_DICT_EXTENDED_PRECISION_LIST_C_INTPctxr rcCs2|\}}}|jdd}td|}||S)z;Replace a type-alias with a concrete ``NBitBase`` subclass..znumpy._typing._nbit.)rsplitr0Z named_type)r3r _apirZname_newr"r"r#_hookzs  r9iterableIterable[Statement]idintcCs8t|D]\}}t|dd|kr|Sqtd|)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#_indexsrBfiler rimportslist[tuple[str, None | str]]NonecCs<t|d|d}d|_|j|jfD] }t||}|||<qdS)zA mypy plugin for handling versus numpy-specific typing tasks.fullnamer*rNone | _HookFunccCs|tvrtSdS)zSet the precision of platform-specific `numpy.number` subclasses. For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`. N)r0r9)selfrJr"r"r#get_type_analyze_hooksz"_NumpyPlugin.get_type_analyze_hookrCr list[tuple[int, str, int]]cCsVt|jdfg}|jdkrt|dddtDd|S|jdkr)t|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`. r5numpyz!numpy._typing._extended_precisioncSsg|]}||fqSr"r")r&vr"r"r#r(sz4_NumpyPlugin.get_additional_deps..)rDznumpy.ctypeslibctypesZ_c_intp)r rJrHr1r2)rLrCrr"r"r#get_additional_depss   z _NumpyPlugin.get_additional_depsN)rJr*rrK)rCr rrN)__name__ __module__ __qualname____doc__rMrRr"r"r"r#rIs  rIversiontype[_NumpyPlugin]cCstSzAn entry-point for mypy.)rIrWr"r"r#pluginr[cCstrY)rrZr"r"r#r[r\)rr)rr)rr*)r3r rr)r:r;r<r*rr=)rCr rr*rDrErrF)rWr*rrX)(rV __future__rtypingrrrrOrcollections.abcrZ mypy.typesZmypyrZ mypy.pluginrr Z mypy.nodesr r r Z mypy.buildr Z _HookFuncr__annotations__ModuleNotFoundErrorexrr$r)r/r0r1r2r9rBrHrIr[r"r"r"r#s@ !           &