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

✨ fastapi driver support forward connect

Browse Source
This commit is contained in:
yanyongyu
2021-07-23 00:51:19 +08:00
parent 54a7e4808c
commit cf747f954c
5 changed files with 244 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
nonebot/drivers/aiohttp.py
View File

@ -212,6 +212,9 @@ class Driver(ForwardDriver):
BotClass = self._adapters[setup.adapter]
bot = BotClass(setup.self_id, request)
self._bot_connect(bot)
logger.opt(colors=True).info(
f"Start http polling for <y>{setup.adapter.upper()} "
f"Bot {setup.self_id}</y>")
headers = request.headers
timeout = aiohttp.ClientTimeout(30)
@ -289,11 +292,13 @@ class Driver(ForwardDriver):
)
try:
async with session.ws_connect(url) as ws:
logger.opt(colors=True).info(
f"WebSocket Connection to <y>{setup.adapter.upper()} "
f"Bot {setup.self_id}</y> succeeded!")
request = WebSocket(
setup.http_version, url.scheme, url.path,
url.raw_query_string.encode("latin-1"), {
**setup.headers, "host": host
}, ws)
url.raw_query_string.encode("latin-1"), headers,
ws)
BotClass = self._adapters[setup.adapter]
bot = BotClass(setup.self_id, request)