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

👽 🐛 fix quart.exception removal caused import error

Browse Source
This commit is contained in:
Mix 2021-07-26 10:41:38 +08:00
parent 904f285f62
commit c1b10a0e48
1 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
nonebot/drivers/quart.py
View File

@ -21,7 +21,7 @@ from nonebot.config import Env, Config as NoneBotConfig
from nonebot.drivers import ReverseDriver, HTTPRequest, WebSocket as BaseWebSocket from nonebot.drivers import ReverseDriver, HTTPRequest, WebSocket as BaseWebSocket
try: try:
from quart import exceptions from werkzeug import exceptions
from quart import request as _request from quart import request as _request
from quart import websocket as _websocket from quart import websocket as _websocket
from quart import Quart, Request, Response from quart import Quart, Request, Response
@ -167,10 +167,8 @@ class Driver(ReverseDriver):
self_id, response = await BotClass.check_permission(self, http_request) self_id, response = await BotClass.check_permission(self, http_request)
if not self_id: if not self_id:
raise exceptions.HTTPException( raise exceptions.Unauthorized(
response and response.status or 401, description=(response and response.body or b"").decode())
description=(response and response.body or b"").decode(),
name="Request Denied")
if self_id in self._clients: if self_id in self._clients:
logger.warning("There's already a reverse websocket connection," logger.warning("There's already a reverse websocket connection,"
"so the event may be handled twice.") "so the event may be handled twice.")
@ -195,18 +193,14 @@ class Driver(ReverseDriver):
self_id, response = await BotClass.check_permission(self, ws) self_id, response = await BotClass.check_permission(self, ws)
if not self_id: if not self_id:
raise exceptions.HTTPException( raise exceptions.Unauthorized(
response and response.status or 401, description=(response and response.body or b"").decode())
description=(response and response.body or b"").decode(),
name="Request Denied")
if self_id in self._clients: if self_id in self._clients:
logger.opt(colors=True).warning( logger.opt(colors=True).warning(
"There's already a reverse websocket connection, " "There's already a reverse websocket connection, "
f"<y>{adapter.upper()} Bot {self_id}</y> ignored.") f"<y>{adapter.upper()} Bot {self_id}</y> ignored.")
raise exceptions.HTTPException(403, raise exceptions.Forbidden(description='Client already exists.')
description="Client already exists",
name="Request Denied")
bot = BotClass(self_id, ws) bot = BotClass(self_id, ws)
await ws.accept() await ws.accept()