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

🐛 fix log not escaped in forward driver

Browse Source
This commit is contained in:
yanyongyu
2021-09-25 19:14:16 +08:00
parent 049d14295e
commit cfdebb30b7
2 changed files with 23 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
nonebot/drivers/aiohttp.py
View File

@ -9,18 +9,19 @@ import signal
import asyncio
import threading
from dataclasses import dataclass
from typing import Set, List, cast, Union, Optional, Callable, Awaitable
from typing import Set, List, Union, Callable, Optional, Awaitable, cast
import aiohttp
from yarl import URL
from nonebot.log import logger
from nonebot.adapters import Bot
from nonebot.utils import escape_tag
from nonebot.typing import overrides
from nonebot.utils import escape_tag
from nonebot.config import Env, Config
from nonebot.drivers import (ForwardDriver, HTTPPollingSetup, WebSocketSetup,
HTTPRequest, WebSocket as BaseWebSocket)
from nonebot.drivers import WebSocket as BaseWebSocket
from nonebot.drivers import (HTTPRequest, ForwardDriver, WebSocketSetup,
HTTPPollingSetup)
STARTUP_FUNC = Callable[[], Awaitable[None]]
SHUTDOWN_FUNC = Callable[[], Awaitable[None]]
@ -242,8 +243,8 @@ class Driver(ForwardDriver):
setup_ = setup
except Exception as e:
logger.opt(colors=True, exception=e).error(
f"<r><bg #f8bbd0>Error while parsing setup {setup!r}.</bg #f8bbd0></r>"
)
"<r><bg #f8bbd0>Error while parsing setup "
f"{escape_tag(repr(setup))}.</bg #f8bbd0></r>")
await asyncio.sleep(3)
continue
@ -324,8 +325,8 @@ class Driver(ForwardDriver):
setup_ = setup
except Exception as e:
logger.opt(colors=True, exception=e).error(
f"<r><bg #f8bbd0>Error while parsing setup {setup!r}.</bg #f8bbd0></r>"
)
"<r><bg #f8bbd0>Error while parsing setup "
f"{escape_tag(repr(setup))}.</bg #f8bbd0></r>")
await asyncio.sleep(3)
continue