From f94adfe0ac0909d5468146a1dec86bd4d40c1e33 Mon Sep 17 00:00:00 2001
From: Nikos Skalkotos <skalkoto@grnet.gr>
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:
-- 
GitLab