From: Nikos Skalkotos <skalkoto@grnet.gr>
Date: Wed, 10 Jul 2013 07:31:33 +0000 (+0300)
Subject: Add kwargs parameter to sysprep decorator
X-Git-Tag: 0.5~1^2~13^2~20
X-Git-Url: https://code.grnet.gr/git/snf-image-creator/commitdiff_plain/f94adfe0ac0909d5468146a1dec86bd4d40c1e33

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
--- 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: