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

add types

Browse Source
This commit is contained in:
yanyongyu
2020-08-10 13:06:02 +08:00
parent 00913f1a8f
commit 9e33a605a6
9 changed files with 87 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
nonebot/adapters/__init__.py
View File

@ -3,13 +3,10 @@
import abc
from functools import reduce
from dataclasses import dataclass
# from pydantic.dataclasses import dataclass # dataclass with validation
from dataclasses import dataclass, field
from nonebot.config import Config
from nonebot.drivers import BaseWebSocket
from nonebot.typing import Dict, Union, Iterable, Optional
from nonebot.typing import Dict, Union, Iterable, WebSocket
class BaseBot(abc.ABC):
@ -20,7 +17,7 @@ class BaseBot(abc.ABC):
config: Config,
self_id: int,
*,
websocket: BaseWebSocket = None):
websocket: WebSocket = None):
self.connection_type = connection_type
self.config = config
self.self_id = self_id
@ -43,7 +40,7 @@ class BaseBot(abc.ABC):
@dataclass
class BaseMessageSegment(abc.ABC):
type: str
data: Dict[str, str] = {}
data: Dict[str, str] = field(default_factory=lambda: {})
@abc.abstractmethod
def __str__(self):

5
nonebot/adapters/cqhttp.py
View File

@ -8,10 +8,9 @@ import httpx
from nonebot.event import Event
from nonebot.config import Config
from nonebot.message import handle_event
from nonebot.drivers import BaseWebSocket
from nonebot.exception import ApiNotAvailable
from nonebot.typing import Tuple, Iterable, Optional, overrides
from nonebot.adapters import BaseBot, BaseMessage, BaseMessageSegment
from nonebot.typing import Tuple, Iterable, Optional, overrides, WebSocket
def escape(s: str, *, escape_comma: bool = True) -> str:
@ -47,7 +46,7 @@ class Bot(BaseBot):
config: Config,
self_id: int,
*,
websocket: BaseWebSocket = None):
websocket: WebSocket = None):
if connection_type not in ["http", "websocket"]:
raise ValueError("Unsupported connection type")