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

⚗️ add proxy support

Browse Source
This commit is contained in:
yanyongyu
2021-12-25 14:04:53 +08:00
parent e62ce93011
commit 188eb110c5
4 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nonebot/drivers/aiohttp.py
View File

@ -40,7 +40,7 @@ class Mixin(ForwardMixin):
files = aiohttp.FormData()
for name, file in setup.files:
files.add_field(name, file[1], content_type=file[2], filename=file[0])
async with aiohttp.ClientSession(version=version) as session:
async with aiohttp.ClientSession(version=version, trust_env=True) as session:
async with session.request(
setup.method,
setup.url,
@ -48,6 +48,7 @@ class Mixin(ForwardMixin):
json=setup.json,
headers=setup.headers,
timeout=timeout,
proxy=setup.proxy,
) as response:
res = Response(
response.status,
@ -66,12 +67,13 @@ class Mixin(ForwardMixin):
else:
raise RuntimeError(f"Unsupported HTTP version: {setup.version}")
session = aiohttp.ClientSession(version=version)
session = aiohttp.ClientSession(version=version, trust_env=True)
ws = await session.ws_connect(
setup.url,
method=setup.method,
timeout=setup.timeout or 10,
headers=setup.headers,
proxy=setup.proxy,
)
websocket = WebSocket(request=setup, session=session, websocket=ws)
return websocket