From f94adfe0ac0909d5468146a1dec86bd4d40c1e33 Mon Sep 17 00:00:00 2001 From: Nikos Skalkotos Date: Wed, 10 Jul 2013 10:31:33 +0300 Subject: [PATCH] Add kwargs parameter to sysprep decorator This can be used to add arbitrary attributes to the function it decorates --- image_creator/os_type/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/image_creator/os_type/__init__.py b/image_creator/os_type/__init__.py index 855022f..e41c52c 100644 --- a/image_creator/os_type/__init__.py +++ b/image_creator/os_type/__init__.py @@ -68,13 +68,16 @@ def add_prefix(target): return wrapper -def sysprep(message, enabled=True): +def sysprep(message, enabled=True, **kwargs): """Decorator for system preparation tasks""" def wrapper1(func): func.sysprep = True func.enabled = enabled func.executed = False + for key, val in kwargs.items(): + setattr(func, key, val) + @wraps(func) def wrapper2(self, print_message=True): if print_message: -- 1.7.10.4