Decorated methods should forward args and kwargs

yourlabs · Nov 26, 2012 · 5c5a460 · 5c5a460
1 parent 61016ae
commit 5c5a460
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/rules_light/class_decorator.py b/rules_light/class_decorator.py
@@ -14,8 +14,8 @@
 def patch_get_object(cls, suffix, override):
     old_get_object = cls.get_object
 
-    def new_get_object(self):
-        obj = old_get_object(self)
+    def new_get_object(self, *args, **kwargs):
+        obj = old_get_object(self, *args, **kwargs)
 
         if self.get_object._rule_override:
             rule_name = self.get_object._rule_override
@@ -67,14 +67,14 @@ def __new__(self, *args):
         if issubclass(cls, generic.CreateView):
             old_get_form = cls.get_form
 
-            def new_get_form(self, form_class):
+            def new_get_form(self, *args, **kwargs):
                 model = form_class.Meta.model
                 rule_name = '%s.%s.create' % (model._meta.app_label,
                     model._meta.module_name)
 
                 registry.require(self.request.user, rule_name)
 
-                return old_get_form(self, form_class)
+                return old_get_form(self, *args, **kwargs)
 
             cls.get_form = new_get_form