Compatible with cyfunction

Author: FrostyLeaf

Diff for: ajsonrpc/manager.py

@@ -37,9 +37,13 @@ async def get_response_for_request(self, request: JSONRPC20Request) -> Optional[
             )
         else:
             try:
-                result = await method(*request.args, **request.kwargs) \
-                    if inspect.iscoroutinefunction(method) \
-                    else method(*request.args, **request.kwargs)
+                # Compatible with coroutine cyfunction or other kind of functions which return coroutines
+                # but are not coroutine functions
+                co_or_ret = method(*request.args, **request.kwargs)
+                # Maybe `inspect.isawaitable()` is better?
+                result = await co_or_ret \
+                    if inspect.iscoroutine(co_or_ret) \
+                    else co_or_ret
             except JSONRPC20DispatchException as dispatch_error:
                 # Dispatcher method raised exception with controlled "data"
                 output = JSONRPC20Response(

Diff for: ajsonrpc/scripts/server.py

@@ -36,7 +36,7 @@ def data_received(self, data):
         _, payload = request_message.split('\r\n\r\n', 1)
         task = create_task(self.json_rpc_manager.get_payload_for_payload(payload))
         task.add_done_callback(self.handle_task_result)
-    
+
     def handle_task_result(self, task):
         res = task.result()
         self.transport.write((
@@ -49,6 +49,12 @@ def handle_task_result(self, task):
         logger.info('Close the client socket')
         self.transport.close()
 
+# Compatible with cyfunction.
+# This change helps to work with some modules like `pydantic`, which wraps
+# functions (includes coroutine functions) into cyfunctions. The method
+# `inspect.isfunction()` cannot recognize them for now.
+def is_func_or_cyfunc(object):
+    return isfunction(object) or (type(object).__name__ == "cython_function_or_method")
 
 def main():
     """Usage: % examples.methods"""
@@ -68,7 +74,7 @@ def main():
     module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(module)
     # get functions from the module
-    methods = getmembers(module, isfunction)
+    methods = getmembers(module, is_func_or_cyfunc)
     logger.info('Extracted methods: {}'.format(methods))
     dispatcher = Dispatcher(dict(methods))