hiro/hiroplotutils
1
0
Fork 0
mirror of https://github.com/vale981/hiroplotutils synced 2025-03-05 09:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

get the kwargs too

Browse source
This commit is contained in:
Valentin Boettcher 2025-01-29 17:40:07 -05:00
parent 45f6520ef7
commit d13f5a529d
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
hiroplotutils/__init__.py
View file

@ -32,7 +32,7 @@ def make_figure(fig_name: str | None = None, *args, **kwargs):
fig_name = fig_name or inspect.stack()[1].function fig_name = fig_name or inspect.stack()[1].function
fig = plt.figure(fig_name, *args, **kwargs) fig = plt.figure(fig_name, *args, **kwargs)
fig.__dict__["__hiro_filename_function"] = get_file_and_function() fig.__dict__["__hiro_filename_function"] = get_function_meta()
fig.clf() fig.clf()
return fig return fig
@ -97,7 +97,7 @@ def get_jj_info(type):
).stdout.decode("utf-8") ).stdout.decode("utf-8")
def get_file_and_function(): def get_function_meta():
i = 1 i = 1
stack = inspect.stack() stack = inspect.stack()
frame = stack[i] frame = stack[i]
@ -112,7 +112,7 @@ def get_file_and_function():
filename = pathlib.Path(frame.filename) if frame else "<unknown>" filename = pathlib.Path(frame.filename) if frame else "<unknown>"
function = frame.function if frame else "<unknown>" function = frame.function if frame else "<unknown>"
return filename, function return filename, function, get_kwargs()
def write_meta(path, include_kwags=True, filename_function_override=None, **kwargs): def write_meta(path, include_kwags=True, filename_function_override=None, **kwargs):
@ -135,7 +135,7 @@ def write_meta(path, include_kwags=True, filename_function_override=None, **kwar
.strip() .strip()
) )
filename, function = filename_function_override or get_file_and_function() filename, function, fn_kwargs = filename_function_override or get_function_meta()
if filename != "<unknown>": if filename != "<unknown>":
filename = str(filename.relative_to(project_dir)) filename = str(filename.relative_to(project_dir))
@ -146,7 +146,7 @@ def write_meta(path, include_kwags=True, filename_function_override=None, **kwar
dict( dict(
source=filename, source=filename,
function=function, function=function,
function_args=get_kwargs() if include_kwags else {}, function_args=fn_kwargs if include_kwags else {},
change_id=change_id, change_id=change_id,
commit_id=commit_id, commit_id=commit_id,
description=description.strip(), description=description.strip(),