Skip to content

Commit

Permalink
Different colour for function name
Browse files Browse the repository at this point in the history
  • Loading branch information
bwohlberg committed Aug 1, 2024
1 parent baf116f commit ab5e3eb
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions scico/trace.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@

if have_colorama:
clr_main = colorama.Fore.LIGHTRED_EX
clr_func = colorama.Fore.RED
clr_args = colorama.Fore.LIGHTBLUE_EX
clr_reset = colorama.Fore.RESET
else:
Expand Down Expand Up @@ -148,16 +149,18 @@ def wrapper(*args, **kwargs):
):
argidx = 1
if args[0].__hash__() in call_trace.instance_hash:
name = f"{call_trace.instance_hash[args[0].__hash__()]}.{func.__name__}"
name = f"{call_trace.instance_hash[args[0].__hash__()]}.{clr_func}{func.__name__}"
elif hasattr(args[0], "__class__"):
name = (
f"{args[0].__class__.__module__}.{args[0].__class__.__name__}.{func.__name__}"
f"{args[0].__class__.__module__}.{args[0].__class__.__name__}."
f"{clr_func}{func.__name__}"
)
argsrep = [_trace_arg_repr(val) for val in args[argidx:]]
kwargrep = [f"{key}={_trace_arg_repr(val)}" for key, val in kwargs.items()]
argstr = clr_args + ", ".join(argsrep + kwargrep) + clr_main
print(
f"{clr_main}>> {' ' * 2 * call_trace.trace_level}{name}({argstr}){clr_reset}",
f"{clr_main}>> {' ' * 2 * call_trace.trace_level}{name}"
f"({argstr}{clr_func}){clr_reset}",
file=sys.stderr,
)
call_trace.trace_level += 1
Expand Down

0 comments on commit ab5e3eb

Please sign in to comment.