Add kwargs parameter to sysprep decorator

[snf-image-creator] / image_creator / os_type / __init__.py

diff --git a/image_creator/os_type/__init__.py b/image_creator/os_type/__init__.py
index 855022f..e41c52c 100644 (file)
--- 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: