Fix: Data type '...' is misinterpreted

nutti · Dec 31, 2024 · f987ede · f987ede
1 parent 4c8d511
commit f987ede
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 0 deletions.
diff --git a/src/fake_bpy_module/transformer/data_type_refiner.py b/src/fake_bpy_module/transformer/data_type_refiner.py
@@ -203,6 +203,9 @@ def _get_refined_data_type_fast(  # noqa: C901, PLR0911, PLR0912
             if s:
                 return [make_data_type_node(f"`{s}`")]
 
+        if dtype_str == "...":
+            return [make_data_type_node("...")]
+
         if dtype_str in ("type", "object", "function"):
             return [make_data_type_node("typing.Any")]
 

diff --git a/...bpy_module_test/transformer_test_data/data_type_refiner_test/expect/special_data_type.xml b/...bpy_module_test/transformer_test_data/data_type_refiner_test/expect/special_data_type.xml
@@ -49,6 +49,13 @@
                 <class-ref>
                     ClassA
                 ], None]
+    <data>
+        <name>
+            data_python_typing_syntax_6
+        <description>
+        <data-type-list>
+            <data-type>
+                Callable[[float, int], tuple[float, int]
     <data>
         <name>
             data_parentheses_pair

diff --git a/...est/transformer_test_data/data_type_refiner_test/expect/special_data_type_transformed.xml b/...est/transformer_test_data/data_type_refiner_test/expect/special_data_type_transformed.xml
@@ -63,6 +63,24 @@
                 , 
                 None
                 ]
+    <data>
+        <name>
+            data_python_typing_syntax_6
+        <description>
+        <data-type-list>
+            <data-type>
+                Callable[
+                [
+                float
+                , 
+                int
+                ]
+                , 
+                tuple[
+                float
+                , 
+                int
+                ]
     <data>
         <name>
             data_parentheses_pair

diff --git a/...e_test/transformer_test_data/data_type_refiner_test/input/special_data_type.rst b/...e_test/transformer_test_data/data_type_refiner_test/input/special_data_type.rst
@@ -20,6 +20,10 @@
 
    :type: Callable[[int, :class:`ClassA`], None]
 
+.. data:: data_python_typing_syntax_6
+
+   :type: Callable[[float, int], tuple[float, int]]
+
 .. data:: data_parentheses_pair
 
    :type: (:class:`ClassA`, :class:`module_2.ClassB`) pair