Use type dict to convert primitive type

Author: sitaowang1998

python/spider-py/src/spider_py/type/tdl_convert.py

@@ -20,6 +20,16 @@
 from spider_py.type.type import Double, Float, Int8, Int16, Int32, Int64
 from spider_py.type.utils import get_class_name
 
+TypeDict = {
+    Int8: Int8Type(),
+    Int16: Int16Type(),
+    Int32: Int32Type(),
+    Int64: Int64Type(),
+    Float: FloatType(),
+    Double: DoubleType(),
+    bool: BoolType(),
+}
+
 
 def _to_primitive_tdl_type(native_type: type | GenericAlias) -> TdlType | None:
     """
@@ -30,27 +40,14 @@ def _to_primitive_tdl_type(native_type: type | GenericAlias) -> TdlType | None:
         - None if `native_type` is not a primitive Python type.
     :raises TypeError: If `native_type` is a primitive Python type not supported by TDL.
     """
-    tdl_type: TdlType | None = None
-    if native_type is Int8:
-        tdl_type = Int8Type()
-    elif native_type is Int16:
-        tdl_type = Int16Type()
-    elif native_type is Int32:
-        tdl_type = Int32Type()
-    elif native_type is Int64:
-        tdl_type = Int64Type()
-    elif native_type is Float:
-        tdl_type = FloatType()
-    elif native_type is Double:
-        tdl_type = DoubleType()
-    elif native_type is bool:
-        tdl_type = BoolType()
+    if isinstance(native_type, type) and native_type in TypeDict:
+        return TypeDict[native_type]
 
     if native_type in (int, float, str, complex, bytes):
         msg = f"{native_type} is not a TDL type. Please use the corresponding TDL primitive type."
         raise TypeError(msg)
 
-    return tdl_type
+    return None
 
 
 def to_tdl_type(native_type: type | GenericAlias) -> TdlType: