mirror of
https://github.com/nonebot/nonebot2.git
synced 2025-02-21 18:15:50 +08:00
commit
a98417a878
@ -21,7 +21,7 @@ _✨ Python 异步机器人框架 ✨_
|
|||||||
<a href="https://github.com/howmanybots/onebot/blob/master/README.md">
|
<a href="https://github.com/howmanybots/onebot/blob/master/README.md">
|
||||||
<img src="https://img.shields.io/badge/OneBot-v11-black?style=social&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAEAAAABABAMAAABYR2ztAAAAIVBMVEUAAAAAAAADAwMHBwceHh4UFBQNDQ0ZGRkoKCgvLy8iIiLWSdWYAAAAAXRSTlMAQObYZgAAAQVJREFUSMftlM0RgjAQhV+0ATYK6i1Xb+iMd0qgBEqgBEuwBOxU2QDKsjvojQPvkJ/ZL5sXkgWrFirK4MibYUdE3OR2nEpuKz1/q8CdNxNQgthZCXYVLjyoDQftaKuniHHWRnPh2GCUetR2/9HsMAXyUT4/3UHwtQT2AggSCGKeSAsFnxBIOuAggdh3AKTL7pDuCyABcMb0aQP7aM4AnAbc/wHwA5D2wDHTTe56gIIOUA/4YYV2e1sg713PXdZJAuncdZMAGkAukU9OAn40O849+0ornPwT93rphWF0mgAbauUrEOthlX8Zu7P5A6kZyKCJy75hhw1Mgr9RAUvX7A3csGqZegEdniCx30c3agAAAABJRU5ErkJggg==" alt="cqhttp">
|
<img src="https://img.shields.io/badge/OneBot-v11-black?style=social&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAEAAAABABAMAAABYR2ztAAAAIVBMVEUAAAAAAAADAwMHBwceHh4UFBQNDQ0ZGRkoKCgvLy8iIiLWSdWYAAAAAXRSTlMAQObYZgAAAQVJREFUSMftlM0RgjAQhV+0ATYK6i1Xb+iMd0qgBEqgBEuwBOxU2QDKsjvojQPvkJ/ZL5sXkgWrFirK4MibYUdE3OR2nEpuKz1/q8CdNxNQgthZCXYVLjyoDQftaKuniHHWRnPh2GCUetR2/9HsMAXyUT4/3UHwtQT2AggSCGKeSAsFnxBIOuAggdh3AKTL7pDuCyABcMb0aQP7aM4AnAbc/wHwA5D2wDHTTe56gIIOUA/4YYV2e1sg713PXdZJAuncdZMAGkAukU9OAn40O849+0ornPwT93rphWF0mgAbauUrEOthlX8Zu7P5A6kZyKCJy75hhw1Mgr9RAUvX7A3csGqZegEdniCx30c3agAAAABJRU5ErkJggg==" alt="cqhttp">
|
||||||
</a>
|
</a>
|
||||||
<a href="https://ding-doc.dingtalk.com/doc#/serverapi2/krgddi">
|
<a href="https://ding-doc.dingtalk.com/document#/org-dev-guide/elzz1p">
|
||||||
<img src="https://img.shields.io/badge/%E9%92%89%E9%92%89-Bot-lightgrey?style=social&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAEAAAABACAMAAACdt4HsAAAAnFBMVEUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD4jUzeAAAAM3RSTlMAQKSRaA+/f0YyFevh29R3cyklIfrlyrGsn41tVUs48c/HqJm9uZdhX1otGwkF9IN8V1CX0Q+IAAABY0lEQVRYw+3V2W7CMBAF0JuNQAhhX9OEfYdu9///rUVWpagE27Ef2gfO+0zGozsKnv6bMGzAhkNytIe5gDdzrwtTCwrbI8x4/NF668NAxgI3Q3UtFi3TyPwNQtPLUUmDd8YfqGLNe4v22XwEYb5zoOuF5baHq2UHtsKe5ivWfGAwrWu2mC34QM0PoCAuqZdOmiwV+5BLyMRtZ7dTSEcs48rzWfzwptMLyzpApka1SJ5FtR4kfCqNIBPEVDmqoqgwUYY5plQOlf6UEjNoOPnuKB6wzDyCrks///TDza8+PnR109WQdxLo8RKWq0PPnuXG0OXKQ6wWLFnCg75uYYbhmMIVVdQ709q33aHbGIj6Duz+2k1HQFX9VwqmY8xYsEJll2ahvhWgsjYLHFRXvIi2Qb0jzMQCzC3FAoydxCma88UCzE3JCWwkjCNYyMUCzHX4DiuTMawEwwhW6hnshPhjZzzJfAH0YacpbmRd7QAAAABJRU5ErkJggg==" alt="ding">
|
<img src="https://img.shields.io/badge/%E9%92%89%E9%92%89-Bot-lightgrey?style=social&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAEAAAABACAMAAACdt4HsAAAAnFBMVEUAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD4jUzeAAAAM3RSTlMAQKSRaA+/f0YyFevh29R3cyklIfrlyrGsn41tVUs48c/HqJm9uZdhX1otGwkF9IN8V1CX0Q+IAAABY0lEQVRYw+3V2W7CMBAF0JuNQAhhX9OEfYdu9///rUVWpagE27Ef2gfO+0zGozsKnv6bMGzAhkNytIe5gDdzrwtTCwrbI8x4/NF668NAxgI3Q3UtFi3TyPwNQtPLUUmDd8YfqGLNe4v22XwEYb5zoOuF5baHq2UHtsKe5ivWfGAwrWu2mC34QM0PoCAuqZdOmiwV+5BLyMRtZ7dTSEcs48rzWfzwptMLyzpApka1SJ5FtR4kfCqNIBPEVDmqoqgwUYY5plQOlf6UEjNoOPnuKB6wzDyCrks///TDza8+PnR109WQdxLo8RKWq0PPnuXG0OXKQ6wWLFnCg75uYYbhmMIVVdQ709q33aHbGIj6Duz+2k1HQFX9VwqmY8xYsEJll2ahvhWgsjYLHFRXvIi2Qb0jzMQCzC3FAoydxCma88UCzE3JCWwkjCNYyMUCzHX4DiuTMawEwwhW6hnshPhjZzzJfAH0YacpbmRd7QAAAABJRU5ErkJggg==" alt="ding">
|
||||||
</a>
|
</a>
|
||||||
<a href="https://core.telegram.org/bots/api">
|
<a href="https://core.telegram.org/bots/api">
|
||||||
@ -71,7 +71,7 @@ NoneBot2 的驱动框架 `Driver` 以及通信协议 `Adapter` 均可**自定义
|
|||||||
目前 NoneBot2 内置的协议适配:
|
目前 NoneBot2 内置的协议适配:
|
||||||
|
|
||||||
- [OneBot(CQHTTP) 协议](https://github.com/howmanybots/onebot/blob/master/README.md) (QQ 等)
|
- [OneBot(CQHTTP) 协议](https://github.com/howmanybots/onebot/blob/master/README.md) (QQ 等)
|
||||||
- [钉钉](https://ding-doc.dingtalk.com/doc#/serverapi2/krgddi) _开发中_
|
- [钉钉](https://ding-doc.dingtalk.com/document#/org-dev-guide/elzz1p) _开发中_
|
||||||
- [Telegram](https://core.telegram.org/bots/api) _计划中_
|
- [Telegram](https://core.telegram.org/bots/api) _计划中_
|
||||||
|
|
||||||
## 即刻开始
|
## 即刻开始
|
||||||
|
@ -370,7 +370,7 @@ Event 基类。提供获取关键信息的方法,其余信息可直接获取
|
|||||||
* **参数**
|
* **参数**
|
||||||
|
|
||||||
|
|
||||||
* `message: Union[str, dict, list, BaseModel, MessageSegment, Message]`: 消息内容
|
* `message: Union[str, list, dict, MessageSegment, Message, Any]`: 消息内容
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -63,7 +63,7 @@ sidebarDepth: 0
|
|||||||
|
|
||||||
## _exception_ `SessionExpired`
|
## _exception_ `SessionExpired`
|
||||||
|
|
||||||
基类:[`nonebot.exception.ApiNotAvailable`](../exception.md#nonebot.exception.ApiNotAvailable), `nonebot.adapters.ding.exception.DingAdapterException`
|
基类:`nonebot.adapters.ding.exception.ApiNotAvailable`, `nonebot.adapters.ding.exception.DingAdapterException`
|
||||||
|
|
||||||
|
|
||||||
* **说明**
|
* **说明**
|
||||||
@ -94,7 +94,7 @@ sidebarDepth: 0
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
### _async_ `handle_message(body)`
|
### _async_ `handle_message(message)`
|
||||||
|
|
||||||
|
|
||||||
* **说明**
|
* **说明**
|
||||||
@ -195,147 +195,290 @@ sidebarDepth: 0
|
|||||||
|
|
||||||
基类:[`nonebot.adapters.Event`](README.md#nonebot.adapters.Event)
|
基类:[`nonebot.adapters.Event`](README.md#nonebot.adapters.Event)
|
||||||
|
|
||||||
钉钉 协议 Event 适配。继承属性参考 [BaseEvent](./#class-baseevent) 。
|
钉钉 协议 Event 适配。各事件字段参考 [钉钉文档](https://ding-doc.dingtalk.com/document#/org-dev-guide/elzz1p)
|
||||||
|
|
||||||
|
|
||||||
### _property_ `raw_event`
|
### `get_type()`
|
||||||
|
|
||||||
原始上报消息
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
### _property_ `id`
|
获取事件类型的方法,类型通常为 NoneBot 内置的四种类型。
|
||||||
|
|
||||||
|
|
||||||
* 类型: `Optional[str]`
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
* 说明: 消息 ID
|
|
||||||
|
|
||||||
|
* `Literal["message", "notice", "request", "meta_event"]`
|
||||||
|
|
||||||
### _property_ `name`
|
|
||||||
|
|
||||||
|
|
||||||
* 类型: `str`
|
### `get_event_name()`
|
||||||
|
|
||||||
|
|
||||||
* 说明: 事件名称,由 type.\`detail_type\` 组合而成
|
* **说明**
|
||||||
|
|
||||||
|
获取事件名称的方法。
|
||||||
|
|
||||||
### _property_ `self_id`
|
|
||||||
|
|
||||||
|
|
||||||
* 类型: `str`
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
* 说明: 机器人自身 ID
|
* `str`
|
||||||
|
|
||||||
|
|
||||||
### _property_ `time`
|
|
||||||
|
|
||||||
|
### `get_event_description()`
|
||||||
|
|
||||||
* 类型: `int`
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
* 说明: 消息的时间戳,单位 s
|
获取事件描述的方法,通常为事件具体内容。
|
||||||
|
|
||||||
|
|
||||||
### _property_ `type`
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
* 类型: `str`
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
* 说明: 事件类型
|
|
||||||
|
|
||||||
|
|
||||||
### _property_ `detail_type`
|
### `get_message()`
|
||||||
|
|
||||||
|
|
||||||
* 类型: `str`
|
* **说明**
|
||||||
|
|
||||||
|
获取事件消息内容的方法。
|
||||||
|
|
||||||
* 说明: 事件详细类型
|
|
||||||
|
|
||||||
|
|
||||||
### _property_ `sub_type`
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
* 类型: `None`
|
* `Message`
|
||||||
|
|
||||||
|
|
||||||
* 说明: 钉钉适配器无事件子类型
|
|
||||||
|
|
||||||
|
### `get_plaintext()`
|
||||||
|
|
||||||
### _property_ `user_id`
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
* 类型: `Optional[str]`
|
获取消息纯文本的方法,通常不需要修改,默认通过 `get_message().extract_plain_text` 获取。
|
||||||
|
|
||||||
|
|
||||||
* 说明: 发送者 ID
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
### _property_ `group_id`
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
* 类型: `Optional[str]`
|
|
||||||
|
|
||||||
|
|
||||||
* 说明: 事件主体群 ID
|
### `get_user_id()`
|
||||||
|
|
||||||
|
|
||||||
### _property_ `to_me`
|
* **说明**
|
||||||
|
|
||||||
|
获取事件主体 id 的方法,通常是用户 id 。
|
||||||
|
|
||||||
* 类型: `Optional[bool]`
|
|
||||||
|
|
||||||
|
|
||||||
* 说明: 消息是否与机器人相关
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
### _property_ `message`
|
* `str`
|
||||||
|
|
||||||
|
|
||||||
* 类型: `Optional[Message]`
|
|
||||||
|
|
||||||
|
### `get_session_id()`
|
||||||
|
|
||||||
* 说明: 消息内容
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
### _property_ `reply`
|
获取会话 id 的方法,用于判断当前事件属于哪一个会话,通常是用户 id、群组 id 组合。
|
||||||
|
|
||||||
|
|
||||||
* 类型: `None`
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
* 说明: 回复消息详情
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
### _property_ `raw_message`
|
|
||||||
|
|
||||||
|
|
||||||
* 类型: `Optional[str]`
|
### `is_tome()`
|
||||||
|
|
||||||
|
|
||||||
* 说明: 原始消息
|
* **说明**
|
||||||
|
|
||||||
|
获取事件是否与机器人有关的方法。
|
||||||
|
|
||||||
### _property_ `plain_text`
|
|
||||||
|
|
||||||
|
|
||||||
* 类型: `Optional[str]`
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
* 说明: 纯文本消息内容
|
* `bool`
|
||||||
|
|
||||||
|
|
||||||
### _property_ `sender`
|
|
||||||
|
|
||||||
|
## _class_ `ConversationType`
|
||||||
|
|
||||||
* 类型: `Optional[dict]`
|
基类:`str`, `enum.Enum`
|
||||||
|
|
||||||
|
An enumeration.
|
||||||
|
|
||||||
|
|
||||||
|
### `_member_type_`
|
||||||
|
|
||||||
|
`builtins.str` 的别名
|
||||||
|
|
||||||
|
|
||||||
|
## _class_ `MessageEvent`
|
||||||
|
|
||||||
|
基类:`nonebot.adapters.ding.event.Event`
|
||||||
|
|
||||||
|
|
||||||
|
### `get_type()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取事件类型的方法,类型通常为 NoneBot 内置的四种类型。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `Literal["message", "notice", "request", "meta_event"]`
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
### `get_event_name()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取事件名称的方法。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
### `get_event_description()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取事件描述的方法,通常为事件具体内容。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
### `get_message()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取事件消息内容的方法。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `Message`
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
### `get_plaintext()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取消息纯文本的方法,通常不需要修改,默认通过 `get_message().extract_plain_text` 获取。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
### `get_user_id()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取事件主体 id 的方法,通常是用户 id 。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
### `get_session_id()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取会话 id 的方法,用于判断当前事件属于哪一个会话,通常是用户 id、群组 id 组合。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `str`
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## _class_ `PrivateMessageEvent`
|
||||||
|
|
||||||
|
基类:`nonebot.adapters.ding.event.MessageEvent`
|
||||||
|
|
||||||
|
|
||||||
|
## _class_ `GroupMessageEvent`
|
||||||
|
|
||||||
|
基类:`nonebot.adapters.ding.event.MessageEvent`
|
||||||
|
|
||||||
|
|
||||||
|
### `is_tome()`
|
||||||
|
|
||||||
|
|
||||||
|
* **说明**
|
||||||
|
|
||||||
|
获取事件是否与机器人有关的方法。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
* **返回**
|
||||||
|
|
||||||
|
|
||||||
|
* `bool`
|
||||||
|
|
||||||
* 说明: 消息发送者信息
|
|
||||||
|
|
||||||
|
|
||||||
## _class_ `MessageSegment`
|
## _class_ `MessageSegment`
|
||||||
@ -345,7 +488,12 @@ sidebarDepth: 0
|
|||||||
钉钉 协议 MessageSegment 适配。具体方法参考协议消息段类型或源码。
|
钉钉 协议 MessageSegment 适配。具体方法参考协议消息段类型或源码。
|
||||||
|
|
||||||
|
|
||||||
### _static_ `actionCardSingleMultiBtns(title, text, btns=[], hideAvatar=False, btnOrientation='1')`
|
### _static_ `extension(dict_)`
|
||||||
|
|
||||||
|
"标记 text 文本的 extension 属性,需要与 text 消息段相加。
|
||||||
|
|
||||||
|
|
||||||
|
### _static_ `actionCardMultiBtns(title, text, btns, hideAvatar=False, btnOrientation='1')`
|
||||||
|
|
||||||
|
|
||||||
* **参数**
|
* **参数**
|
||||||
@ -358,7 +506,7 @@ sidebarDepth: 0
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
### _static_ `feedCard(links=[])`
|
### _static_ `feedCard(links)`
|
||||||
|
|
||||||
|
|
||||||
* **参数**
|
* **参数**
|
||||||
@ -368,11 +516,6 @@ sidebarDepth: 0
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
### _static_ `empty()`
|
|
||||||
|
|
||||||
不想回复消息到群里
|
|
||||||
|
|
||||||
|
|
||||||
## _class_ `Message`
|
## _class_ `Message`
|
||||||
|
|
||||||
基类:[`nonebot.adapters.Message`](README.md#nonebot.adapters.Message)
|
基类:[`nonebot.adapters.Message`](README.md#nonebot.adapters.Message)
|
||||||
|
@ -6,6 +6,7 @@
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
import abc
|
import abc
|
||||||
|
from copy import copy
|
||||||
from typing_extensions import Literal
|
from typing_extensions import Literal
|
||||||
from functools import reduce, partial
|
from functools import reduce, partial
|
||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
@ -164,7 +165,7 @@ class MessageSegment(abc.ABC):
|
|||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def __add__(self: T_MessageSegment, other: Union[str, T_MessageSegment,
|
def __add__(self: T_MessageSegment, other: Union[str, T_MessageSegment,
|
||||||
T_Message]) -> "T_Message":
|
T_Message]) -> T_Message:
|
||||||
"""你需要在这里实现不同消息段的合并:
|
"""你需要在这里实现不同消息段的合并:
|
||||||
比如:
|
比如:
|
||||||
if isinstance(other, str):
|
if isinstance(other, str):
|
||||||
@ -198,6 +199,9 @@ class MessageSegment(abc.ABC):
|
|||||||
def get(self, key, default=None):
|
def get(self, key, default=None):
|
||||||
return getattr(self, key, default)
|
return getattr(self, key, default)
|
||||||
|
|
||||||
|
def copy(self: T_MessageSegment) -> T_MessageSegment:
|
||||||
|
return copy(self)
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def is_text(self) -> bool:
|
def is_text(self) -> bool:
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
@ -207,22 +211,22 @@ class Message(list, abc.ABC):
|
|||||||
"""消息数组"""
|
"""消息数组"""
|
||||||
|
|
||||||
def __init__(self,
|
def __init__(self,
|
||||||
message: Union[str, dict, list, T_MessageSegment,
|
message: Union[str, list, dict, T_MessageSegment, T_Message,
|
||||||
T_Message] = None,
|
Any] = None,
|
||||||
*args,
|
*args,
|
||||||
**kwargs):
|
**kwargs):
|
||||||
"""
|
"""
|
||||||
:参数:
|
:参数:
|
||||||
|
|
||||||
* ``message: Union[str, dict, list, MessageSegment, Message]``: 消息内容
|
* ``message: Union[str, list, dict, MessageSegment, Message, Any]``: 消息内容
|
||||||
"""
|
"""
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
if isinstance(message, (str, dict, list)):
|
if isinstance(message, Message):
|
||||||
self.extend(self._construct(message))
|
|
||||||
elif isinstance(message, Message):
|
|
||||||
self.extend(message)
|
self.extend(message)
|
||||||
elif isinstance(message, MessageSegment):
|
elif isinstance(message, MessageSegment):
|
||||||
self.append(message)
|
self.append(message)
|
||||||
|
else:
|
||||||
|
self.extend(self._construct(message))
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return ''.join((str(seg) for seg in self))
|
return ''.join((str(seg) for seg in self))
|
||||||
@ -238,8 +242,7 @@ class Message(list, abc.ABC):
|
|||||||
@staticmethod
|
@staticmethod
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def _construct(
|
def _construct(
|
||||||
msg: Union[str, dict, list,
|
msg: Union[str, list, dict, Any]) -> Iterable[T_MessageSegment]:
|
||||||
BaseModel]) -> Iterable[T_MessageSegment]:
|
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
def __add__(self: T_Message, other: Union[str, T_MessageSegment,
|
def __add__(self: T_Message, other: Union[str, T_MessageSegment,
|
||||||
|
@ -6,7 +6,6 @@ import asyncio
|
|||||||
from typing import Any, Dict, Union, Optional, TYPE_CHECKING
|
from typing import Any, Dict, Union, Optional, TYPE_CHECKING
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
from nonebot.log import logger
|
from nonebot.log import logger
|
||||||
from nonebot.config import Config
|
from nonebot.config import Config
|
||||||
from nonebot.typing import overrides
|
from nonebot.typing import overrides
|
||||||
|
@ -5,13 +5,13 @@
|
|||||||
协议详情请看: `钉钉文档`_
|
协议详情请看: `钉钉文档`_
|
||||||
|
|
||||||
.. _钉钉文档:
|
.. _钉钉文档:
|
||||||
https://ding-doc.dingtalk.com/doc#/serverapi2/krgddi
|
https://ding-doc.dingtalk.com/document#/org-dev-guide/elzz1p/
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from .utils import log
|
from .utils import log
|
||||||
from .bot import Bot
|
from .bot import Bot
|
||||||
from .event import Event
|
|
||||||
from .message import Message, MessageSegment
|
from .message import Message, MessageSegment
|
||||||
|
from .event import Event, MessageEvent, PrivateMessageEvent, GroupMessageEvent
|
||||||
from .exception import (DingAdapterException, ApiNotAvailable, NetworkError,
|
from .exception import (DingAdapterException, ApiNotAvailable, NetworkError,
|
||||||
ActionFailed, SessionExpired)
|
ActionFailed, SessionExpired)
|
||||||
|
@ -6,18 +6,20 @@ from typing import Any, Union, Optional, TYPE_CHECKING
|
|||||||
import httpx
|
import httpx
|
||||||
from nonebot.log import logger
|
from nonebot.log import logger
|
||||||
from nonebot.config import Config
|
from nonebot.config import Config
|
||||||
|
from nonebot.typing import overrides
|
||||||
from nonebot.message import handle_event
|
from nonebot.message import handle_event
|
||||||
from nonebot.adapters import Bot as BaseBot
|
from nonebot.adapters import Bot as BaseBot
|
||||||
from nonebot.exception import RequestDenied
|
from nonebot.exception import RequestDenied
|
||||||
|
|
||||||
from .utils import log
|
from .utils import log
|
||||||
from .event import Event
|
|
||||||
from .model import MessageModel
|
|
||||||
from .message import Message, MessageSegment
|
from .message import Message, MessageSegment
|
||||||
from .exception import NetworkError, ApiNotAvailable, ActionFailed, SessionExpired
|
from .exception import NetworkError, ApiNotAvailable, ActionFailed, SessionExpired
|
||||||
|
from .event import Event, MessageEvent, PrivateMessageEvent, GroupMessageEvent, ConversationType
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from nonebot.drivers import BaseDriver as Driver
|
from nonebot.drivers import Driver
|
||||||
|
|
||||||
|
SEND_BY_SESSION_WEBHOOK = "send_by_sessionWebhook"
|
||||||
|
|
||||||
|
|
||||||
class Bot(BaseBot):
|
class Bot(BaseBot):
|
||||||
@ -38,6 +40,7 @@ class Bot(BaseBot):
|
|||||||
return "ding"
|
return "ding"
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
@overrides(BaseBot)
|
||||||
async def check_permission(cls, driver: "Driver", connection_type: str,
|
async def check_permission(cls, driver: "Driver", connection_type: str,
|
||||||
headers: dict, body: Optional[dict]) -> str:
|
headers: dict, body: Optional[dict]) -> str:
|
||||||
"""
|
"""
|
||||||
@ -50,7 +53,8 @@ class Bot(BaseBot):
|
|||||||
|
|
||||||
# 检查连接方式
|
# 检查连接方式
|
||||||
if connection_type not in ["http"]:
|
if connection_type not in ["http"]:
|
||||||
raise RequestDenied(405, "Unsupported connection type")
|
raise RequestDenied(
|
||||||
|
405, "Unsupported connection type, available type: `http`")
|
||||||
|
|
||||||
# 检查 timestamp
|
# 检查 timestamp
|
||||||
if not timestamp:
|
if not timestamp:
|
||||||
@ -72,13 +76,25 @@ class Bot(BaseBot):
|
|||||||
log("WARNING", "Ding signature check ignored!")
|
log("WARNING", "Ding signature check ignored!")
|
||||||
return body["chatbotUserId"]
|
return body["chatbotUserId"]
|
||||||
|
|
||||||
async def handle_message(self, body: dict):
|
@overrides(BaseBot)
|
||||||
message = MessageModel.parse_obj(body)
|
async def handle_message(self, message: dict):
|
||||||
if not message:
|
if not message:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
# 判断消息类型,生成不同的 Event
|
||||||
|
try:
|
||||||
|
conversation_type = message["conversationType"]
|
||||||
|
if conversation_type == ConversationType.private:
|
||||||
|
event = PrivateMessageEvent.parse_obj(message)
|
||||||
|
elif conversation_type == ConversationType.group:
|
||||||
|
event = GroupMessageEvent.parse_obj(message)
|
||||||
|
else:
|
||||||
|
raise ValueError("Unsupported conversation type")
|
||||||
|
except Exception as e:
|
||||||
|
log("ERROR", "Event Parser Error", e)
|
||||||
|
return
|
||||||
|
|
||||||
try:
|
try:
|
||||||
event = Event(message)
|
|
||||||
await handle_event(self, event)
|
await handle_event(self, event)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.opt(colors=True, exception=e).error(
|
logger.opt(colors=True, exception=e).error(
|
||||||
@ -86,9 +102,10 @@ class Bot(BaseBot):
|
|||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
@overrides(BaseBot)
|
||||||
async def call_api(self,
|
async def call_api(self,
|
||||||
api: str,
|
api: str,
|
||||||
event: Optional[Event] = None,
|
event: Optional[MessageEvent] = None,
|
||||||
**data) -> Any:
|
**data) -> Any:
|
||||||
"""
|
"""
|
||||||
:说明:
|
:说明:
|
||||||
@ -120,28 +137,27 @@ class Bot(BaseBot):
|
|||||||
|
|
||||||
log("DEBUG", f"Calling API <y>{api}</y>")
|
log("DEBUG", f"Calling API <y>{api}</y>")
|
||||||
|
|
||||||
if api == "send_message":
|
if api == SEND_BY_SESSION_WEBHOOK:
|
||||||
if event:
|
if event:
|
||||||
# 确保 sessionWebhook 没有过期
|
# 确保 sessionWebhook 没有过期
|
||||||
if int(datetime.now().timestamp()) > int(
|
if int(datetime.now().timestamp()) > int(
|
||||||
event.raw_event.sessionWebhookExpiredTime / 1000):
|
event.sessionWebhookExpiredTime / 1000):
|
||||||
raise SessionExpired
|
raise SessionExpired
|
||||||
|
|
||||||
target = event.raw_event.sessionWebhook
|
target = event.sessionWebhook
|
||||||
else:
|
else:
|
||||||
target = None
|
|
||||||
|
|
||||||
if not target:
|
|
||||||
raise ApiNotAvailable
|
raise ApiNotAvailable
|
||||||
|
|
||||||
headers = {}
|
headers = {}
|
||||||
segment: MessageSegment = data["message"][0]
|
message: Message = data.get("message", None)
|
||||||
|
if not message:
|
||||||
|
raise ValueError("Message not found")
|
||||||
try:
|
try:
|
||||||
async with httpx.AsyncClient(headers=headers) as client:
|
async with httpx.AsyncClient(headers=headers) as client:
|
||||||
response = await client.post(
|
response = await client.post(
|
||||||
target,
|
target,
|
||||||
params={"access_token": self.config.access_token},
|
params={"access_token": self.config.access_token},
|
||||||
json=segment.data,
|
json=message._produce(),
|
||||||
timeout=self.config.api_timeout)
|
timeout=self.config.api_timeout)
|
||||||
|
|
||||||
if 200 <= response.status_code < 300:
|
if 200 <= response.status_code < 300:
|
||||||
@ -158,8 +174,9 @@ class Bot(BaseBot):
|
|||||||
except httpx.HTTPError:
|
except httpx.HTTPError:
|
||||||
raise NetworkError("HTTP request failed")
|
raise NetworkError("HTTP request failed")
|
||||||
|
|
||||||
|
@overrides(BaseBot)
|
||||||
async def send(self,
|
async def send(self,
|
||||||
event: Event,
|
event: MessageEvent,
|
||||||
message: Union[str, "Message", "MessageSegment"],
|
message: Union[str, "Message", "MessageSegment"],
|
||||||
at_sender: bool = False,
|
at_sender: bool = False,
|
||||||
**kwargs) -> Any:
|
**kwargs) -> Any:
|
||||||
@ -187,14 +204,14 @@ class Bot(BaseBot):
|
|||||||
"""
|
"""
|
||||||
msg = message if isinstance(message, Message) else Message(message)
|
msg = message if isinstance(message, Message) else Message(message)
|
||||||
|
|
||||||
at_sender = at_sender and bool(event.user_id)
|
at_sender = at_sender and bool(event.senderId)
|
||||||
params = {}
|
params = {}
|
||||||
params["event"] = event
|
params["event"] = event
|
||||||
params.update(kwargs)
|
params.update(kwargs)
|
||||||
|
|
||||||
if at_sender and event.detail_type != "private":
|
if at_sender and event.conversationType != ConversationType.private:
|
||||||
params["message"] = f"@{event.user_id} " + msg
|
params["message"] = f"@{event.senderNick} " + msg
|
||||||
else:
|
else:
|
||||||
params["message"] = msg
|
params["message"] = msg
|
||||||
|
|
||||||
return await self.call_api("send_message", **params)
|
return await self.call_api(SEND_BY_SESSION_WEBHOOK, **params)
|
||||||
|
@ -1,197 +1,142 @@
|
|||||||
from typing import Union, Optional
|
from enum import Enum
|
||||||
|
from typing import List, Optional
|
||||||
|
from typing_extensions import Literal
|
||||||
|
|
||||||
|
from pydantic import BaseModel, root_validator
|
||||||
|
|
||||||
|
from nonebot.typing import overrides
|
||||||
from nonebot.adapters import Event as BaseEvent
|
from nonebot.adapters import Event as BaseEvent
|
||||||
|
|
||||||
from .message import Message
|
from .message import Message
|
||||||
from .model import MessageModel, ConversationType, TextMessage
|
|
||||||
|
|
||||||
|
|
||||||
class Event(BaseEvent):
|
class Event(BaseEvent):
|
||||||
"""
|
"""
|
||||||
钉钉 协议 Event 适配。继承属性参考 `BaseEvent <./#class-baseevent>`_ 。
|
钉钉 协议 Event 适配。各事件字段参考 `钉钉文档`_
|
||||||
|
|
||||||
|
.. _钉钉文档:
|
||||||
|
https://ding-doc.dingtalk.com/document#/org-dev-guide/elzz1p
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, message: MessageModel):
|
chatbotUserId: str
|
||||||
super().__init__(message)
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def get_type(self) -> Literal["message", "notice", "request", "meta_event"]:
|
||||||
|
raise ValueError("Event has no type!")
|
||||||
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def get_event_name(self) -> str:
|
||||||
|
raise ValueError("Event has no name!")
|
||||||
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def get_event_description(self) -> str:
|
||||||
|
raise ValueError("Event has no description!")
|
||||||
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def get_message(self) -> "Message":
|
||||||
|
raise ValueError("Event has no message!")
|
||||||
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def get_plaintext(self) -> str:
|
||||||
|
raise ValueError("Event has no plaintext!")
|
||||||
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def get_user_id(self) -> str:
|
||||||
|
raise ValueError("Event has no user_id!")
|
||||||
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def get_session_id(self) -> str:
|
||||||
|
raise ValueError("Event has no session_id!")
|
||||||
|
|
||||||
|
@overrides(BaseEvent)
|
||||||
|
def is_tome(self) -> bool:
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
class TextMessage(BaseModel):
|
||||||
|
content: str
|
||||||
|
|
||||||
|
|
||||||
|
class AtUsersItem(BaseModel):
|
||||||
|
dingtalkId: str
|
||||||
|
staffId: Optional[str]
|
||||||
|
|
||||||
|
|
||||||
|
class ConversationType(str, Enum):
|
||||||
|
private = "1"
|
||||||
|
group = "2"
|
||||||
|
|
||||||
|
|
||||||
|
class MessageEvent(Event):
|
||||||
|
msgtype: str
|
||||||
|
text: TextMessage
|
||||||
|
msgId: str
|
||||||
|
createAt: int # ms
|
||||||
|
conversationType: ConversationType
|
||||||
|
conversationId: str
|
||||||
|
senderId: str
|
||||||
|
senderNick: str
|
||||||
|
senderCorpId: str
|
||||||
|
sessionWebhook: str
|
||||||
|
sessionWebhookExpiredTime: int
|
||||||
|
isAdmin: bool
|
||||||
|
|
||||||
|
message: Message
|
||||||
|
|
||||||
|
@root_validator(pre=True)
|
||||||
|
def gen_message(cls, values: dict):
|
||||||
|
assert "msgtype" in values, "msgtype must be specified"
|
||||||
# 其实目前钉钉机器人只能接收到 text 类型的消息
|
# 其实目前钉钉机器人只能接收到 text 类型的消息
|
||||||
self._message = Message(getattr(message, message.msgtype or "text"))
|
assert values[
|
||||||
|
"msgtype"] in values, f"{values['msgtype']} must be specified"
|
||||||
|
content = values[values['msgtype']]['content']
|
||||||
|
# 如果是被 @,第一个字符将会为空格,移除特殊情况
|
||||||
|
if content[0] == ' ':
|
||||||
|
content = content[1:]
|
||||||
|
values["message"] = content
|
||||||
|
return values
|
||||||
|
|
||||||
@property
|
@overrides(Event)
|
||||||
def raw_event(self) -> MessageModel:
|
def get_type(self) -> Literal["message", "notice", "request", "meta_event"]:
|
||||||
"""原始上报消息"""
|
|
||||||
return self._raw_event
|
|
||||||
|
|
||||||
@property
|
|
||||||
def id(self) -> Optional[str]:
|
|
||||||
"""
|
|
||||||
- 类型: ``Optional[str]``
|
|
||||||
- 说明: 消息 ID
|
|
||||||
"""
|
|
||||||
return self.raw_event.msgId
|
|
||||||
|
|
||||||
@property
|
|
||||||
def name(self) -> str:
|
|
||||||
"""
|
|
||||||
- 类型: ``str``
|
|
||||||
- 说明: 事件名称,由 `type`.`detail_type` 组合而成
|
|
||||||
"""
|
|
||||||
return self.type + "." + self.detail_type
|
|
||||||
|
|
||||||
@property
|
|
||||||
def self_id(self) -> str:
|
|
||||||
"""
|
|
||||||
- 类型: ``str``
|
|
||||||
- 说明: 机器人自身 ID
|
|
||||||
"""
|
|
||||||
return str(self.raw_event.chatbotUserId)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def time(self) -> int:
|
|
||||||
"""
|
|
||||||
- 类型: ``int``
|
|
||||||
- 说明: 消息的时间戳,单位 s
|
|
||||||
"""
|
|
||||||
# 单位 ms -> s
|
|
||||||
return int(self.raw_event.createAt / 1000)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def type(self) -> str:
|
|
||||||
"""
|
|
||||||
- 类型: ``str``
|
|
||||||
- 说明: 事件类型
|
|
||||||
"""
|
|
||||||
return "message"
|
return "message"
|
||||||
|
|
||||||
@type.setter
|
@overrides(Event)
|
||||||
def type(self, value) -> None:
|
def get_event_name(self) -> str:
|
||||||
pass
|
return f"{self.get_type()}.{self.conversationType.name}"
|
||||||
|
|
||||||
@property
|
@overrides(Event)
|
||||||
def detail_type(self) -> str:
|
def get_event_description(self) -> str:
|
||||||
"""
|
return f'Message[{self.msgtype}] {self.msgId} from {self.senderId} "{self.text.content}"'
|
||||||
- 类型: ``str``
|
|
||||||
- 说明: 事件详细类型
|
|
||||||
"""
|
|
||||||
return self.raw_event.conversationType.name
|
|
||||||
|
|
||||||
@detail_type.setter
|
@overrides(Event)
|
||||||
def detail_type(self, value) -> None:
|
def get_message(self) -> Message:
|
||||||
if value == "private":
|
return self.message
|
||||||
self.raw_event.conversationType = ConversationType.private
|
|
||||||
if value == "group":
|
|
||||||
self.raw_event.conversationType = ConversationType.group
|
|
||||||
|
|
||||||
@property
|
@overrides(Event)
|
||||||
def sub_type(self) -> None:
|
def get_plaintext(self) -> str:
|
||||||
"""
|
return self.text.content
|
||||||
- 类型: ``None``
|
|
||||||
- 说明: 钉钉适配器无事件子类型
|
|
||||||
"""
|
|
||||||
return None
|
|
||||||
|
|
||||||
@sub_type.setter
|
@overrides(Event)
|
||||||
def sub_type(self, value) -> None:
|
def get_user_id(self) -> str:
|
||||||
pass
|
return self.senderId
|
||||||
|
|
||||||
@property
|
@overrides(Event)
|
||||||
def user_id(self) -> Optional[str]:
|
def get_session_id(self) -> str:
|
||||||
"""
|
return self.senderId
|
||||||
- 类型: ``Optional[str]``
|
|
||||||
- 说明: 发送者 ID
|
|
||||||
"""
|
|
||||||
return self.raw_event.senderId
|
|
||||||
|
|
||||||
@user_id.setter
|
|
||||||
def user_id(self, value) -> None:
|
|
||||||
self.raw_event.senderId = value
|
|
||||||
|
|
||||||
@property
|
class PrivateMessageEvent(MessageEvent):
|
||||||
def group_id(self) -> Optional[str]:
|
chatbotCorpId: str
|
||||||
"""
|
senderStaffId: Optional[str]
|
||||||
- 类型: ``Optional[str]``
|
conversationType: ConversationType = ConversationType.private
|
||||||
- 说明: 事件主体群 ID
|
|
||||||
"""
|
|
||||||
return self.raw_event.conversationId
|
|
||||||
|
|
||||||
@group_id.setter
|
|
||||||
def group_id(self, value) -> None:
|
|
||||||
self.raw_event.conversationId = value
|
|
||||||
|
|
||||||
@property
|
class GroupMessageEvent(MessageEvent):
|
||||||
def to_me(self) -> Optional[bool]:
|
atUsers: List[AtUsersItem]
|
||||||
"""
|
conversationType: ConversationType = ConversationType.group
|
||||||
- 类型: ``Optional[bool]``
|
conversationTitle: str
|
||||||
- 说明: 消息是否与机器人相关
|
isInAtList: bool
|
||||||
"""
|
|
||||||
return self.detail_type == "private" or self.raw_event.isInAtList
|
|
||||||
|
|
||||||
@property
|
@overrides(MessageEvent)
|
||||||
def message(self) -> Optional["Message"]:
|
def is_tome(self) -> bool:
|
||||||
"""
|
return self.isInAtList
|
||||||
- 类型: ``Optional[Message]``
|
|
||||||
- 说明: 消息内容
|
|
||||||
"""
|
|
||||||
return self._message
|
|
||||||
|
|
||||||
@message.setter
|
|
||||||
def message(self, value) -> None:
|
|
||||||
self._message = value
|
|
||||||
|
|
||||||
@property
|
|
||||||
def reply(self) -> None:
|
|
||||||
"""
|
|
||||||
- 类型: ``None``
|
|
||||||
- 说明: 回复消息详情
|
|
||||||
"""
|
|
||||||
raise ValueError("暂不支持 reply")
|
|
||||||
|
|
||||||
@property
|
|
||||||
def raw_message(self) -> Optional[Union[TextMessage]]:
|
|
||||||
"""
|
|
||||||
- 类型: ``Optional[str]``
|
|
||||||
- 说明: 原始消息
|
|
||||||
"""
|
|
||||||
return getattr(self.raw_event, self.raw_event.msgtype)
|
|
||||||
|
|
||||||
@raw_message.setter
|
|
||||||
def raw_message(self, value) -> None:
|
|
||||||
setattr(self.raw_event, self.raw_event.msgtype, value)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def plain_text(self) -> Optional[str]:
|
|
||||||
"""
|
|
||||||
- 类型: ``Optional[str]``
|
|
||||||
- 说明: 纯文本消息内容
|
|
||||||
"""
|
|
||||||
return self.message and self.message.extract_plain_text().strip()
|
|
||||||
|
|
||||||
@property
|
|
||||||
def sender(self) -> Optional[dict]:
|
|
||||||
"""
|
|
||||||
- 类型: ``Optional[dict]``
|
|
||||||
- 说明: 消息发送者信息
|
|
||||||
"""
|
|
||||||
result = {
|
|
||||||
# 加密的发送者ID。
|
|
||||||
"senderId": self.raw_event.senderId,
|
|
||||||
# 发送者昵称。
|
|
||||||
"senderNick": self.raw_event.senderNick,
|
|
||||||
# 企业内部群有的发送者当前群的企业 corpId。
|
|
||||||
"senderCorpId": self.raw_event.senderCorpId,
|
|
||||||
# 企业内部群有的发送者在企业内的 userId。
|
|
||||||
"senderStaffId": self.raw_event.senderStaffId,
|
|
||||||
"role": "admin" if self.raw_event.isAdmin else "member"
|
|
||||||
}
|
|
||||||
return result
|
|
||||||
|
|
||||||
@sender.setter
|
|
||||||
def sender(self, value) -> None:
|
|
||||||
|
|
||||||
def set_wrapper(name):
|
|
||||||
if value.get(name):
|
|
||||||
setattr(self.raw_event, name, value.get(name))
|
|
||||||
|
|
||||||
set_wrapper("senderId")
|
|
||||||
set_wrapper("senderNick")
|
|
||||||
set_wrapper("senderCorpId")
|
|
||||||
set_wrapper("senderStaffId")
|
|
||||||
|
@ -37,7 +37,10 @@ class ActionFailed(BaseActionFailed, DingAdapterException):
|
|||||||
self.errmsg = errmsg
|
self.errmsg = errmsg
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return f"<ApiError errcode={self.errcode} errmsg={self.errmsg}>"
|
return f"<ApiError errcode={self.errcode} errmsg=\"{self.errmsg}\">"
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
class ApiNotAvailable(BaseApiNotAvailable, DingAdapterException):
|
class ApiNotAvailable(BaseApiNotAvailable, DingAdapterException):
|
||||||
@ -66,7 +69,7 @@ class NetworkError(BaseNetworkError, DingAdapterException):
|
|||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
class SessionExpired(BaseApiNotAvailable, DingAdapterException):
|
class SessionExpired(ApiNotAvailable, DingAdapterException):
|
||||||
"""
|
"""
|
||||||
:说明:
|
:说明:
|
||||||
|
|
||||||
@ -75,3 +78,6 @@ class SessionExpired(BaseApiNotAvailable, DingAdapterException):
|
|||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return f"<Session Webhook is Expired>"
|
return f"<Session Webhook is Expired>"
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return self.__repr__()
|
||||||
|
@ -1,82 +1,87 @@
|
|||||||
|
from copy import copy
|
||||||
from typing import Any, Dict, Union, Iterable
|
from typing import Any, Dict, Union, Iterable
|
||||||
|
|
||||||
|
from nonebot.typing import overrides
|
||||||
from nonebot.adapters import Message as BaseMessage, MessageSegment as BaseMessageSegment
|
from nonebot.adapters import Message as BaseMessage, MessageSegment as BaseMessageSegment
|
||||||
|
|
||||||
from .utils import log
|
|
||||||
from .model import TextMessage
|
|
||||||
|
|
||||||
|
|
||||||
class MessageSegment(BaseMessageSegment):
|
class MessageSegment(BaseMessageSegment):
|
||||||
"""
|
"""
|
||||||
钉钉 协议 MessageSegment 适配。具体方法参考协议消息段类型或源码。
|
钉钉 协议 MessageSegment 适配。具体方法参考协议消息段类型或源码。
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, type_: str, msg: Dict[str, Any]) -> None:
|
@overrides(BaseMessageSegment)
|
||||||
data = {
|
def __init__(self, type_: str, data: Dict[str, Any]) -> None:
|
||||||
"msgtype": type_,
|
|
||||||
}
|
|
||||||
if msg:
|
|
||||||
data.update(msg)
|
|
||||||
log("DEBUG", f"data {data}")
|
|
||||||
super().__init__(type=type_, data=data)
|
super().__init__(type=type_, data=data)
|
||||||
|
|
||||||
@classmethod
|
@overrides(BaseMessageSegment)
|
||||||
def from_segment(cls, segment: "MessageSegment"):
|
|
||||||
return MessageSegment(segment.type, segment.data)
|
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
log("DEBUG", f"__str__: self.type {self.type} data {self.data}")
|
|
||||||
if self.type == "text":
|
if self.type == "text":
|
||||||
return str(self.data["text"]["content"].strip())
|
return str(self.data["content"])
|
||||||
|
elif self.type == "markdown":
|
||||||
|
return str(self.data["text"])
|
||||||
return ""
|
return ""
|
||||||
|
|
||||||
|
@overrides(BaseMessageSegment)
|
||||||
def __add__(self, other) -> "Message":
|
def __add__(self, other) -> "Message":
|
||||||
if isinstance(other, str):
|
|
||||||
if self.type == 'text':
|
|
||||||
self.data['text']['content'] += other
|
|
||||||
return MessageSegment.from_segment(self)
|
|
||||||
return Message(self) + other
|
return Message(self) + other
|
||||||
|
|
||||||
def atMobile(self, mobileNumber):
|
@overrides(BaseMessageSegment)
|
||||||
self.data.setdefault("at", {})
|
def __radd__(self, other) -> "Message":
|
||||||
self.data["at"].setdefault("atMobiles", [])
|
return Message(other) + self
|
||||||
self.data["at"]["atMobiles"].append(mobileNumber)
|
|
||||||
|
|
||||||
def atAll(self, value):
|
@overrides(BaseMessageSegment)
|
||||||
self.data.setdefault("at", {})
|
def is_text(self) -> bool:
|
||||||
self.data["at"]["isAtAll"] = value
|
return self.type == "text"
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def text(text_: str) -> "MessageSegment":
|
def atAll() -> "MessageSegment":
|
||||||
return MessageSegment("text", {"text": {"content": text_.strip()}})
|
return MessageSegment("at", {"isAtAll": True})
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def atMobiles(*mobileNumber: str) -> "MessageSegment":
|
||||||
|
return MessageSegment("at", {"atMobiles": list(mobileNumber)})
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def text(text: str) -> "MessageSegment":
|
||||||
|
return MessageSegment("text", {"content": text})
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def image(picURL: str) -> "MessageSegment":
|
||||||
|
return MessageSegment("image", {"picURL": picURL})
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def extension(dict_: dict) -> "MessageSegment":
|
||||||
|
""""标记 text 文本的 extension 属性,需要与 text 消息段相加。
|
||||||
|
"""
|
||||||
|
return MessageSegment("extension", dict_)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def markdown(title: str, text: str) -> "MessageSegment":
|
def markdown(title: str, text: str) -> "MessageSegment":
|
||||||
return MessageSegment("markdown", {
|
return MessageSegment(
|
||||||
"markdown": {
|
"markdown",
|
||||||
|
{
|
||||||
"title": title,
|
"title": title,
|
||||||
"text": text,
|
"text": text,
|
||||||
},
|
},
|
||||||
})
|
)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def actionCardSingleBtn(title: str, text: str, btnTitle: str,
|
def actionCardSingleBtn(title: str, text: str, singleTitle: str,
|
||||||
btnUrl) -> "MessageSegment":
|
singleURL) -> "MessageSegment":
|
||||||
return MessageSegment(
|
return MessageSegment(
|
||||||
"actionCard", {
|
"actionCard", {
|
||||||
"actionCard": {
|
|
||||||
"title": title,
|
"title": title,
|
||||||
"text": text,
|
"text": text,
|
||||||
"singleTitle": btnTitle,
|
"singleTitle": singleTitle,
|
||||||
"singleURL": btnUrl
|
"singleURL": singleURL
|
||||||
}
|
|
||||||
})
|
})
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def actionCardSingleMultiBtns(
|
def actionCardMultiBtns(
|
||||||
title: str,
|
title: str,
|
||||||
text: str,
|
text: str,
|
||||||
btns: list = [],
|
btns: list,
|
||||||
hideAvatar: bool = False,
|
hideAvatar: bool = False,
|
||||||
btnOrientation: str = '1',
|
btnOrientation: str = '1',
|
||||||
) -> "MessageSegment":
|
) -> "MessageSegment":
|
||||||
@ -89,28 +94,36 @@ class MessageSegment(BaseMessageSegment):
|
|||||||
"""
|
"""
|
||||||
return MessageSegment(
|
return MessageSegment(
|
||||||
"actionCard", {
|
"actionCard", {
|
||||||
"actionCard": {
|
|
||||||
"title": title,
|
"title": title,
|
||||||
"text": text,
|
"text": text,
|
||||||
"hideAvatar": "1" if hideAvatar else "0",
|
"hideAvatar": "1" if hideAvatar else "0",
|
||||||
"btnOrientation": btnOrientation,
|
"btnOrientation": btnOrientation,
|
||||||
"btns": btns
|
"btns": btns
|
||||||
}
|
|
||||||
})
|
})
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def feedCard(links: list = [],) -> "MessageSegment":
|
def feedCard(links: list) -> "MessageSegment":
|
||||||
"""
|
"""
|
||||||
:参数:
|
:参数:
|
||||||
|
|
||||||
* ``links``: [{ "title": xxx, "messageURL": xxx, "picURL": xxx }, ...]
|
* ``links``: [{ "title": xxx, "messageURL": xxx, "picURL": xxx }, ...]
|
||||||
"""
|
"""
|
||||||
return MessageSegment("feedCard", {"feedCard": {"links": links}})
|
return MessageSegment("feedCard", {"links": links})
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def empty() -> "MessageSegment":
|
def raw(data) -> "MessageSegment":
|
||||||
"""不想回复消息到群里"""
|
return MessageSegment('raw', data)
|
||||||
return MessageSegment("empty")
|
|
||||||
|
def to_dict(self) -> dict:
|
||||||
|
# 让用户可以直接发送原始的消息格式
|
||||||
|
if self.type == "raw":
|
||||||
|
return copy(self.data)
|
||||||
|
|
||||||
|
# 不属于消息内容,只是作为消息段的辅助
|
||||||
|
if self.type in ["at", "extension"]:
|
||||||
|
return {self.type: copy(self.data)}
|
||||||
|
|
||||||
|
return {"msgtype": self.type, self.type: copy(self.data)}
|
||||||
|
|
||||||
|
|
||||||
class Message(BaseMessage):
|
class Message(BaseMessage):
|
||||||
@ -119,17 +132,24 @@ class Message(BaseMessage):
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _construct(
|
@overrides(BaseMessage)
|
||||||
msg: Union[str, dict, list,
|
def _construct(msg: Union[str, dict, list]) -> Iterable[MessageSegment]:
|
||||||
TextMessage]) -> Iterable[MessageSegment]:
|
|
||||||
if isinstance(msg, dict):
|
if isinstance(msg, dict):
|
||||||
yield MessageSegment(msg["type"], msg.get("data") or {})
|
yield MessageSegment(msg["type"], msg.get("data") or {})
|
||||||
return
|
|
||||||
elif isinstance(msg, list):
|
elif isinstance(msg, list):
|
||||||
for seg in msg:
|
for seg in msg:
|
||||||
yield MessageSegment(seg["type"], seg.get("data") or {})
|
yield MessageSegment(seg["type"], seg.get("data") or {})
|
||||||
return
|
|
||||||
elif isinstance(msg, TextMessage):
|
|
||||||
yield MessageSegment("text", {"text": msg.dict()})
|
|
||||||
elif isinstance(msg, str):
|
elif isinstance(msg, str):
|
||||||
yield MessageSegment.text(msg)
|
yield MessageSegment.text(msg)
|
||||||
|
|
||||||
|
def _produce(self) -> dict:
|
||||||
|
data = {}
|
||||||
|
for segment in self:
|
||||||
|
# text 可以和 text 合并
|
||||||
|
if segment.type == "text" and data.get("msgtype") == 'text':
|
||||||
|
data.setdefault("text", {})
|
||||||
|
data["text"]["content"] = data["text"].setdefault(
|
||||||
|
"content", "") + segment.data["content"]
|
||||||
|
else:
|
||||||
|
data.update(segment.to_dict())
|
||||||
|
return data
|
||||||
|
@ -1,48 +0,0 @@
|
|||||||
from enum import Enum
|
|
||||||
from typing import List, Optional
|
|
||||||
|
|
||||||
from pydantic import BaseModel
|
|
||||||
|
|
||||||
|
|
||||||
class Headers(BaseModel):
|
|
||||||
sign: str
|
|
||||||
token: str
|
|
||||||
# ms
|
|
||||||
timestamp: int
|
|
||||||
|
|
||||||
|
|
||||||
class TextMessage(BaseModel):
|
|
||||||
content: str
|
|
||||||
|
|
||||||
|
|
||||||
class AtUsersItem(BaseModel):
|
|
||||||
dingtalkId: str
|
|
||||||
staffId: Optional[str]
|
|
||||||
|
|
||||||
|
|
||||||
class ConversationType(str, Enum):
|
|
||||||
private = '1'
|
|
||||||
group = '2'
|
|
||||||
|
|
||||||
|
|
||||||
class MessageModel(BaseModel):
|
|
||||||
msgtype: str = None
|
|
||||||
text: Optional[TextMessage] = None
|
|
||||||
msgId: str
|
|
||||||
# ms
|
|
||||||
createAt: int = None
|
|
||||||
conversationType: ConversationType = None
|
|
||||||
conversationId: str = None
|
|
||||||
conversationTitle: str = None
|
|
||||||
senderId: str = None
|
|
||||||
senderNick: str = None
|
|
||||||
senderCorpId: str = None
|
|
||||||
senderStaffId: str = None
|
|
||||||
chatbotUserId: str = None
|
|
||||||
chatbotCorpId: str = None
|
|
||||||
atUsers: List[AtUsersItem] = None
|
|
||||||
sessionWebhook: str = None
|
|
||||||
# ms
|
|
||||||
sessionWebhookExpiredTime: int = None
|
|
||||||
isAdmin: bool = None
|
|
||||||
isInAtList: bool = None
|
|
@ -113,7 +113,7 @@ class Matcher(metaclass=MatcherMeta):
|
|||||||
self.state = self._default_state.copy()
|
self.state = self._default_state.copy()
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return (f"<Matcher from {self.module or 'unknow'}, type={self.type}, "
|
return (f"<Matcher from {self.module or 'unknown'}, type={self.type}, "
|
||||||
f"priority={self.priority}, temp={self.temp}>")
|
f"priority={self.priority}, temp={self.temp}>")
|
||||||
|
|
||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
@ -460,13 +460,23 @@ class Matcher(metaclass=MatcherMeta):
|
|||||||
if not hasattr(handler, "__params__"):
|
if not hasattr(handler, "__params__"):
|
||||||
self.process_handler(handler)
|
self.process_handler(handler)
|
||||||
params = getattr(handler, "__params__")
|
params = getattr(handler, "__params__")
|
||||||
|
|
||||||
BotType = ((params["bot"] is not inspect.Parameter.empty) and
|
BotType = ((params["bot"] is not inspect.Parameter.empty) and
|
||||||
inspect.isclass(params["bot"]) and params["bot"])
|
inspect.isclass(params["bot"]) and params["bot"])
|
||||||
|
if BotType and not isinstance(bot, BotType):
|
||||||
|
logger.debug(
|
||||||
|
f"Matcher {self} bot type {type(bot)} not match annotation {BotType}, ignored"
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
EventType = ((params["event"] is not inspect.Parameter.empty) and
|
EventType = ((params["event"] is not inspect.Parameter.empty) and
|
||||||
inspect.isclass(params["event"]) and params["event"])
|
inspect.isclass(params["event"]) and params["event"])
|
||||||
if (BotType and not isinstance(bot, BotType)) or (
|
if EventType and not isinstance(event, EventType):
|
||||||
EventType and not isinstance(event, EventType)):
|
logger.debug(
|
||||||
|
f"Matcher {self} event type {type(event)} not match annotation {EventType}, ignored"
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
args = {"bot": bot, "event": event, "state": state, "matcher": self}
|
args = {"bot": bot, "event": event, "state": state, "matcher": self}
|
||||||
await handler(
|
await handler(
|
||||||
**{k: v for k, v in args.items() if params[k] is not None})
|
**{k: v for k, v in args.items() if params[k] is not None})
|
||||||
|
160
tests/test_plugins/test_ding.py
Normal file
160
tests/test_plugins/test_ding.py
Normal file
@ -0,0 +1,160 @@
|
|||||||
|
from nonebot.rule import to_me
|
||||||
|
from nonebot.plugin import on_command
|
||||||
|
from nonebot.adapters.ding import Bot as DingBot, MessageSegment, MessageEvent
|
||||||
|
|
||||||
|
markdown = on_command("markdown", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@markdown.handle()
|
||||||
|
async def test_handler(bot: DingBot):
|
||||||
|
message = MessageSegment.markdown(
|
||||||
|
"Hello, This is NoneBot",
|
||||||
|
"#### NoneBot \n> Nonebot 是一款高性能的 Python 机器人框架\n> \n> [GitHub 仓库地址](https://github.com/nonebot/nonebot2) \n"
|
||||||
|
)
|
||||||
|
await markdown.finish(message)
|
||||||
|
|
||||||
|
|
||||||
|
actionCardSingleBtn = on_command("actionCardSingleBtn", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@actionCardSingleBtn.handle()
|
||||||
|
async def test_handler(bot: DingBot):
|
||||||
|
message = MessageSegment.actionCardSingleBtn(
|
||||||
|
title="打造一间咖啡厅",
|
||||||
|
text=
|
||||||
|
" \n #### 乔布斯 20 年前想打造的苹果咖啡厅 \n\n Apple Store 的设计正从原来满满的科技感走向生活化,而其生活化的走向其实可以追溯到 20 年前苹果一个建立咖啡馆的计划",
|
||||||
|
singleTitle="阅读全文",
|
||||||
|
singleURL="https://www.dingtalk.com/")
|
||||||
|
await actionCardSingleBtn.finish(message)
|
||||||
|
|
||||||
|
|
||||||
|
actionCard = on_command("actionCard", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@actionCard.handle()
|
||||||
|
async def test_handler(bot: DingBot):
|
||||||
|
message = MessageSegment.raw({
|
||||||
|
"msgtype": "actionCard",
|
||||||
|
"actionCard": {
|
||||||
|
"title":
|
||||||
|
"乔布斯 20 年前想打造一间苹果咖啡厅,而它正是 Apple Store 的前身",
|
||||||
|
"text":
|
||||||
|
" \n\n #### 乔布斯 20 年前想打造的苹果咖啡厅 \n\n Apple Store 的设计正从原来满满的科技感走向生活化,而其生活化的走向其实可以追溯到 20 年前苹果一个建立咖啡馆的计划",
|
||||||
|
"hideAvatar":
|
||||||
|
"0",
|
||||||
|
"btnOrientation":
|
||||||
|
"0",
|
||||||
|
"btns": [{
|
||||||
|
"title": "内容不错",
|
||||||
|
"actionURL": "https://www.dingtalk.com/"
|
||||||
|
}, {
|
||||||
|
"title": "不感兴趣",
|
||||||
|
"actionURL": "https://www.dingtalk.com/"
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
})
|
||||||
|
await actionCard.finish(message)
|
||||||
|
|
||||||
|
|
||||||
|
feedCard = on_command("feedCard", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@feedCard.handle()
|
||||||
|
async def test_handler(bot: DingBot):
|
||||||
|
message = MessageSegment.raw({
|
||||||
|
"msgtype": "feedCard",
|
||||||
|
"feedCard": {
|
||||||
|
"links": [{
|
||||||
|
"title":
|
||||||
|
"时代的火车向前开1",
|
||||||
|
"messageURL":
|
||||||
|
"https://www.dingtalk.com/",
|
||||||
|
"picURL":
|
||||||
|
"https://img.alicdn.com/tfs/TB1NwmBEL9TBuNjy1zbXXXpepXa-2400-1218.png"
|
||||||
|
}, {
|
||||||
|
"title":
|
||||||
|
"时代的火车向前开2",
|
||||||
|
"messageURL":
|
||||||
|
"https://www.dingtalk.com/",
|
||||||
|
"picURL":
|
||||||
|
"https://img.alicdn.com/tfs/TB1NwmBEL9TBuNjy1zbXXXpepXa-2400-1218.png"
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
})
|
||||||
|
await feedCard.finish(message)
|
||||||
|
|
||||||
|
|
||||||
|
atme = on_command("atme", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@atme.handle()
|
||||||
|
async def test_handler(bot: DingBot, event: MessageEvent):
|
||||||
|
message = f"@{event.senderNick} at you" + MessageSegment.atMobiles(
|
||||||
|
"13800000001")
|
||||||
|
await atme.finish(message)
|
||||||
|
|
||||||
|
|
||||||
|
image = on_command("image", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@image.handle()
|
||||||
|
async def test_handler(bot: DingBot, event: MessageEvent):
|
||||||
|
message = MessageSegment.image(
|
||||||
|
"https://static-aliyun-doc.oss-accelerate.aliyuncs.com/assets/img/zh-CN/0634199951/p158167.png"
|
||||||
|
)
|
||||||
|
await image.finish(message)
|
||||||
|
|
||||||
|
|
||||||
|
textAdd = on_command("t", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@textAdd.handle()
|
||||||
|
async def test_handler(bot: DingBot, event: MessageEvent):
|
||||||
|
message = "第一段消息\n" + MessageSegment.text("asdawefaefa\n")
|
||||||
|
await textAdd.send(message)
|
||||||
|
|
||||||
|
message = message + MessageSegment.text("第二段消息\n")
|
||||||
|
await textAdd.send(message)
|
||||||
|
|
||||||
|
message = message + MessageSegment.text(
|
||||||
|
"\n第三段消息\n") + "adfkasfkhsdkfahskdjasdashdkjasdf"
|
||||||
|
message = message + MessageSegment.extension({
|
||||||
|
"text_type": "code_snippet",
|
||||||
|
"code_language": "C#"
|
||||||
|
})
|
||||||
|
await textAdd.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
code = on_command("code", to_me())
|
||||||
|
|
||||||
|
|
||||||
|
@code.handle()
|
||||||
|
async def test_handler(bot: DingBot, event: MessageEvent):
|
||||||
|
raw = MessageSegment.raw({
|
||||||
|
"msgtype": "text",
|
||||||
|
"text": {
|
||||||
|
"content": 'print("hello world")'
|
||||||
|
},
|
||||||
|
"extension": {
|
||||||
|
"text_type": "code_snippet",
|
||||||
|
"code_language": "Python",
|
||||||
|
}
|
||||||
|
})
|
||||||
|
await code.send(raw)
|
||||||
|
message = MessageSegment.text("""using System;
|
||||||
|
|
||||||
|
namespace HelloWorld
|
||||||
|
{
|
||||||
|
class Program
|
||||||
|
{
|
||||||
|
static void Main(string[] args)
|
||||||
|
{
|
||||||
|
Console.WriteLine("Hello World!");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}""")
|
||||||
|
message += MessageSegment.extension({
|
||||||
|
"text_type": "code_snippet",
|
||||||
|
"code_language": "C#"
|
||||||
|
})
|
||||||
|
await code.finish(message)
|
Loading…
x
Reference in New Issue
Block a user