Add kwargs parameter to sysprep decorator

This can be used to add arbitrary attributes to the function it
decorates

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
 
 
     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
 
     """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:
         @wraps(func)
         def wrapper2(self, print_message=True):
             if print_message: