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

put kwargs into filename

Browse source
This commit is contained in:
Valentin Boettcher 2025-01-31 11:24:45 -05:00
parent c0077e8a90
commit f2fa4db7fe
No known key found for this signature in database
GPG key ID: E034E12B7AF56ACE
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
hiroplotutils/__init__.py
View file

@ -29,7 +29,11 @@ def noop_if_interactive(f):
def make_figure(fig_name: str | None = None, *args, **kwargs): def make_figure(fig_name: str | None = None, *args, **kwargs):
fig_name = fig_name or inspect.stack()[1].function meta = get_function_meta()
fig_name = fig_name or (meta[1] if isinstance(meta[1], str) else meta[1].name)
fig_name += "".join(f"_{k}={str(v)}" for k, v in meta[2].items()) if meta[2] else ""
fig = plt.figure(fig_name, *args, **kwargs) fig = plt.figure(fig_name, *args, **kwargs)
fig.__dict__["__hiro_filename_function"] = get_function_meta() fig.__dict__["__hiro_filename_function"] = get_function_meta()
@ -112,7 +116,7 @@ def get_function_meta():
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, get_kwargs() return filename, function, get_kwargs(frame)
def write_meta(path, include_kwags=True, filename_function_override=None, **kwargs): def write_meta(path, include_kwags=True, filename_function_override=None, **kwargs):
@ -160,13 +164,14 @@ def write_meta(path, include_kwags=True, filename_function_override=None, **kwar
print(f"Metadata written to {outpath}") print(f"Metadata written to {outpath}")
def get_kwargs(): def get_kwargs(frame=None):
frame = inspect.currentframe().f_back.f_back.f_back.f_back frame = frame.frame or inspect.currentframe().f_back.f_back.f_back.f_back
keys, _, _, values = inspect.getargvalues(frame) keys, _, _, values = inspect.getargvalues(frame)
kwargs = {} kwargs = {}
for key in keys: for key in keys:
if key != "self": if key != "self":
kwargs[key] = values[key] kwargs[key] = values[key]
return kwargs return kwargs