🎨 simplify ctypeslib.as_ctypes_type using _numtype.ToDType*

Author: jorenham

src/numpy-stubs/@test/static/accept/ctypeslib.pyi

@@ -78,6 +78,3 @@ assert_type(np.ctypeslib.as_ctypes(AR_f8), ct.Array[ct.c_double])
 assert_type(np.ctypeslib.as_array(AR_u1), npt.NDArray[np.ubyte])
 assert_type(np.ctypeslib.as_array(1), npt.NDArray[Any])
 assert_type(np.ctypeslib.as_array(pointer), npt.NDArray[Any])
-
-assert_type(np.ctypeslib.as_ctypes_type(int), type[ct.c_ssize_t])
-assert_type(np.ctypeslib.as_ctypes_type("N"), type[ct.c_size_t])

src/numpy-stubs/ctypeslib.pyi

@@ -9,24 +9,8 @@ import _numtype as _nt
 import numpy as np
 from numpy._core._internal import _ctypes
 from numpy._core.multiarray import flagsobj
-from numpy._typing import _ArrayLike, _BoolCodes, _DTypeLike, _Shape, _ShapeLike, _VoidDTypeLike
-from numpy._typing._char_codes import (
-    _Float32Codes,
-    _Float64Codes,
-    _Int8Codes,
-    _Int16Codes,
-    _Int32Codes,
-    _Int64Codes,
-    _IntPCodes,
-    _LongCodes,
-    _LongDoubleCodes,
-    _UInt8Codes,
-    _UInt16Codes,
-    _UInt32Codes,
-    _UInt64Codes,
-    _UIntPCodes,
-    _ULongCodes,
-)
+from numpy._typing import _ArrayLike, _DTypeLike, _Shape, _ShapeLike, _VoidDTypeLike
+from numpy._typing._char_codes import _LongCodes, _ULongCodes
 
 __all__ = ["as_array", "as_ctypes", "as_ctypes_type", "c_intp", "load_library", "ndpointer"]
 
@@ -176,43 +160,33 @@ def as_ctypes(obj: _nt.Array[np.longdouble]) -> ct.Array[ct.c_longdouble]: ...
 
 # TODO(jorenham): https://github.com/numpy/numtype/issues/522
 @overload
-def as_ctypes_type(dtype: _DTypeLike[np.bool] | type[bool | ct.c_bool] | _BoolCodes) -> type[ct.c_bool]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeBool) -> type[ct.c_bool]: ...
 @overload
-def as_ctypes_type(dtype: _DTypeLike[np.int8] | type[ct.c_int8 | ct.c_byte] | _Int8Codes) -> type[ct.c_int8]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeInt8) -> type[ct.c_int8]: ...
 @overload
-def as_ctypes_type(dtype: _DTypeLike[np.int16] | type[ct.c_int16 | ct.c_short] | _Int16Codes) -> type[ct.c_int16]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeUInt8) -> type[ct.c_uint8]: ...
 @overload
-def as_ctypes_type(dtype: _DTypeLike[np.int32] | type[ct.c_int32 | ct.c_int] | _Int32Codes) -> type[ct.c_int32]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeInt16) -> type[ct.c_int16]: ...
 @overload
-def as_ctypes_type(dtype: type[ct.c_long] | _LongCodes) -> type[ct.c_long]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeUInt16) -> type[ct.c_uint16]: ...
 @overload
-def as_ctypes_type(dtype: type[_nt.JustInt | ct.c_ssize_t] | _IntPCodes) -> type[ct.c_ssize_t]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeInt32) -> type[ct.c_int32]: ...
 @overload
-def as_ctypes_type(dtype: _DTypeLike[np.int64] | type[ct.c_int64] | _Int64Codes) -> type[ct.c_int64]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeUInt32) -> type[ct.c_uint32]: ...
 @overload
-def as_ctypes_type(dtype: _DTypeLike[np.uint8] | type[ct.c_uint8 | ct.c_ubyte] | _UInt8Codes) -> type[ct.c_uint8]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeInt64) -> type[ct.c_int64]: ...
 @overload
-def as_ctypes_type(
-    dtype: _DTypeLike[np.uint16] | type[ct.c_uint16 | ct.c_ushort] | _UInt16Codes,
-) -> type[ct.c_uint16]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeUInt64) -> type[ct.c_uint64]: ...
 @overload
-def as_ctypes_type(
-    dtype: _DTypeLike[np.uint32] | type[ct.c_uint32 | ct.c_uint] | _UInt32Codes,
-) -> type[ct.c_uint32]: ...
+def as_ctypes_type(dtype: type[ct.c_long] | _LongCodes) -> type[ct.c_long]: ...
 @overload
 def as_ctypes_type(dtype: type[ct.c_ulong] | _ULongCodes) -> type[ct.c_ulong]: ...
 @overload
-def as_ctypes_type(dtype: type[ct.c_void_p | ct.c_size_t] | _UIntPCodes) -> type[ct.c_size_t]: ...
-@overload
-def as_ctypes_type(dtype: _DTypeLike[np.uint64] | type[ct.c_uint64] | _UInt64Codes) -> type[ct.c_uint64]: ...
-@overload
-def as_ctypes_type(dtype: _DTypeLike[np.float32] | type[ct.c_float] | _Float32Codes) -> type[ct.c_float]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeFloat32) -> type[ct.c_float]: ...
 @overload
-def as_ctypes_type(dtype: _DTypeLike[np.float64] | type[ct.c_double] | _Float64Codes) -> type[ct.c_double]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeFloat64) -> type[ct.c_double]: ...
 @overload
-def as_ctypes_type(
-    dtype: _DTypeLike[np.longdouble] | type[ct.c_longdouble] | _LongDoubleCodes,
-) -> type[ct.c_longdouble]: ...
+def as_ctypes_type(dtype: _nt.ToDTypeLongDouble) -> type[ct.c_longdouble]: ...
 @overload
 def as_ctypes_type(dtype: _VoidDTypeLike) -> _ct._UnionType | _ct._PyCStructType: ...
 @overload