B .uf8 @s\UdZddlmZddlZddlZydddlZddlm Z ddl m Z m Z ddl mZmZmZddlmZeje ge fZdZded <Wn&ek rZzeZWddZ[XYnXgZd ed <d d ddZd d ddZeZded<eZded<dddddZejsedkrHddddddZ Gd d!d!e Z!dd"d#d$d%Z"ndd"d#d&d%Z"dS)'z6A module containing `numpy`-specific plugins for mypy.) annotationsN)Type)PluginAnalyzeTypeContext)MypyFile ImportFrom Statement)PRI_MEDzt.Optional[ModuleNotFoundError]MYPY_EXz t.List[str]__all__zt.Dict[str, str])returnc Csdtjfdtjfdtjfdtjfdtjfdtjfdtjfdtjfd tj fd tj fg }i}x4|D],\}}d |j j }d |d |d|<q^W|S)NZ _NBitByteZ _NBitShortZ _NBitIntCZ _NBitIntPZ_NBitIntZ _NBitLongLongZ _NBitHalfZ _NBitSingleZ _NBitDoubleZ_NBitLongDoubleznumpy._ZBitznumpy.typing._nbit.) npbyteshortZintcZintpint_longlongZhalfZsingledouble longdoubleZdtypeitemsize)namesretnametypnrZ/home/ankuromar296_gmail_com/myenv/lib/python3.7/site-packages/numpy/typing/mypy_plugin.py_get_precision_dicts rc sBtjtjtjtjg}ddddddddd d d d h fd d|DS)NZuint128Zuint256Zint128Zint256Zfloat80Zfloat96Zfloat128Zfloat256Z complex160Z complex192Z complex256Z complex512csg|]}|jkr|jqSr)__name__).0i)extended_namesrr =sz0_get_extended_precision_list..)rZ ulonglongrrZ clongdouble)Zextended_typesr)r!r_get_extended_precision_list-sr#zt.Final_PRECISION_DICT_EXTENDED_PRECISION_LISTrr)ctxr cCs2|\}}}|jdd}td|}||S)z;Replace a type-alias with a concrete ``NBitBase`` subclass..znumpy.typing._nbit.)rsplitr$Z named_type)r&r_apirZname_newrrr_hookHs r,zt.Iterable[Statement]strint)iterableidr cCs<x6t|D]\}}t|dd|kr |Sq Wtd|dS)z>Identify the first ``ImportFrom`` instance the specified `id`.r0NzBFailed to identify a `ImportFrom` instance with the following id: ) enumerategetattr ValueError)r/r0r valuerrr_indexQsr5c@s0eZdZdZdddddZddd d d Zd S) _NumpyPluginzCA plugin for assigning platform-specific `numpy.number` precisions.r-zt.Optional[_HookFunc])fullnamer cCs|tkr tSdS)zSet the precision of platform-specific `numpy.number` subclasses. For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`. N)r$r,)selfr7rrrget_type_analyze_hook]sz"_NumpyPlugin.get_type_analyze_hookrzt.List[t.Tuple[int, str, int]])filer cCsbt|jdfg}|jdkr^tddddtDd}d|_x&|j|jgD]}t|d}|||<qDW|S) zImport platform-specific extended-precision `numpy.number` subclasses. For example: `numpy.float96`, `numpy.float128` and `numpy.complex256`. r(numpyz numpy.typing._extended_precisionrcSsg|] }||fqSrr)rvrrrr"qsz4_NumpyPlugin.get_additional_deps..)rT)r r7rr%Z is_top_levelZdefsimportsr5)r8r:rr=lstr rrrget_additional_depsfs   z _NumpyPlugin.get_additional_depsN)r __module__ __qualname____doc__r9r?rrrrr6Zs r6zt.Type[_NumpyPlugin])versionr cCstS)zAn entry-point for mypy.)r6)rCrrrplugin|srDcCstdS)zAn entry-point for mypy.N)r )rCrrrrDs)#rB __future__rtypingtr;rZ mypy.typesZmypyrZ mypy.pluginrrZ mypy.nodesrrrZ mypy.buildr CallableZ _HookFuncr __annotations__ModuleNotFoundErrorexr rr#r$r%r, TYPE_CHECKINGr5r6rDrrrrs0     "