From 22d76a3dac72add693baa7161edc436048ceb56e Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Tue, 19 Jul 2022 12:42:28 +0200 Subject: [PATCH] added option to specify newly appended kwargs for the argument forwarding --- slic/utils/argfwd.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/slic/utils/argfwd.py b/slic/utils/argfwd.py index 21eeab55..3caac418 100644 --- a/slic/utils/argfwd.py +++ b/slic/utils/argfwd.py @@ -1,7 +1,7 @@ import inspect -def forwards_to(func_inner, nfilled=0): +def forwards_to(func_inner, nfilled=0, appended_kwargs=None): """ Decorator inserting an explicit signature for functions that forward their arguments to other function via *args/**kwargs @@ -18,6 +18,9 @@ def forwards_to(func_inner, nfilled=0): pos_comb = merge_lists_unique(pos_outer, pos_inner) kw_comb = merge_dicts_unique(kw_outer, kw_inner) + if appended_kwargs: + kw_comb = merge_dicts_unique(kw_comb, appended_kwargs) + func_outer.__signature__ = make_signature(pos_comb, kw_comb) return func_outer