🐛 fix import error for extra dependencies

This commit is contained in:
yanyongyu
2021-12-23 14:29:21 +08:00
parent cda49551a8
commit 0839d6d062
3 changed files with 15 additions and 6 deletions

View File

@ -23,7 +23,6 @@ from nonebot.config import Env
from nonebot.typing import overrides
from nonebot.utils import escape_tag
from nonebot.drivers.httpx import HttpxMixin
from nonebot.drivers.aiohttp import AiohttpMixin
from nonebot.config import Config as NoneBotConfig
from nonebot.drivers import Request as BaseRequest
from nonebot.drivers import WebSocket as BaseWebSocket
@ -35,6 +34,11 @@ from nonebot.drivers import (
combine_driver,
)
try:
from nonebot.drivers.aiohttp import AiohttpMixin
except ImportError:
AiohttpMixin = None
class Config(BaseSettings):
"""
@ -316,4 +320,5 @@ class FastAPIWebSocket(BaseWebSocket):
FullDriver = combine_driver(Driver, HttpxMixin, WebSocketsMixin)
AiohttpDriver = combine_driver(Driver, AiohttpMixin)
if AiohttpMixin:
AiohttpDriver = combine_driver(Driver, AiohttpMixin)