vbatts/maubot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Allow plugins to add more classes as handler sources

Browse source
This commit is contained in:
Tulir Asokan 2019-12-26 17:47:30 +02:00
parent 4c1fe0fb37
commit 50addc2ceb
1 changed files with 17 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
maubot/plugin_base.py
View file

@ -20,6 +20,7 @@ from asyncio import AbstractEventLoop
from sqlalchemy.engine.base import Engine from sqlalchemy.engine.base import Engine
from aiohttp import ClientSession from aiohttp import ClientSession
from yarl import URL
if TYPE_CHECKING: if TYPE_CHECKING:
from mautrix.util.config import BaseProxyConfig from mautrix.util.config import BaseProxyConfig
@ -35,7 +36,7 @@ class Plugin(ABC):
config: Optional['BaseProxyConfig'] config: Optional['BaseProxyConfig']
database: Optional[Engine] database: Optional[Engine]
webapp: Optional['PluginWebApp'] webapp: Optional['PluginWebApp']
webapp_url: Optional[str] webapp_url: Optional[URL]
def __init__(self, client: 'MaubotMatrixClient', loop: AbstractEventLoop, http: ClientSession, def __init__(self, client: 'MaubotMatrixClient', loop: AbstractEventLoop, http: ClientSession,
instance_id: str, log: Logger, config: Optional['BaseProxyConfig'], instance_id: str, log: Logger, config: Optional['BaseProxyConfig'],
@ -49,12 +50,12 @@ class Plugin(ABC):
self.config = config self.config = config
self.database = database self.database = database
self.webapp = webapp self.webapp = webapp
self.webapp_url = webapp_url self.webapp_url = URL(webapp_url)
self._handlers_at_startup = [] self._handlers_at_startup = []
async def internal_start(self) -> None: def register_handler_class(self, obj) -> None:
for key in dir(self): for key in dir(obj):
val = getattr(self, key) val = getattr(obj, key)
try: try:
if val.__mb_event_handler__: if val.__mb_event_handler__:
self._handlers_at_startup.append((val, val.__mb_event_type__)) self._handlers_at_startup.append((val, val.__mb_event_type__))
@ -67,12 +68,23 @@ class Plugin(ABC):
self.webapp.add_route(method=method, path=path, handler=val, **kwargs) self.webapp.add_route(method=method, path=path, handler=val, **kwargs)
except AttributeError: except AttributeError:
pass pass
async def pre_start(self) -> None:
pass
async def internal_start(self) -> None:
await self.pre_start()
self.register_handler_class(self)
await self.start() await self.start()
async def start(self) -> None: async def start(self) -> None:
pass pass
async def pre_stop(self) -> None:
pass
async def internal_stop(self) -> None: async def internal_stop(self) -> None:
await self.pre_stop()
for func, event_type in self._handlers_at_startup: for func, event_type in self._handlers_at_startup:
self.client.remove_event_handler(event_type, func) self.client.remove_event_handler(event_type, func)
if self.webapp is not None: if self.webapp is not None: