Merge pull request #165 from skogsbaer/typealias-on-stack

Visualize type aliases on the stack

Author: skogsbaer

pytrace-generator/main.py

@@ -11,6 +11,7 @@
 import socket
 import sys
 import types
+import typing
 
 def eprint(*args, **kwargs):
     print(*args, file=sys.stderr, **kwargs)
@@ -29,6 +30,7 @@ def eprint(*args, **kwargs):
     str: "str",
     type(None): "none",
     type: "type",
+    typing.TypeAliasType: "type",
     types.FunctionType: "function"
 }
 HEAP_TYPES = {
@@ -120,6 +122,8 @@ def format(self):
             if search_result is not None:
                 type_name = f"<class '{search_result.group(1)}'>"
             d["value"] = type_name
+        elif type(d["value"]) == typing.TypeAliasType:
+            d["value"] = "<TypeAlias>"
         elif inspect.isfunction(d["value"]):
             function_desc = str(d["value"])
             search_result = function_str_regex.search(function_desc)