add: support for non-awaitable function

snamper · Mar 3, 2020 · 165e7a2 · 165e7a2
1 parent 0300725
commit 165e7a2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/mirai/session.py b/mirai/session.py
@@ -318,9 +318,15 @@ async def main_entrance(self, run_body, event_context, queue):
               for normal_middleware in normal_middlewares:
                 SessionLogger.debug(f"a event called {event_context.name}, enter a currect context.")
                 normal_stack.enter_context(normal_middleware)
-              return await callable_target(**translated_mapping)
+              if inspect.iscoroutinefunction(callable_target):
+                return await callable_target(**translated_mapping)
+              else:
+                return callable_target(**translated_mapping)
 
-      return await callable_target(**translated_mapping)
+      if inspect.iscoroutinefunction(callable_target):
+        return await callable_target(**translated_mapping)
+      else:
+        return callable_target(**translated_mapping)
     except (NameError, TypeError) as e:
       EventLogger.error(f"threw a exception by {event_context.name}, it's about Annotations Checker, please report to developer.")
       traceback.print_exc()