Re-added __call__ method on step, transform, and hook implementation …

…objects
pigletto · May 11, 2010 · 7366074 · 7366074
1 parent 6155c93
commit 7366074
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/freshen/stepregistry.py b/freshen/stepregistry.py
@@ -37,6 +37,9 @@ def __init__(self, step_type, spec, func):
 
     def run(self, *args, **kwargs):
         self.func(*args, **kwargs)
+
+    def __call__(self, *args, **kwargs):
+        self.func(*args, **kwargs)
 
     def match(self, match):
         return self.re_spec.match(match)
@@ -59,7 +62,9 @@ def __repr__(self):
 
     def run(self, scenario):
         self.func(scenario)
-
+
+    def __call__(self, *args, **kwargs):
+        self.func(*args, **kwargs)
 
 class TransformImpl(object):
 
@@ -75,6 +80,9 @@ def transform_arg(self, arg):
         match = self.re_spec.match(arg)
         if match:
             return self.func(*match.groups())
+
+    def __call__(self, *args, **kwargs):
+        self.func(*args, **kwargs)
 
 class StepImplLoader(object):