From 9e8e445bb351be2cacda31541c6e9dd2544930bf Mon Sep 17 00:00:00 2001
From: Hannes Braun <hannes@hannesbraun.net>
Date: Tue, 3 Dec 2024 15:23:01 +0100
Subject: [PATCH] Visualize type aliases on the stack

---
 pytrace-generator/main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pytrace-generator/main.py b/pytrace-generator/main.py
index 15ec11f8..96237237 100644
--- a/pytrace-generator/main.py
+++ b/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)