get_underlying_type

Author: SeanMcOwen

src/math_spec_mapping/Convenience/cadCAD.py

@@ -1,5 +1,13 @@
 from copy import deepcopy
-from typing import _TypedDictMeta
+from typing import _TypedDictMeta, _GenericAlias
+
+
+def get_underlying_type(data):
+    if type(data) == _GenericAlias:
+        data.__args__ = [get_underlying_type(x) for x in data.__args__]
+    elif type(data) == _TypedDictMeta:
+        data = get_nested_types(data.__annotations__)
+    return data
 
 
 def get_nested_types(data):
@@ -8,5 +16,7 @@ def get_nested_types(data):
         print(type(data[key]))
         if type(data[key]) == _TypedDictMeta:
             data[key] = get_nested_types(data[key].__annotations__)
+        elif type(data[key]) == _GenericAlias:
+            data[key] = get_underlying_type(data[key])
 
     return data