ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-09-07 04:26:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

✏️ add typing

Browse Source
This commit is contained in:
yanyongyu
2021-12-21 18:22:14 +08:00
parent b900133ab4
commit 9b2fa46921
8 changed files with 35 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
nonebot/drivers/fastapi.py
View File

@ -333,8 +333,8 @@ class WebSocketsWS(BaseWebSocket):
raise NotImplementedError
@overrides(BaseWebSocket)
async def close(self, code: int = 1000):
await self.websocket.close(code)
async def close(self, code: int = 1000, reason: str = ""):
await self.websocket.close(code, reason)
@overrides(BaseWebSocket)
async def receive(self) -> str:
@ -374,11 +374,13 @@ class FastAPIWebSocket(BaseWebSocket):
)
@overrides(BaseWebSocket)
async def accept(self):
async def accept(self) -> None:
await self.websocket.accept()
@overrides(BaseWebSocket)
async def close(self, code: int = status.WS_1000_NORMAL_CLOSURE):
async def close(
self, code: int = status.WS_1000_NORMAL_CLOSURE, reason: str = ""
) -> None:
await self.websocket.close(code)
@overrides(BaseWebSocket)