Add private field to access declared types

jspahrsummers · Feb 15, 2020 · a0a9315 · a0a9315
1 parent 0c33d30
commit a0a9315
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/adt/decorator.py b/adt/decorator.py
@@ -27,6 +27,18 @@ def adt(cls):
     cls._Key = Enum(  # type: ignore
         '_Key', list(caseConstructors.keys()))
 
+    values = list(caseConstructors.values())
+    types = []
+    for i in range(len(caseConstructors.values())):
+        try:
+            types.append(values[i]._argType)
+        except AttributeError:
+            try:
+                types.append(values[i]._types)
+            except AttributeError:
+                types.append(None)
+    cls._types = types
+
     _installInit(cls)
     _installRepr(cls)
     _installStr(cls)