gFUdZddlmZddlmZmZmZddlZerddl m Z  ddl Z ddl m Z ddlmZmZddlmZmZmZdd lmZeege fZdZd ed <n#e$r ZeZYdZ[ndZ[wwxYwgZd ed <d.dZd/dZd0dZeZ ded<eZ!ded<eZ"ded<d1dZ#esed2d Z$d3d'Z%Gd(d)eZ&d4d,Z'dSd4d-Z'dS)5aA 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: 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 dtjfdtjfdtjfdtjfdtjfdtjfdtjfdtjfd tj fd tj fd tj fg }i}d }|D],\}}d |j j z}|d|d||d|<-|S)N _NBitByte _NBitShort _NBitIntC _NBitIntP_NBitInt _NBitLong _NBitLongLong _NBitHalf _NBitSingle _NBitDouble_NBitLongDoublez numpy._typingz ._nbit_base._Bitz._nbit.)npbyteshortintcintpint_longlonglonghalfsingledouble longdoubledtypeitemsize)namesretmodulenametypns V/var/www/html/mpstechhub/venv/lib/python3.11/site-packages/numpy/typing/mypy_plugin.py_get_precision_dictr7:s bg rx bg bg RW bg "+& bg  "  " BM* E C FGG cSSUU[))+1(F(F(F(F(Fv $ $d $ $%% Jc"gd}d|DS)N) uint128uint256int128int256float80float96float128float256 complex160 complex192 complex256 complex512c<g|]}tt||S)hasattrr").0is r6 z0_get_extended_precision_list..`s% 8 8 8!Q 8A 8 8 8r8rG)extended_namess r6_get_extended_precision_listrMQs)   N 9 8~ 8 8 88r8strchtjdj}|dkrdS|dkrdS|dkrdSdS)Nr5rJc_intlc_longq c_longlong)r"r.char)rUs r6_get_c_intp_namerVbsA 8C== D s{{w x |xr8r_PRECISION_DICT_EXTENDED_PRECISION_LIST_C_INTPctxr rc|\}}}|jdd}td|}||S)z;Replace a type-alias with a concrete ``NBitBase`` subclass..znumpy._typing._nbit.)r3splitrW named_type)rZr4_apir3name_news r6_hookrczsJKCC 8>>#  r "D>( # ##r8iterableIterable[Statement]idintct|D]\}}t|dd|kr|cStd|)z>Identify the first ``ImportFrom`` instance the specified `id`.rfNzBFailed to identify a `ImportFrom` instance with the following id: ) enumerategetattr ValueError)rdrfrJvalues r6_indexrmsf!(++  HAuudD))R//0:35::;; ;r8filer r2importslist[tuple[str, None | str]]Nonect|d|}d|_|j|jfD]}t ||}|||<dS)zA mypy plugin for handling versus numpy-specific typing tasks.fullnamerNrNone | _HookFuncc&|tvrtSdS)zSet the precision of platform-specific `numpy.number` subclasses. For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`. N)rWrc)selfrzs r6get_type_analyze_hookz"_NumpyPlugin.get_type_analyze_hooks ?** 4r8rnr list[tuple[int, str, int]]ct|jdfg}|jdkr"t|ddtDn%|jdkrt|dtdfg|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_precisioncg|]}||fSrGrG)rIvs r6rKz4_NumpyPlugin.get_additional_deps..sFFFaVFFFr8)roznumpy.ctypeslibctypes_c_intp)rrzrwrXrY)r}rnr1s r6get_additional_depsz _NumpyPlugin.get_additional_depssT]B/0C}''!=FF-EFFF"333!(%y12Jr8N)rzrNrr{)rnr rr)__name__ __module__ __qualname____doc__r~rrGr8r6ryrysBLL          r8ryversiontype[_NumpyPlugin]ctSzAn entry-point for mypy.)ryrs r6pluginrsr8ctr)rrs r6rrs r8)rr)rr)rrN)rZr rr)rdrerfrNrrg)rnr r2rNrorprrq)rrNrr)(r __future__rtypingrrrrr"collections.abcr mypy.typesmypyr mypy.pluginr r mypy.nodesr r r mypy.buildr _HookFuncr__annotations__ModuleNotFoundErrorexrr7rMrVrWrXrYrcrmrwryrrGr8r6rsB#"""""1111111111)(((((( 66666666::::::::::"""""",-t34I*.G.....GGGGGG.9999"    -,......#?">"@"@@@@@"!######$$$$EGO;;;;     $$$$$v$$$L s4AA'A""A'