misc updates

- Support custom json encoder for serializing response data
- Support registering global decorators for jsonrpc methods
- Support custom method prefix for Dispatcher.add_class
- Support explicitly exporting subset of class methods with Dispatcher.add_class

Author: wonderbeyond

jsonrpc/backend/flask.py

@@ -19,7 +19,8 @@
 
 
 class JSONRPCAPI(object):
-    def __init__(self, dispatcher=None, check_content_type=True):
+    def __init__(self, dispatcher=None, check_content_type=True,
+                 json_encoder=None):
         """
 
         :param dispatcher: methods dispatcher
@@ -31,6 +32,7 @@ def __init__(self, dispatcher=None, check_content_type=True):
         self.dispatcher = dispatcher if dispatcher is not None \
             else Dispatcher()
         self.check_content_type = check_content_type
+        self.json_encoder = json_encoder or DatetimeDecimalEncoder
 
     def as_blueprint(self, name=None):
         blueprint = Blueprint(name if name else str(uuid4()), __name__)
@@ -83,9 +85,9 @@ def _get_request_str(self):
             return request.data
         return list(request.form.keys())[0]
 
-    @staticmethod
-    def _serialize(s):
-        return json.dumps(s, cls=DatetimeDecimalEncoder)
+    # @staticmethod
+    def _serialize(self, s):
+        return json.dumps(s, cls=self.json_encoder)
 
 
 api = JSONRPCAPI()

jsonrpc/dispatcher.py

@@ -4,6 +4,7 @@
 
 """
 import collections
+from functools import wraps
 
 
 class Dispatcher(collections.MutableMapping):
@@ -27,6 +28,7 @@ def __init__(self, prototype=None):
         None
 
         """
+        self._decorators = []
         self.method_map = dict()
 
         if prototype is not None:
@@ -36,7 +38,7 @@ def __getitem__(self, key):
         return self.method_map[key]
 
     def __setitem__(self, key, value):
-        self.method_map[key] = value
+        self.method_map[key] = self._wrap_method(value)
 
     def __delitem__(self, key):
         del self.method_map[key]
@@ -50,8 +52,24 @@ def __iter__(self):
     def __repr__(self):
         return repr(self.method_map)
 
+    def register_decorator(self, a):
+        self._decorators.extend(a if hasattr(a, '__iter__') else [a])
+
+    def _wrap_method(self, f):
+        @wraps(f)
+        def _method(*args, **kwargs):
+            nf = f
+            for deco in reversed(self._decorators):
+                nf = deco(nf)
+            return nf(*args, **kwargs)
+
+        return _method
+
     def add_class(self, cls):
-        prefix = cls.__name__.lower() + '.'
+        if hasattr(cls, 'rpc_method_prefix'):
+            prefix = cls.rpc_method_prefix + '.'
+        else:
+            prefix = cls.__name__.lower() + '.'
         self.build_method_map(cls(), prefix)
 
     def add_object(self, obj):
@@ -94,7 +112,7 @@ def mymethod(*args, **kwargs):
                 print(args, kwargs)
 
         """
-        self.method_map[name or f.__name__] = f
+        self[name or f.__name__] = f
         return f
 
     def build_method_map(self, prototype, prefix=''):
@@ -112,10 +130,19 @@ def build_method_map(self, prototype, prefix=''):
             Prefix of methods
 
         """
+        rpc_exports = getattr(prototype, 'rpc_exports', None)
+
+        def _should_export(method):
+            if method.startswith('_'):
+                return False
+            if rpc_exports is None:
+                return True
+            return method in rpc_exports
+
         if not isinstance(prototype, dict):
             prototype = dict((method, getattr(prototype, method))
                              for method in dir(prototype)
-                             if not method.startswith('_'))
+                             if _should_export(method))
 
         for attr, method in prototype.items():
             if callable(method):

jsonrpc/tests/test_dispatcher.py

@@ -90,10 +90,10 @@ def test_to_dict(self):
         d = Dispatcher()
 
         def func():
-            return ""
+            return "x"
 
         d["method"] = func
-        self.assertEqual(dict(d), {"method": func})
+        self.assertEqual(dict(d)["method"](), "x")
 
     def test_init_from_object_instance(self):