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

⚗️ new driver combine expr support

Browse Source
This commit is contained in:
yanyongyu
2021-12-23 17:20:26 +08:00
parent b9f1890d80
commit 8fb394e4c3
11 changed files with 83 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
nonebot/drivers/httpx.py
View File

@ -1,5 +1,3 @@
import httpx
from nonebot.typing import overrides
from nonebot.drivers._block_driver import BlockDriver
from nonebot.drivers import (
@ -11,8 +9,13 @@ from nonebot.drivers import (
combine_driver,
)
try:
import httpx
except ImportError:
raise ImportError("Please install httpx by using `pip install nonebot2[httpx]`")
class HttpxMixin(ForwardMixin):
class Mixin(ForwardMixin):
@property
@overrides(ForwardMixin)
def type(self) -> str:
@ -39,7 +42,7 @@ class HttpxMixin(ForwardMixin):
@overrides(ForwardMixin)
async def websocket(self, setup: Request) -> WebSocket:
return await super(HttpxMixin, self).websocket(setup)
return await super(Mixin, self).websocket(setup)
Driver = combine_driver(BlockDriver, HttpxMixin)
Driver = combine_driver(BlockDriver, Mixin)