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"" d["value"] = type_name + elif type(d["value"]) == typing.TypeAliasType: + d["value"] = "" elif inspect.isfunction(d["value"]): function_desc = str(d["value"]) search_result = function_str_regex.search(function_desc)