ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-16 02:50:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 Fix: websockets 驱动器连接关闭 code 获取错误 (#2537)

Browse Source
This commit is contained in:
Ju4tCode
2024-01-17 16:39:35 +08:00
committed by GitHub
parent c2d2169a9f
commit 2c6affecea
5 changed files with 97 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
nonebot/drivers/websockets.py
View File

@ -50,10 +50,7 @@ def catch_closed(func: Callable[P, Awaitable[T]]) -> Callable[P, Awaitable[T]]:
try:
return await func(*args, **kwargs)
except ConnectionClosed as e:
if e.rcvd_then_sent:
raise WebSocketClosed(e.rcvd.code, e.rcvd.reason) # type: ignore
else:
raise WebSocketClosed(e.sent.code, e.sent.reason) # type: ignore
raise WebSocketClosed(e.code, e.reason)
return decorator