agatha/maubot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add warning if webapp is disabled when using @web decorators. Fixes #65

Browse Source
This commit is contained in:
Tulir Asokan 2021-06-11 12:42:06 +03:00
parent b7f6c08b2b
commit 8f806a2594
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
maubot/plugin_base.py
View File

@ -55,6 +55,7 @@ class Plugin(ABC):
self._handlers_at_startup = []
def register_handler_class(self, obj) -> None:
warned_webapp = False
for key in dir(obj):
val = getattr(obj, key)
try:
@ -65,10 +66,17 @@ class Plugin(ABC):
pass
try:
web_handlers = val.__mb_web_handler__
for method, path, kwargs in web_handlers:
self.webapp.add_route(method=method, path=path, handler=val, **kwargs)
except AttributeError:
pass
else:
if len(web_handlers) > 0 and self.webapp is None:
if not warned_webapp:
self.log.warning(f"{type(obj).__name__} has web handlers, but the webapp"
" feature isn't enabled in the plugin's maubot.yaml")
warned_webapp = True
continue
for method, path, kwargs in web_handlers:
self.webapp.add_route(method=method, path=path, handler=val, **kwargs)
async def pre_start(self) -> None:
pass