G/Ph UdZddlmZddlmZddlmZmZmZddl 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)Iterable)Final TYPE_CHECKINGCallableN)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 |j j z}d |d|d|<)|S)N _NBitByte _NBitShort _NBitIntC _NBitIntP_NBitInt _NBitLong _NBitLongLong _NBitHalf _NBitSingle _NBitDouble_NBitLongDoubleznumpy._Bitnumpy._typing._nbit.)npbyteshortintcintpint_longlonglonghalfsingledouble longdoubledtypeitemsize)namesretnametypns X/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numpy/typing/mypy_plugin.py_get_precision_dictr78s bg rx bg bg RW bg "+& bg  "  " BM* E C>> cSSUU[))-=q-=-=-= )4 ) )** 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_listrMNs)   N 9 8~ 8 8 88r8strchtjdj}|dkrdS|dkrdS|dkrdSdS)Nr5rJc_intlc_longq c_longlong)r#r/char)rUs r6_get_c_intp_namerV`sA 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..r")r3splitrW named_type)rZr4_apir3name_news r6_hookrcxsJKCC 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 moduleimportslist[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)selfr{s 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)rpznumpy.ctypeslibctypes_c_intp)rr{rxrXrY)r~rnr2s r6get_additional_depsz _NumpyPlugin.get_additional_depssT]B/0C}''!=FF-EFFF"333!(%y12Jr8N)r{rNrr|)rnr rr)__name__ __module__ __qualname____doc__rrrGr8r6rzrzsBLL          r8rzversiontype[_NumpyPlugin]ctSzAn entry-point for mypy.)rzrs r6pluginrsr8ctr)rrs r6rrs r8)rr)rr)rrN)rZr rr)rdrerfrNrrg)rnr rorNrprqrrr)rrNrr)(r __future__rcollections.abcrtypingrrrrr# mypy.typesmypyr mypy.pluginr r mypy.nodesr r r mypy.buildr _HookFuncr__annotations__ModuleNotFoundErrorexrr7rMrVrWrXrYrcrmrxrzrrGr8r6rs{B#"""""$$$$$$1111111111 66666666::::::::::"""""",-t34I*.G.....GGGGGG,9999$    -,......#?">"@"@@@@@"!######$$$$EGO;;;;     $$$$$v$$$L s4AA%A  A%