Forward kwargs from raise_ajax to AjaxException init.

vispa/remote/__init__.py

@@ -36,16 +36,18 @@ class AjaxException(Exception):
         self.alert = alert
 
 
-def raise_ajax(fn):
+def raise_ajax(fn=None, **kwargs):
     """
     Decorator that transforms raised exceptions into AjaxExceptions.
     """
-    def wrapper(*args, **kwargs):
-        try:
-            return fn(*args, **kwargs)
-        except Exception as e:
-            if isinstance(e, AjaxException):
-                raise
-            else:
-                raise AjaxException(str(e))
-    return wrapper
+    def decorator(fn):
+        def wrapper(*args, **_kwargs):
+            try:
+                return fn(*args, **_kwargs)
+            except Exception as e:
+                if isinstance(e, AjaxException):
+                    raise
+                else:
+                    raise AjaxException(str(e), **kwargs)
+        return wrapper
+    return decorator if fn is None else decorator(fn)