nonebot2/nonebot/event.py
2020-08-07 17:05:08 +08:00

128 lines
3.4 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from nonebot.typing import Any, Dict, Optional
class Event(dict):
"""
封装从 CQHTTP 收到的事件数据对象(字典),提供属性以获取其中的字段。
除 `type` 和 `detail_type` 属性对于任何事件都有效外,其它属性存在与否(不存在则返回
`None`)依事件不同而不同。
"""
@staticmethod
def from_payload(payload: Dict[str, Any]) -> Optional["Event"]:
"""
从 CQHTTP 事件数据构造 `Event` 对象。
"""
try:
e = Event(payload)
_ = e.type, e.detail_type
return e
except KeyError:
return None
@property
def type(self) -> str:
"""
事件类型,有 ``message``、``notice``、``request``、``meta_event`` 等。
"""
return self["post_type"]
@property
def detail_type(self) -> str:
"""
事件具体类型,依 `type` 的不同而不同,以 ``message`` 类型为例,有
``private``、``group``、``discuss`` 等。
"""
return self[f"{self.type}_type"]
@property
def sub_type(self) -> Optional[str]:
"""
事件子类型,依 `detail_type` 不同而不同,以 ``message.private`` 为例,有
``friend``、``group``、``discuss``、``other`` 等。
"""
return self.get("sub_type")
@property
def name(self):
"""
事件名,对于有 `sub_type` 的事件,为 ``{type}.{detail_type}.{sub_type}``,否则为
``{type}.{detail_type}``。
"""
n = self.type + "." + self.detail_type
if self.sub_type:
n += "." + self.sub_type
return n
@property
def self_id(self) -> int:
"""机器人自身 ID。"""
return self["self_id"]
@property
def user_id(self) -> Optional[int]:
"""用户 ID。"""
return self.get("user_id")
@property
def operator_id(self) -> Optional[int]:
"""操作者 ID。"""
return self.get("operator_id")
@property
def group_id(self) -> Optional[int]:
"""群 ID。"""
return self.get("group_id")
@property
def discuss_id(self) -> Optional[int]:
"""讨论组 ID。"""
return self.get("discuss_id")
@property
def message_id(self) -> Optional[int]:
"""消息 ID。"""
return self.get("message_id")
@property
def message(self) -> Optional[Any]:
"""消息。"""
return self.get("message")
@property
def raw_message(self) -> Optional[str]:
"""未经 CQHTTP 处理的原始消息。"""
return self.get("raw_message")
@property
def sender(self) -> Optional[Dict[str, Any]]:
"""消息发送者信息。"""
return self.get("sender")
@property
def anonymous(self) -> Optional[Dict[str, Any]]:
"""匿名信息。"""
return self.get("anonymous")
@property
def file(self) -> Optional[Dict[str, Any]]:
"""文件信息。"""
return self.get("file")
@property
def comment(self) -> Optional[str]:
"""请求验证消息。"""
return self.get("comment")
@property
def flag(self) -> Optional[str]:
"""请求标识。"""
return self.get("flag")
def __repr__(self) -> str:
return f"<Event, {super().__repr__()}>"