mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-25 01:25:04 +08:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
from pprint import pprint
|
|
from typing import Optional
|
|
|
|
from nonebot.adapters import Bot as BaseBot
|
|
from nonebot.adapters import Event as BaseEvent
|
|
from nonebot.drivers import Driver, WebSocket
|
|
from nonebot.message import handle_event
|
|
from nonebot.typing import overrides
|
|
|
|
from .event import Event
|
|
|
|
|
|
class MiraiBot(BaseBot):
|
|
|
|
def __init__(self,
|
|
connection_type: str,
|
|
self_id: str,
|
|
*,
|
|
websocket: Optional["WebSocket"] = None):
|
|
super().__init__(connection_type, self_id, websocket=websocket)
|
|
|
|
@property
|
|
@overrides(BaseBot)
|
|
def type(self) -> str:
|
|
return "mirai"
|
|
|
|
@classmethod
|
|
@overrides(BaseBot)
|
|
async def check_permission(cls, driver: "Driver", connection_type: str,
|
|
headers: dict, body: Optional[dict]) -> str:
|
|
return ''
|
|
|
|
@overrides(BaseBot)
|
|
async def handle_message(self, message: dict):
|
|
event = Event.new(message)
|
|
await handle_event(self, event)
|
|
|
|
@overrides(BaseBot)
|
|
async def call_api(self, api: str, **data):
|
|
return super().call_api(api, **data)
|
|
|
|
@overrides(BaseBot)
|
|
async def send(self, event: "BaseEvent", message: str, **kwargs):
|
|
return super().send(event, message, **kwargs)
|