Issue 49: Add register_api to SimpleJSONRPCServer

jsonrpclib/SimpleJSONRPCServer.py

@@ -218,6 +218,11 @@ def __init__(self, addr, requestHandler=SimpleJSONRPCRequestHandler,
             flags |= fcntl.FD_CLOEXEC
             fcntl.fcntl(self.fileno(), fcntl.F_SETFD, flags)
 
+    def register_api(self, api_obj):
+        methods = dir(api_obj)
+        apis = filter(lambda m: not m.startswith('_'), methods)
+        [self.register_function(getattr(api_obj, api)) for api in apis]
+
 
 class CGIJSONRPCRequestHandler(SimpleJSONRPCDispatcher):
 

tests.py

@@ -365,6 +365,12 @@ def test_proxy_object_reuse_is_allowed(self):
         result = sub_service_proxy.add(21, 21)
         self.assertTrue(result == 42)
 
+    def test_registered_by_register_api(self):
+        client = self.get_client()
+        result = client.echo(['hello, world'])
+        self.assertEqual(result, ['hello, world'])
+
+
 if jsonrpc.USE_UNIX_SOCKETS:
     # We won't do these tests unless Unix Sockets are supported
 
@@ -450,6 +456,13 @@ def ping():
         return True
 
 
+class ExampleApi(object):
+
+    @staticmethod
+    def echo(args):
+        return args
+
+
 class ExampleAggregateService(ExampleService):
     """
     Exposes the inherited ExampleService and a second copy as sub_service
@@ -473,6 +486,8 @@ def log_request(self, *args, **kwargs):
     server.register_function(service.summation, 'sum')
     server.register_function(service.summation, 'notify_sum')
     server.register_function(service.summation, 'namespace.sum')
+    example_api = ExampleApi()
+    server.register_api(example_api)
     server_proc = Thread(target=server.serve_forever)
     server_proc.daemon = True
     server_proc.start()