mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-28 08:12:14 +08:00
86 lines
1.8 KiB
Python
86 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import abc
|
|
from ipaddress import IPv4Address
|
|
|
|
from nonebot.config import Config
|
|
from nonebot.adapters import BaseBot
|
|
from nonebot.typing import Dict, Optional
|
|
|
|
|
|
class BaseDriver(abc.ABC):
|
|
|
|
@abc.abstractmethod
|
|
def __init__(self, config: Config):
|
|
self.config = config
|
|
self._clients: Dict[int, BaseBot] = {}
|
|
|
|
@property
|
|
@abc.abstractmethod
|
|
def server_app(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
@abc.abstractmethod
|
|
def asgi(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
@abc.abstractmethod
|
|
def logger(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
def bots(self) -> Dict[int, BaseBot]:
|
|
return self._clients
|
|
|
|
@abc.abstractmethod
|
|
def run(self,
|
|
host: Optional[IPv4Address] = None,
|
|
port: Optional[int] = None,
|
|
*args,
|
|
**kwargs):
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
async def _handle_http(self):
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
async def _handle_ws_reverse(self):
|
|
raise NotImplementedError
|
|
|
|
|
|
class BaseWebSocket(object):
|
|
|
|
@abc.abstractmethod
|
|
def __init__(self, websocket):
|
|
self._websocket = websocket
|
|
|
|
@property
|
|
@abc.abstractmethod
|
|
def websocket(self):
|
|
return self._websocket
|
|
|
|
@property
|
|
@abc.abstractmethod
|
|
def closed(self):
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
async def accept(self):
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
async def close(self, code: int):
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
async def receive(self) -> dict:
|
|
raise NotImplementedError
|
|
|
|
@abc.abstractmethod
|
|
async def send(self, data: dict):
|
|
raise NotImplementedError
|