From 3500ab90c0c1a6023724e9e7ab0790826f0f30c7 Mon Sep 17 00:00:00 2001 From: bDgcgsX4bnc3Duprdb2S <157499110+bDgcgsX4bnc3Duprdb2S@users.noreply.github.com> Date: Mon, 17 Mar 2025 20:51:44 +0100 Subject: [PATCH] Update macro.py Add macro support for args and kwargs Eg: """set ip {{ ip | IP | macro("myfunc", "arg1", "arg2", key="value") }}""" TODO: Fix macro doc saying macros are based on exec (they are NOT) and describe new args and kwargs behavior --- ttp/match/macro.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ttp/match/macro.py b/ttp/match/macro.py index 1f4ea58..006a02f 100644 --- a/ttp/match/macro.py +++ b/ttp/match/macro.py @@ -1,7 +1,7 @@ -def macro(data, macro_name): +def macro(data, macro_name, *args, **kwargs): result = None if macro_name in _ttp_["macro"]: - result = _ttp_["macro"][macro_name](data) + result = _ttp_["macro"][macro_name](data, *args, **kwargs) # process macro result if result is True: return data, True