mirror of
https://github.com/kuitoi/kuitoi-Server.git
synced 2025-08-17 08:15:42 +00:00
SantaSpeen
c6c6ec31b0
Add async support to call_event(); Add **kwargs t0 call_event(); Sends new data to events;
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import asyncio
|
|
import builtins
|
|
import inspect
|
|
import time
|
|
|
|
from core import get_logger
|
|
|
|
|
|
# noinspection PyShadowingBuiltins
|
|
class EventsSystem:
|
|
|
|
def __init__(self):
|
|
# TODO: default events
|
|
self.log = get_logger("EventsSystem")
|
|
self.loop = asyncio.get_event_loop()
|
|
self.__events = {
|
|
"server_started": [],
|
|
"_plugins_start": [],
|
|
"auth_sent_key": [],
|
|
"auth_ok": [],
|
|
"chat_receive": [],
|
|
"_plugins_unload": [],
|
|
"server_stopped": [],
|
|
}
|
|
|
|
def builtins_hook(self):
|
|
self.log.debug("used builtins_hook")
|
|
builtins.ev = self
|
|
|
|
def register_event(self, event_name, event_func):
|
|
self.log.debug(f"register_event({event_name}, {event_func}):")
|
|
if not callable(event_func):
|
|
# TODO: i18n
|
|
self.log.error(f"Cannot add event '{event_name}'. "
|
|
f"Use `KuiToi.add_event({event_name}', function)` instead. Skipping it...")
|
|
return
|
|
if event_name not in self.__events:
|
|
self.__events.update({str(event_name): [event_func]})
|
|
else:
|
|
self.__events[event_name].append(event_func)
|
|
|
|
def call_event(self, event_name, *args, **kwargs):
|
|
self.log.debug(f"Using event '{event_name}'")
|
|
funcs_data = []
|
|
|
|
if event_name in self.__events.keys():
|
|
for func in self.__events[event_name]:
|
|
try:
|
|
event_data = {"event_name": event_name, "args": args, "kwargs": kwargs}
|
|
if inspect.iscoroutinefunction(func):
|
|
d = self.loop.run_until_complete(func(event_data))
|
|
else:
|
|
d = func(event_data)
|
|
funcs_data.append(d)
|
|
except Exception as e:
|
|
# TODO: i18n
|
|
self.log.error(f'Error while calling "{event_name}"; In function: "{func.__name__}"')
|
|
self.log.exception(e)
|
|
else:
|
|
# TODO: i18n
|
|
self.log.warning(f"Event {event_name} does not exist. Just skipping it...")
|
|
|
|
return funcs_data
|