ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-09-06 20:16:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add startup shutdown deco

Browse Source
This commit is contained in:
yanyongyu
2020-08-11 10:44:05 +08:00
parent 2d90c35df6
commit b32d4a24d1
5 changed files with 36 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
nonebot/drivers/fastapi.py
View File

@ -14,9 +14,9 @@ from fastapi import Body, Header, Response, WebSocket as FastAPIWebSocket
from nonebot.log import logger
from nonebot.config import Env, Config
from nonebot.utils import DataclassEncoder
from nonebot.typing import Optional, overrides
from nonebot.adapters.cqhttp import Bot as CQBot
from nonebot.drivers import BaseDriver, BaseWebSocket
from nonebot.typing import Optional, Callable, overrides
class Driver(BaseDriver):
@ -38,7 +38,7 @@ class Driver(BaseDriver):
@property
@overrides(BaseDriver)
def server_app(self):
def server_app(self) -> FastAPI:
return self._server_app
@property
@ -51,6 +51,14 @@ class Driver(BaseDriver):
def logger(self) -> logging.Logger:
return logging.getLogger("fastapi")
@overrides(BaseDriver)
def on_startup(self, func: Callable) -> Callable:
return self.server_app.on_event("startup")(func)
@overrides(BaseDriver)
def on_shutdown(self, func: Callable) -> Callable:
return self.server_app.on_event("shutdown")(func)
@overrides(BaseDriver)
def run(self,
host: Optional[IPv4Address] = None,