simple execute fix

shueteam · Apr 12, 2020 · 6d18102 · 6d18102
1 parent 3a11591
commit 6d18102
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/vkwave/bots/easy/easy_bot.py b/vkwave/bots/easy/easy_bot.py
@@ -1,5 +1,6 @@
 import asyncio
 import typing
+import inspect
 
 from vkwave.api.methods import API
 from vkwave.api.token.token import BotSyncPoolTokens, BotSyncSingleToken, Token
@@ -118,7 +119,9 @@ def __init__(self, func: typing.Callable[[BaseEvent], typing.Awaitable[typing.An
 
         async def execute(self, event: BotEvent) -> typing.Any:
             new_event = SimpleBotEvent(event)
-            return await self.func(new_event)
+            if inspect.iscoroutinefunction(self.func):
+                return await self.func(new_event)
+            return self.func(new_event)
 
     def message_handler(self, *filters: BaseFilter):
         def decorator(func: typing.Callable[..., typing.Any]):