mirror of
https://github.com/ChenXu233/nonebot_plugin_dialectlist.git
synced 2026-01-25 21:22:16 +00:00
Compare commits
34 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3434a5352f | ||
|
|
0954eb4f3e | ||
|
|
5b94bc0fff | ||
|
|
e1a20922c4 | ||
|
|
5e00605dfb | ||
|
|
833cdb4a75 | ||
|
|
29b4598e5f | ||
|
|
10cc5cf278 | ||
|
|
ed167a56b5 | ||
|
|
b40c7be1f9 | ||
|
|
07f85be527 | ||
|
|
f2cdfe7f8a | ||
|
|
e0ca7907c0 | ||
|
|
53b2a33923 | ||
|
|
4e339bcdb1 | ||
|
|
0b85f6fb34 | ||
|
|
83d9cda2eb | ||
|
|
dfde832dba | ||
|
|
987b34e282 | ||
|
|
49ce2ffb7c | ||
|
|
fcce94250c | ||
|
|
53e52eabb5 | ||
|
|
633ffbde9a | ||
|
|
d7baa9b04b | ||
|
|
df2cd495c0 | ||
|
|
d2e185c189 | ||
|
|
d42140cc95 | ||
|
|
a753305282 | ||
|
|
ad5f556229 | ||
|
|
3efaafe81c | ||
|
|
c6b87bf981 | ||
|
|
e62f2ed488 | ||
|
|
e5d107c520 | ||
|
|
bf577b5a31 |
1
.vscode/settings.json
vendored
1
.vscode/settings.json
vendored
@@ -19,6 +19,7 @@
|
|||||||
"pygal",
|
"pygal",
|
||||||
"sqlalchemy",
|
"sqlalchemy",
|
||||||
"timecost",
|
"timecost",
|
||||||
|
"uninfo",
|
||||||
"userinfo",
|
"userinfo",
|
||||||
"whereclause",
|
"whereclause",
|
||||||
"xaxis",
|
"xaxis",
|
||||||
|
|||||||
21
LICENSE
Normal file
21
LICENSE
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
MIT License
|
||||||
|
|
||||||
|
Copyright (c) 2024 ChenXu233
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
||||||
94
README.md
94
README.md
@@ -26,15 +26,16 @@ nonebot-plugin-dialectlist
|
|||||||
|
|
||||||
## 💿 安装
|
## 💿 安装
|
||||||
|
|
||||||
通过`pip`或`nb`安装;
|
通过`pip`或`nb`安装:
|
||||||
|
- 通过 pip 安装
|
||||||
|
```bash
|
||||||
|
pip install nonebot-plugin-dialectlist
|
||||||
|
```
|
||||||
|
- 通过 nb-cli 安装
|
||||||
|
```bash
|
||||||
|
nb plugin install nonebot-plugin-dialectlist
|
||||||
|
```
|
||||||
|
|
||||||
>**通过 pip **安装
|
|
||||||
|
|
||||||
`pip install nonebot-plugin-dialectlist`
|
|
||||||
|
|
||||||
>**通过 nb **安装
|
|
||||||
|
|
||||||
`nb plugin install nonebot-plugin-dialectlist`
|
|
||||||
|
|
||||||
### ✅ 插件依赖于
|
### ✅ 插件依赖于
|
||||||
|
|
||||||
@@ -53,15 +54,20 @@ nonebot-plugin-dialectlist
|
|||||||
|
|
||||||
在 .env 中,可以添加以下配置项
|
在 .env 中,可以添加以下配置项
|
||||||
```python
|
```python
|
||||||
dialectlist__string_format = "第{index}名:\n{nickname},{chatdatanum}条消息、\n" #消息格式
|
dialectlist__get_num: int = 5 # 获取人数数量
|
||||||
dialectlist__get_num = 10 #获取人数数量
|
dialectlist__font: str = "SimHei" # 字体格式
|
||||||
dialectlist__visualization = True #是否可视化
|
dialectlist__suffix: bool = True # 是否显示后缀
|
||||||
# dialectlist__visualization_type = "圆环图" #可视化方案 (不再支持)
|
dialectlist__excluded_self: bool = True # 是否排除自己
|
||||||
dialectlist__font = "SimHei" #字体格式
|
dialectlist__visualization: bool = True # 是否可视化
|
||||||
dialectlist__excluded_people = [] #排除的人的 QQ 号(或频道号?(未经测试))
|
dialectlist__show_text_rank: bool = True # 是否显示文本排名
|
||||||
dialectlist__excluded_self = True #是否排除机器人自己 QQ
|
dialectlist__counting_cache: bool = False # 计数缓存(能够提高回复速度)
|
||||||
dialectlist__suffix: bool = False # 是否显示后缀
|
dialectlist__excluded_people: List[str] = [] # 排除的人的QQ号
|
||||||
dialectlist__string_suffix: str = "统计花费时间{timecost}" # 消息格式后缀
|
dialectlist__use_user_info_cache: bool = False # 是否使用用户信息缓存
|
||||||
|
dialectlist__aggregate_transmission: bool = False # 是否聚合转发消息
|
||||||
|
dialectlist__timezone: Optional[str] = "Asia/Shanghai" # 时区,影响统计时间
|
||||||
|
dialectlist__string_suffix: str = "统计花费时间:{timecost}秒" # 消息格式后缀
|
||||||
|
dialectlist__template_path: str = "./template/rank_template.j2" # 模板路径
|
||||||
|
dialectlist__string_format: str = "第{index}名:\n{nickname},{chatdatanum}条消息\n" # 消息格式
|
||||||
```
|
```
|
||||||
💭也可以不进行配置,这将会使插件按照默认配置运行
|
💭也可以不进行配置,这将会使插件按照默认配置运行
|
||||||
|
|
||||||
@@ -76,6 +82,8 @@ __!!注意!!__
|
|||||||
|
|
||||||
### 🎨一般用法
|
### 🎨一般用法
|
||||||
|
|
||||||
|
#### B话榜
|
||||||
|
|
||||||
-`/B话榜` ————看看有史以来(机器人存在以来)群友们发了多少消息! (好像没写)
|
-`/B话榜` ————看看有史以来(机器人存在以来)群友们发了多少消息! (好像没写)
|
||||||
|
|
||||||
-`/今日B话榜` ————看看今天的群友发了多少消息!
|
-`/今日B话榜` ————看看今天的群友发了多少消息!
|
||||||
@@ -94,11 +102,37 @@ __!!注意!!__
|
|||||||
|
|
||||||
-`/历史B话榜` ————看看历史上(机器人存在以来)的群友发了多少消息!
|
-`/历史B话榜` ————看看历史上(机器人存在以来)的群友发了多少消息!
|
||||||
|
|
||||||
|
#### 看看B话(kkb)
|
||||||
|
|
||||||
|
-`/看看B话 [@某人|QQ号]` ————看看这个b人在这个b群发了多少b话!
|
||||||
|
|
||||||
### 🚀进阶用法
|
### 🚀进阶用法
|
||||||
|
|
||||||
`/{时间类型(今日|年度)?}{B话榜|废话榜} {时间类型?} {ISO8601 格式时间} {群号}`
|
#### B话榜
|
||||||
|
|
||||||
如:`/B话榜 历史 2024-01-01~2024-01-02 12345678`
|
`/{时间类型(今日|年度)?}{B话榜|废话榜} {时间类型?} {ISO8601 格式时间 ?} {群号} {关键词}`
|
||||||
|
|
||||||
|
如:`/B话榜 历史 2024-01-01~2024-01-02 12345678 女装`
|
||||||
|
|
||||||
|
也可以 `/{时间类型(今日|年度)?}{B话榜|废话榜} {时间类型?} {ISO8601 格式时间 ?} -g {群号} -k {关键词}`
|
||||||
|
|
||||||
|
以下调用方法均合法:
|
||||||
|
|
||||||
|
`/今日B话榜 -g 12345678 -k 女装`
|
||||||
|
`/昨日B话榜 -k 女装`
|
||||||
|
`/本周B话榜 -g 12345678`
|
||||||
|
|
||||||
|
#### 看看B话
|
||||||
|
|
||||||
|
`/看看B话 {@|QQ号} {群号?} {关键词?}`
|
||||||
|
|
||||||
|
以下调用方法均合法:
|
||||||
|
|
||||||
|
`/kkb 114514 1919810 ♂`
|
||||||
|
`/kkb @man -k ♂`
|
||||||
|
|
||||||
|
> [!IMPORTANT]
|
||||||
|
> 关键词支持正则表达式!
|
||||||
|
|
||||||
## 💪 目前支持的平台
|
## 💪 目前支持的平台
|
||||||
|
|
||||||
@@ -134,6 +168,9 @@ __!!注意!!__
|
|||||||
|
|
||||||
</details>
|
</details>
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
|
||||||
### 💕感谢
|
### 💕感谢
|
||||||
|
|
||||||
本插件的__init__.py 中的处理函数参考了词云中的方法 ~~(其实大部分都是 Ctrl+C Ctr+V)~~
|
本插件的__init__.py 中的处理函数参考了词云中的方法 ~~(其实大部分都是 Ctrl+C Ctr+V)~~
|
||||||
@@ -158,14 +195,19 @@ __!!注意!!__
|
|||||||
|
|
||||||
- [x] 私聊的查询(超级用户可以任意查询群聊的信息)一半完成
|
- [x] 私聊的查询(超级用户可以任意查询群聊的信息)一半完成
|
||||||
|
|
||||||
- [ ] 特殊的储存方案优化消息统计
|
- [x] 特殊的储存方案优化消息统计
|
||||||
|
|
||||||
- [ ] 查询带某关键词的消息量
|
- [x] 查询带某关键词的消息量
|
||||||
|
|
||||||
|
- [x] 合并转发
|
||||||
|
|
||||||
待补充。.....
|
待补充。.....
|
||||||
|
|
||||||
## 📖版本
|
## 📖版本日志
|
||||||
|
|
||||||
|
<details>
|
||||||
|
<summary>点我展开</summary>
|
||||||
|
|
||||||
### V1.0
|
### V1.0
|
||||||
|
|
||||||
- 看看群里群友能有多话痨
|
- 看看群里群友能有多话痨
|
||||||
@@ -199,5 +241,11 @@ __!!注意!!__
|
|||||||
|
|
||||||
- 优化代码,添加一些新的可配置项。
|
- 优化代码,添加一些新的可配置项。
|
||||||
|
|
||||||
|
### V2.4
|
||||||
|
|
||||||
|
- 添加一些新的可配置项。
|
||||||
|
|
||||||
|
</details>
|
||||||
|
|
||||||
### 👾题外话
|
### 👾题外话
|
||||||
~~整个项目快被我写成屎山了~~
|
~~整个项目快被我写成屎山了~~
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ require("nonebot_plugin_apscheduler")
|
|||||||
require("nonebot_plugin_htmlrender")
|
require("nonebot_plugin_htmlrender")
|
||||||
require("nonebot_plugin_userinfo")
|
require("nonebot_plugin_userinfo")
|
||||||
require("nonebot_plugin_alconna")
|
require("nonebot_plugin_alconna")
|
||||||
|
require("nonebot_plugin_uninfo")
|
||||||
require("nonebot_plugin_cesaa")
|
require("nonebot_plugin_cesaa")
|
||||||
|
|
||||||
import re
|
import re
|
||||||
@@ -23,15 +24,19 @@ from nonebot.params import Arg, Depends
|
|||||||
from nonebot.adapters import Bot, Event
|
from nonebot.adapters import Bot, Event
|
||||||
from nonebot.plugin import PluginMetadata, inherit_supported_adapters
|
from nonebot.plugin import PluginMetadata, inherit_supported_adapters
|
||||||
from nonebot_plugin_alconna import (
|
from nonebot_plugin_alconna import (
|
||||||
|
At,
|
||||||
Args,
|
Args,
|
||||||
|
Field,
|
||||||
|
Match,
|
||||||
Option,
|
Option,
|
||||||
Alconna,
|
Alconna,
|
||||||
on_alconna,
|
on_alconna,
|
||||||
)
|
)
|
||||||
from nonebot_plugin_chatrecorder import get_message_records
|
from nonebot_plugin_chatrecorder import get_message_records
|
||||||
from nonebot_plugin_session import Session, SessionIdType, extract_session
|
from nonebot_plugin_session import Session, SessionIdType, extract_session
|
||||||
|
from nonebot_plugin_uninfo import Uninfo
|
||||||
|
|
||||||
from .storage import get_cache,build_cache
|
from .storage import get_cache, build_cache
|
||||||
from .config import Config, plugin_config
|
from .config import Config, plugin_config
|
||||||
from .usage import __usage__
|
from .usage import __usage__
|
||||||
from .time import (
|
from .time import (
|
||||||
@@ -74,14 +79,74 @@ def wrapper(slot: Union[int, str], content: Optional[str], context) -> str:
|
|||||||
return content
|
return content
|
||||||
return "" # pragma: no cover
|
return "" # pragma: no cover
|
||||||
|
|
||||||
|
|
||||||
build_cache_cmd = on_command("build_cache", aliases={"重建缓存"}, block=True)
|
build_cache_cmd = on_command("build_cache", aliases={"重建缓存"}, block=True)
|
||||||
|
|
||||||
|
|
||||||
@build_cache_cmd.handle()
|
@build_cache_cmd.handle()
|
||||||
async def _build_cache(bot: Bot, event: Event):
|
async def _build_cache(bot: Bot, event: Event):
|
||||||
await saa.Text("正在重建缓存,请稍等。").send(reply=True)
|
await saa.Text("正在重建缓存,请稍等。").send(reply=True)
|
||||||
await build_cache()
|
await build_cache()
|
||||||
await saa.Text("重建缓存完成。").send(reply=True)
|
await saa.Text("重建缓存完成。").send(reply=True)
|
||||||
|
|
||||||
|
|
||||||
|
b_cmd = on_alconna(
|
||||||
|
Alconna(
|
||||||
|
"看看B话",
|
||||||
|
Args["at", [str, At], Field(completion=lambda: "请想要查询的人的QQ号")],
|
||||||
|
Option("-g|--group_id", Args["group_id?", str]),
|
||||||
|
Option("-k|--keyword", Args["keyword?", str]),
|
||||||
|
),
|
||||||
|
aliases={"kkb"},
|
||||||
|
use_cmd_start=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@b_cmd.handle()
|
||||||
|
async def handle_b_cmd(
|
||||||
|
at: Match[str | At],
|
||||||
|
group_id: Match[str],
|
||||||
|
keyword: Match[str],
|
||||||
|
uninfo: Uninfo,
|
||||||
|
session: Session = Depends(extract_session),
|
||||||
|
):
|
||||||
|
id = at.result
|
||||||
|
if isinstance(id, At):
|
||||||
|
id = id.target
|
||||||
|
if group_id.available:
|
||||||
|
gid = group_id.result
|
||||||
|
else:
|
||||||
|
gid = session.id2
|
||||||
|
|
||||||
|
if not gid:
|
||||||
|
await b_cmd.finish("请指定群号。")
|
||||||
|
|
||||||
|
if keyword.available:
|
||||||
|
keywords = keyword.result
|
||||||
|
else:
|
||||||
|
keywords = None
|
||||||
|
|
||||||
|
messages = await get_message_records(
|
||||||
|
id1s=[id],
|
||||||
|
id2s=[gid],
|
||||||
|
id_type=SessionIdType.GROUP,
|
||||||
|
include_bot_id=False,
|
||||||
|
include_bot_type=False,
|
||||||
|
types=["message"], # 排除机器人自己发的消息
|
||||||
|
exclude_id1s=plugin_config.excluded_people,
|
||||||
|
)
|
||||||
|
d = msg_counter(messages, keywords)
|
||||||
|
rank = got_rank(d)
|
||||||
|
if not rank:
|
||||||
|
await b_cmd.finish(
|
||||||
|
f"该用户在群“{uninfo.scene.name}”关于“{keyword}”的B话数量为0。"
|
||||||
|
)
|
||||||
|
|
||||||
|
await saa.Text(
|
||||||
|
f"该用户在群“{uninfo.scene.name}”关于“{keyword}”的B话数量为{rank[0][1]}。"
|
||||||
|
).send(reply=True)
|
||||||
|
|
||||||
|
|
||||||
rank_cmd = on_alconna(
|
rank_cmd = on_alconna(
|
||||||
Alconna(
|
Alconna(
|
||||||
"B话榜",
|
"B话榜",
|
||||||
@@ -90,6 +155,7 @@ rank_cmd = on_alconna(
|
|||||||
str,
|
str,
|
||||||
],
|
],
|
||||||
Option("-g|--group_id", Args["group_id?", str]),
|
Option("-g|--group_id", Args["group_id?", str]),
|
||||||
|
Option("-k|--keyword", Args["keyword?", str]),
|
||||||
behaviors=[SameTime()],
|
behaviors=[SameTime()],
|
||||||
),
|
),
|
||||||
aliases={"废话榜"},
|
aliases={"废话榜"},
|
||||||
@@ -126,6 +192,7 @@ async def _group_message(
|
|||||||
type: Optional[str] = None,
|
type: Optional[str] = None,
|
||||||
time: Optional[str] = None,
|
time: Optional[str] = None,
|
||||||
group_id: Optional[str] = None,
|
group_id: Optional[str] = None,
|
||||||
|
keyword: Optional[str] = None,
|
||||||
):
|
):
|
||||||
t1 = t.time()
|
t1 = t.time()
|
||||||
state["t1"] = t1
|
state["t1"] = t1
|
||||||
@@ -137,6 +204,8 @@ async def _group_message(
|
|||||||
if group_id:
|
if group_id:
|
||||||
state["group_id"] = group_id
|
state["group_id"] = group_id
|
||||||
|
|
||||||
|
state["keyword"] = keyword
|
||||||
|
|
||||||
if not type:
|
if not type:
|
||||||
await rank_cmd.finish(__plugin_meta__.usage)
|
await rank_cmd.finish(__plugin_meta__.usage)
|
||||||
|
|
||||||
@@ -223,7 +292,11 @@ async def handle_rank(
|
|||||||
if not id:
|
if not id:
|
||||||
await saa.Text("没有指定群哦").finish()
|
await saa.Text("没有指定群哦").finish()
|
||||||
|
|
||||||
|
keyword = state["keyword"]
|
||||||
|
|
||||||
if plugin_config.counting_cache:
|
if plugin_config.counting_cache:
|
||||||
|
if keyword:
|
||||||
|
await saa.Text("已开启缓存~缓存不支持关键词查询哦").finish()
|
||||||
t1 = t.time()
|
t1 = t.time()
|
||||||
raw_rank = await get_cache(start, stop, id)
|
raw_rank = await get_cache(start, stop, id)
|
||||||
logger.debug(f"获取计数消息花费时间:{t.time() - t1}")
|
logger.debug(f"获取计数消息花费时间:{t.time() - t1}")
|
||||||
@@ -239,11 +312,13 @@ async def handle_rank(
|
|||||||
time_stop=stop,
|
time_stop=stop,
|
||||||
exclude_id1s=plugin_config.excluded_people,
|
exclude_id1s=plugin_config.excluded_people,
|
||||||
)
|
)
|
||||||
raw_rank = msg_counter(messages)
|
raw_rank = msg_counter(messages, keyword)
|
||||||
logger.debug(f"获取计数消息花费时间:{t.time() - t1}")
|
logger.debug(f"获取计数消息花费时间:{t.time() - t1}")
|
||||||
|
|
||||||
if not raw_rank:
|
if not raw_rank:
|
||||||
await saa.Text("明明这个时间段都没有人说话怎么会有话痨榜呢?").finish()
|
await saa.Text(
|
||||||
|
"没有获取到排行榜数据哦,请确认时间范围和群号是否正确或者关键词是否存在~"
|
||||||
|
).finish()
|
||||||
|
|
||||||
rank = got_rank(raw_rank)
|
rank = got_rank(raw_rank)
|
||||||
ids = await persist_id2user_id([int(i[0]) for i in rank])
|
ids = await persist_id2user_id([int(i[0]) for i in rank])
|
||||||
@@ -256,28 +331,39 @@ async def handle_rank(
|
|||||||
logger.debug(f"获取用户信息花费时间:{t.time() - t1}")
|
logger.debug(f"获取用户信息花费时间:{t.time() - t1}")
|
||||||
|
|
||||||
string: str = ""
|
string: str = ""
|
||||||
for i in rank2:
|
if plugin_config.show_text_rank:
|
||||||
logger.debug(i.user_name)
|
|
||||||
for i in range(len(rank2)):
|
if keyword:
|
||||||
str_example = plugin_config.string_format.format(
|
string += f"关于{keyword}的话痨榜结果:\n"
|
||||||
index=rank2[i].user_index,
|
else:
|
||||||
nickname=rank2[i].user_nickname,
|
string += "话痨榜:\n"
|
||||||
chatdatanum=rank2[i].user_bnum,
|
|
||||||
)
|
for i in rank2:
|
||||||
string += str_example
|
logger.debug(i.user_name)
|
||||||
|
for i in range(len(rank2)):
|
||||||
|
str_example = plugin_config.string_format.format(
|
||||||
|
index=rank2[i].user_index,
|
||||||
|
nickname=rank2[i].user_nickname,
|
||||||
|
chatdatanum=rank2[i].user_bnum,
|
||||||
|
)
|
||||||
|
string += str_example
|
||||||
|
|
||||||
msg = saa.Text(string)
|
msg = saa.Text(string)
|
||||||
t1 = t.time()
|
|
||||||
|
|
||||||
if plugin_config.visualization:
|
if plugin_config.visualization:
|
||||||
|
t1 = t.time()
|
||||||
image = await get_rank_image(rank2)
|
image = await get_rank_image(rank2)
|
||||||
msg += saa.Image(image)
|
msg += saa.Image(image)
|
||||||
|
logger.debug(f"群聊消息渲染图片花费时间:{t.time() - t1}")
|
||||||
|
|
||||||
if plugin_config.suffix:
|
if plugin_config.suffix:
|
||||||
timecost = t.time() - state["t1"]
|
timecost = t.time() - state["t1"]
|
||||||
suffix = saa.Text(plugin_config.string_suffix.format(timecost=timecost))
|
suffix = saa.Text(plugin_config.string_suffix.format(timecost=timecost))
|
||||||
msg += suffix
|
msg += suffix
|
||||||
|
if not msg:
|
||||||
|
await saa.Text("你把可视化都关了哪来的排行榜?").finish()
|
||||||
|
|
||||||
logger.debug(f"群聊消息渲染图片花费时间:{t.time() - t1}")
|
if plugin_config.aggregate_transmission:
|
||||||
|
await saa.AggregatedMessageFactory([msg]).finish()
|
||||||
await msg.finish(reply=True)
|
else:
|
||||||
|
await msg.finish(reply=True)
|
||||||
|
|||||||
@@ -9,8 +9,11 @@ class ScopedConfig(BaseModel):
|
|||||||
suffix: bool = True # 是否显示后缀
|
suffix: bool = True # 是否显示后缀
|
||||||
excluded_self: bool = True # 是否排除自己
|
excluded_self: bool = True # 是否排除自己
|
||||||
visualization: bool = True # 是否可视化
|
visualization: bool = True # 是否可视化
|
||||||
|
show_text_rank: bool = True # 是否显示文本排名
|
||||||
counting_cache: bool = False # 计数缓存(能够提高回复速度)
|
counting_cache: bool = False # 计数缓存(能够提高回复速度)
|
||||||
excluded_people: List[str] = [] # 排除的人的QQ号
|
excluded_people: List[str] = [] # 排除的人的QQ号
|
||||||
|
use_user_info_cache: bool = False # 是否使用用户信息缓存
|
||||||
|
aggregate_transmission: bool = False # 是否聚合转发消息
|
||||||
timezone: Optional[str] = "Asia/Shanghai" # 时区,影响统计时间
|
timezone: Optional[str] = "Asia/Shanghai" # 时区,影响统计时间
|
||||||
string_suffix: str = "统计花费时间:{timecost}秒" # 消息格式后缀
|
string_suffix: str = "统计花费时间:{timecost}秒" # 消息格式后缀
|
||||||
template_path: str = "./template/rank_template.j2" # 模板路径
|
template_path: str = "./template/rank_template.j2" # 模板路径
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ from sqlalchemy import delete, or_, select
|
|||||||
from nonebot import get_driver
|
from nonebot import get_driver
|
||||||
from nonebot.log import logger
|
from nonebot.log import logger
|
||||||
from nonebot.params import Depends
|
from nonebot.params import Depends
|
||||||
from nonebot.adapters import Event,Bot
|
from nonebot.adapters import Event, Bot
|
||||||
from nonebot.message import event_postprocessor
|
from nonebot.message import event_postprocessor
|
||||||
|
|
||||||
from .model import MessageCountCache
|
from .model import MessageCountCache
|
||||||
@@ -56,7 +56,9 @@ async def build_cache():
|
|||||||
where.append(
|
where.append(
|
||||||
or_(
|
or_(
|
||||||
MessageCountCache.time
|
MessageCountCache.time
|
||||||
== remove_timezone(msg.time.replace(hour=1, minute=0, second=0, microsecond=0))
|
== remove_timezone(
|
||||||
|
msg.time.replace(hour=1, minute=0, second=0, microsecond=0)
|
||||||
|
)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
statement = select(MessageCountCache).where(*where)
|
statement = select(MessageCountCache).where(*where)
|
||||||
@@ -68,7 +70,9 @@ async def build_cache():
|
|||||||
else:
|
else:
|
||||||
user_cache = MessageCountCache(
|
user_cache = MessageCountCache(
|
||||||
session_id=msg.session_persist_id,
|
session_id=msg.session_persist_id,
|
||||||
time=remove_timezone(msg.time.replace(hour=1, minute=0, second=0, microsecond=0)),
|
time=remove_timezone(
|
||||||
|
msg.time.replace(hour=1, minute=0, second=0, microsecond=0)
|
||||||
|
),
|
||||||
session_bnum=1,
|
session_bnum=1,
|
||||||
)
|
)
|
||||||
db_session.add(user_cache)
|
db_session.add(user_cache)
|
||||||
@@ -100,7 +104,7 @@ async def _():
|
|||||||
|
|
||||||
|
|
||||||
@event_postprocessor
|
@event_postprocessor
|
||||||
async def _(bot: Bot, event: Event,session: Session = Depends(extract_session)):
|
async def _(bot: Bot, event: Event, session: Session = Depends(extract_session)):
|
||||||
if not plugin_config.counting_cache:
|
if not plugin_config.counting_cache:
|
||||||
return
|
return
|
||||||
if not session.id2:
|
if not session.id2:
|
||||||
@@ -112,7 +116,7 @@ async def _(bot: Bot, event: Event,session: Session = Depends(extract_session)):
|
|||||||
|
|
||||||
async with get_session() as db_session:
|
async with get_session() as db_session:
|
||||||
session_id = await get_session_persist_id(session)
|
session_id = await get_session_persist_id(session)
|
||||||
logger.debug("session_id:"+str(session_id))
|
logger.debug("session_id:" + str(session_id))
|
||||||
where = [or_(MessageCountCache.session_id == session_id)]
|
where = [or_(MessageCountCache.session_id == session_id)]
|
||||||
where.append(or_(MessageCountCache.time == remove_timezone(now)))
|
where.append(or_(MessageCountCache.time == remove_timezone(now)))
|
||||||
statement = select(MessageCountCache).where(*where)
|
statement = select(MessageCountCache).where(*where)
|
||||||
|
|||||||
@@ -2,8 +2,59 @@ from inspect import cleandoc
|
|||||||
|
|
||||||
__usage__ = cleandoc(
|
__usage__ = cleandoc(
|
||||||
"""
|
"""
|
||||||
/今日B话榜 查看今天群里谁B话最多。
|
快速调用:
|
||||||
/昨日B话榜 顾名思义。
|
/今日B话榜 ————看看今天群友发了多少消息。
|
||||||
以此类推,有本周,本月,上周,上月,年度排行榜。
|
|
||||||
|
### 🎨一般用法
|
||||||
|
|
||||||
|
#### B话榜
|
||||||
|
|
||||||
|
-`/B话榜` ————看看有史以来(机器人存在以来)群友们发了多少消息! (好像没写)
|
||||||
|
|
||||||
|
-`/今日B话榜` ————看看今天的群友发了多少消息!
|
||||||
|
|
||||||
|
-`/昨日B话榜` ————看看昨天的群友发了多少消息!
|
||||||
|
|
||||||
|
-`/前日B话榜` ————看看前天的群友发了多少消息!
|
||||||
|
|
||||||
|
-`/本周B话榜` ————看看本周的群友发了多少消息!
|
||||||
|
|
||||||
|
-`/上周B话榜` ————看看上周的群友发了多少消息!
|
||||||
|
|
||||||
|
-`/本月B话榜` ————看看这个月的群友发了多少消息!
|
||||||
|
|
||||||
|
-`/年度B话榜` ————看看今年的群友发了多少消息!
|
||||||
|
|
||||||
|
-`/历史B话榜` ————看看历史上(机器人存在以来)的群友发了多少消息!
|
||||||
|
|
||||||
|
#### 看看B话(kkb)
|
||||||
|
|
||||||
|
-`/看看B话 [@某人|QQ号]` ————看看这个b人在这个b群发了多少b话!
|
||||||
|
|
||||||
|
### 🚀进阶用法
|
||||||
|
|
||||||
|
#### B话榜
|
||||||
|
|
||||||
|
`/{时间类型(今日|年度)?}{B话榜|废话榜} {时间类型?} {ISO8601 格式时间 ?} {群号} {关键词}`
|
||||||
|
|
||||||
|
如:`/B话榜 历史 2024-01-01~2024-01-02 12345678 女装`
|
||||||
|
|
||||||
|
也可以 `/{时间类型(今日|年度)?}{B话榜|废话榜} {时间类型?} {ISO8601 格式时间 ?} -g {群号} -k {关键词}`
|
||||||
|
|
||||||
|
以下调用方法均合法:
|
||||||
|
|
||||||
|
`/今日B话榜 -g 12345678 -k 女装`
|
||||||
|
`/昨日B话榜 -k 女装`
|
||||||
|
`/本周B话榜 -g 12345678`
|
||||||
|
|
||||||
|
#### 看看B话
|
||||||
|
|
||||||
|
`/看看B话 {@|QQ号} {群号?} {关键词?}`
|
||||||
|
|
||||||
|
以下调用方法均合法:
|
||||||
|
|
||||||
|
`/kkb 114514 1919810 ♂`
|
||||||
|
`/kkb @man -k ♂`
|
||||||
|
|
||||||
"""
|
"""
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,9 +1,10 @@
|
|||||||
import os
|
import os
|
||||||
|
import re
|
||||||
import httpx
|
import httpx
|
||||||
import asyncio
|
import asyncio
|
||||||
import unicodedata
|
import unicodedata
|
||||||
|
|
||||||
from typing import Dict, List
|
from typing import Dict, List, Optional
|
||||||
from sqlalchemy import or_, select
|
from sqlalchemy import or_, select
|
||||||
from sqlalchemy.sql import ColumnElement
|
from sqlalchemy.sql import ColumnElement
|
||||||
|
|
||||||
@@ -77,7 +78,9 @@ async def persist_id2group_id(ids: List[str]) -> List[str]:
|
|||||||
return [i.id2 for i in records]
|
return [i.id2 for i in records]
|
||||||
|
|
||||||
|
|
||||||
def msg_counter(msg_list: List[MessageRecord]) -> Dict[str, int]:
|
def msg_counter(
|
||||||
|
msg_list: List[MessageRecord], keyword: Optional[str]
|
||||||
|
) -> Dict[str, int]:
|
||||||
"""### 计算每个人的消息量
|
"""### 计算每个人的消息量
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
@@ -92,6 +95,11 @@ def msg_counter(msg_list: List[MessageRecord]) -> Dict[str, int]:
|
|||||||
logger.info("wow , there are {} msgs to count !!!".format(msg_len))
|
logger.info("wow , there are {} msgs to count !!!".format(msg_len))
|
||||||
|
|
||||||
for i in msg_list:
|
for i in msg_list:
|
||||||
|
# logger.debug(f"processing msg {i.plain_text}")
|
||||||
|
if keyword:
|
||||||
|
match = re.search(keyword, i.plain_text)
|
||||||
|
if not match:
|
||||||
|
continue
|
||||||
try:
|
try:
|
||||||
lst[str(i.session_persist_id)] += 1
|
lst[str(i.session_persist_id)] += 1
|
||||||
except KeyError:
|
except KeyError:
|
||||||
@@ -194,6 +202,7 @@ async def _get_user_default_avatar() -> bytes:
|
|||||||
).read()
|
).read()
|
||||||
return img
|
return img
|
||||||
|
|
||||||
|
|
||||||
async def _get_user_avatar(user: UserInfo, client: httpx.AsyncClient) -> bytes:
|
async def _get_user_avatar(user: UserInfo, client: httpx.AsyncClient) -> bytes:
|
||||||
if not user.user_avatar:
|
if not user.user_avatar:
|
||||||
return await _get_user_default_avatar()
|
return await _get_user_default_avatar()
|
||||||
@@ -208,6 +217,7 @@ async def _get_user_avatar(user: UserInfo, client: httpx.AsyncClient) -> bytes:
|
|||||||
await asyncio.sleep(3)
|
await asyncio.sleep(3)
|
||||||
raise NetworkError(f"{url} 下载失败!")
|
raise NetworkError(f"{url} 下载失败!")
|
||||||
|
|
||||||
|
|
||||||
def get_default_user_info() -> UserInfo:
|
def get_default_user_info() -> UserInfo:
|
||||||
user_info = UserInfo(
|
user_info = UserInfo(
|
||||||
user_id="114514",
|
user_id="114514",
|
||||||
@@ -215,14 +225,18 @@ def get_default_user_info() -> UserInfo:
|
|||||||
)
|
)
|
||||||
return user_info
|
return user_info
|
||||||
|
|
||||||
|
|
||||||
async def get_user_infos(
|
async def get_user_infos(
|
||||||
bot: Bot, event: Event, rank: List, use_cache: bool = True
|
bot: Bot,
|
||||||
|
event: Event,
|
||||||
|
rank: List,
|
||||||
|
use_cache: bool = plugin_config.use_user_info_cache,
|
||||||
) -> List[UserRankInfo]:
|
) -> List[UserRankInfo]:
|
||||||
|
|
||||||
user_ids = [i[0] for i in rank]
|
user_ids = [i[0] for i in rank]
|
||||||
pool = [get_user_info(bot, event, id, use_cache) for id in user_ids]
|
pool = [get_user_info(bot, event, id, use_cache) for id in user_ids]
|
||||||
user_infos = await asyncio.gather(*pool)
|
user_infos = await asyncio.gather(*pool)
|
||||||
|
|
||||||
async with httpx.AsyncClient() as client:
|
async with httpx.AsyncClient() as client:
|
||||||
pool = []
|
pool = []
|
||||||
for i in user_infos:
|
for i in user_infos:
|
||||||
@@ -253,11 +267,13 @@ async def get_user_infos(
|
|||||||
user_nickname=_get_user_nickname(user_info),
|
user_nickname=_get_user_nickname(user_info),
|
||||||
user_avatar_bytes=user_avatars[i],
|
user_avatar_bytes=user_avatars[i],
|
||||||
)
|
)
|
||||||
user.user_gender = (
|
print(user.user_gender)
|
||||||
"♂"
|
if user.user_gender == "male":
|
||||||
if user.user_gender == "male"
|
user.user_gender = "♂"
|
||||||
else "♀" if user.user_gender == "female" else ""
|
elif user.user_gender == "female":
|
||||||
)
|
user.user_gender = "♀"
|
||||||
|
else:
|
||||||
|
user.user_gender = "🤔"
|
||||||
rank2.append(user)
|
rank2.append(user)
|
||||||
|
|
||||||
return rank2
|
return rank2
|
||||||
|
|||||||
10
pdm.lock
generated
10
pdm.lock
generated
@@ -5,10 +5,10 @@
|
|||||||
groups = ["default", "Test", "dev"]
|
groups = ["default", "Test", "dev"]
|
||||||
strategy = ["inherit_metadata"]
|
strategy = ["inherit_metadata"]
|
||||||
lock_version = "4.5.0"
|
lock_version = "4.5.0"
|
||||||
content_hash = "sha256:f31fbbd00506680ed623e0216a48314dcbb142cebec25ebe15f1dc9a50915b25"
|
content_hash = "sha256:41c1ac2ec7d9ce3eb477f442e6ccbeda65fb25eb4f22cd696d361dae97cb065c"
|
||||||
|
|
||||||
[[metadata.targets]]
|
[[metadata.targets]]
|
||||||
requires_python = "~=3.9"
|
requires_python = ">=3.9,<3.13"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "aiofiles"
|
name = "aiofiles"
|
||||||
@@ -267,7 +267,7 @@ name = "charset-normalizer"
|
|||||||
version = "3.3.2"
|
version = "3.3.2"
|
||||||
requires_python = ">=3.7.0"
|
requires_python = ">=3.7.0"
|
||||||
summary = "The Real First Universal Charset Detector. Open, modern and actively maintained alternative to Chardet."
|
summary = "The Real First Universal Charset Detector. Open, modern and actively maintained alternative to Chardet."
|
||||||
groups = ["default", "dev"]
|
groups = ["dev"]
|
||||||
files = [
|
files = [
|
||||||
{file = "charset-normalizer-3.3.2.tar.gz", hash = "sha256:f30c3cb33b24454a82faecaf01b19c18562b1e89558fb6c56de4d9118a032fd5"},
|
{file = "charset-normalizer-3.3.2.tar.gz", hash = "sha256:f30c3cb33b24454a82faecaf01b19c18562b1e89558fb6c56de4d9118a032fd5"},
|
||||||
{file = "charset_normalizer-3.3.2-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:25baf083bf6f6b341f4121c2f3c548875ee6f5339300e08be3f2b2ba1721cdd3"},
|
{file = "charset_normalizer-3.3.2-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:25baf083bf6f6b341f4121c2f3c548875ee6f5339300e08be3f2b2ba1721cdd3"},
|
||||||
@@ -1876,7 +1876,7 @@ name = "requests"
|
|||||||
version = "2.32.3"
|
version = "2.32.3"
|
||||||
requires_python = ">=3.8"
|
requires_python = ">=3.8"
|
||||||
summary = "Python HTTP for Humans."
|
summary = "Python HTTP for Humans."
|
||||||
groups = ["default", "dev"]
|
groups = ["dev"]
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"certifi>=2017.4.17",
|
"certifi>=2017.4.17",
|
||||||
"charset-normalizer<4,>=2",
|
"charset-normalizer<4,>=2",
|
||||||
@@ -2383,7 +2383,7 @@ name = "urllib3"
|
|||||||
version = "2.2.1"
|
version = "2.2.1"
|
||||||
requires_python = ">=3.8"
|
requires_python = ">=3.8"
|
||||||
summary = "HTTP library with thread-safe connection pooling, file post, and more."
|
summary = "HTTP library with thread-safe connection pooling, file post, and more."
|
||||||
groups = ["default", "dev"]
|
groups = ["dev"]
|
||||||
files = [
|
files = [
|
||||||
{file = "urllib3-2.2.1-py3-none-any.whl", hash = "sha256:450b20ec296a467077128bff42b73080516e71b56ff59a60a02bef2232c4fa9d"},
|
{file = "urllib3-2.2.1-py3-none-any.whl", hash = "sha256:450b20ec296a467077128bff42b73080516e71b56ff59a60a02bef2232c4fa9d"},
|
||||||
{file = "urllib3-2.2.1.tar.gz", hash = "sha256:d0570876c61ab9e520d776c38acbbb5b05a776d3f9ff98a5c8fd5162a444cf19"},
|
{file = "urllib3-2.2.1.tar.gz", hash = "sha256:d0570876c61ab9e520d776c38acbbb5b05a776d3f9ff98a5c8fd5162a444cf19"},
|
||||||
|
|||||||
@@ -1,13 +1,12 @@
|
|||||||
[project]
|
[project]
|
||||||
name = "nonebot-plugin-dialectlist"
|
name = "nonebot-plugin-dialectlist"
|
||||||
version = "2.4.1"
|
version = "2.6.1"
|
||||||
description = "看看你群群友有多能说"
|
description = "看看你群群友有多能说"
|
||||||
authors = [
|
authors = [
|
||||||
{name = "Chen_Xu233", email = "woyerpa@outlook.com"},
|
{name = "Chen_Xu233", email = "woyerpa@outlook.com"},
|
||||||
]
|
]
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"nonebot-plugin-chatrecorder>=0.6.0",
|
"nonebot-plugin-chatrecorder>=0.6.0",
|
||||||
"requests>=2.32.3",
|
|
||||||
"nonebot-plugin-orm[default]",
|
"nonebot-plugin-orm[default]",
|
||||||
"nonebot-plugin-apscheduler>=0.4.0",
|
"nonebot-plugin-apscheduler>=0.4.0",
|
||||||
"nonebot-plugin-alconna>=0.50.2",
|
"nonebot-plugin-alconna>=0.50.2",
|
||||||
|
|||||||
112
tests/render.py
112
tests/render.py
@@ -1,112 +0,0 @@
|
|||||||
import PIL
|
|
||||||
import abc
|
|
||||||
from typing import List, Tuple, Union, Iterable
|
|
||||||
from PIL import Image, ImageDraw, ImageFont
|
|
||||||
|
|
||||||
class BaseElement(abc.ABC):
|
|
||||||
def __init__(self, box:Union[Iterable[int],Iterable[float]] = (0, 0, 0, 0)):
|
|
||||||
self.box = [float(i) for i in box]
|
|
||||||
self.ux = None
|
|
||||||
self.ly = None
|
|
||||||
self.dx = None
|
|
||||||
self.ry = None
|
|
||||||
self.hight = None
|
|
||||||
self.width = None
|
|
||||||
self.image = None
|
|
||||||
self.position = (self.ux, self.ly)
|
|
||||||
self.size = (self.width, self.hight)
|
|
||||||
|
|
||||||
@abc.abstractmethod
|
|
||||||
def render(self):
|
|
||||||
raise NotImplementedError
|
|
||||||
|
|
||||||
def get_box(self):
|
|
||||||
return self.box
|
|
||||||
|
|
||||||
def set_box(self, box:Tuple[float, float, float, float]):
|
|
||||||
self.box = box
|
|
||||||
|
|
||||||
def get_size(self):
|
|
||||||
self.size = (self.width, self.hight)
|
|
||||||
return self.size
|
|
||||||
|
|
||||||
def get_position(self):
|
|
||||||
self.position = (self.ux, self.ly)
|
|
||||||
return self.position
|
|
||||||
|
|
||||||
class Board(BaseElement):
|
|
||||||
def __init__(self, width, hight):
|
|
||||||
self.width = width
|
|
||||||
self.hight = hight
|
|
||||||
self.image = Image.new('RGBA', (self.width, self.hight), (255, 255, 255, 0))
|
|
||||||
self.elements = []
|
|
||||||
|
|
||||||
def render(self):
|
|
||||||
for i in self.elements:
|
|
||||||
i.ux = self.width * i.box[0]
|
|
||||||
i.ly = self.hight * i.box[1]
|
|
||||||
i.dx = self.width * i.box[2]
|
|
||||||
i.ry = self.hight * i.box[3]
|
|
||||||
i.hight = i.dx - i.ux
|
|
||||||
i.width = i.ry - i.ly
|
|
||||||
i.position = (i.ux, i.ly)
|
|
||||||
image = i.render()
|
|
||||||
self.image.paste(image, i.get_position())
|
|
||||||
|
|
||||||
def add_element(self, element:BaseElement):
|
|
||||||
self.elements.append(element)
|
|
||||||
|
|
||||||
|
|
||||||
class Container(BaseElement):
|
|
||||||
def __init__(self, box:Union[Iterable[int],Iterable[float]] = (0, 0, 0, 0)):
|
|
||||||
super().__init__(box)
|
|
||||||
self.elements:List[BaseElement] = []
|
|
||||||
|
|
||||||
def render(self,size:Tuple[int, int] = (100, 100)):
|
|
||||||
if not self.width and self.hight:
|
|
||||||
self.width = size[0]
|
|
||||||
self.hight = size[1]
|
|
||||||
if not self.image:
|
|
||||||
self.image = Image.new('RGBA', (self.width, self.hight), (255, 255, 255, 0))
|
|
||||||
for i in self.elements:
|
|
||||||
image = i.render()
|
|
||||||
self.image.paste(image, i.get_position())
|
|
||||||
|
|
||||||
class Element(BaseElement):
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# class BaseContainer(abc.ABC):
|
|
||||||
# def __init__(self, width:int, height:int):
|
|
||||||
# self.width = width
|
|
||||||
# self.height = height
|
|
||||||
# self.image = Image.new('RGBA', (self.width, self.height), (255, 255, 255, 0))
|
|
||||||
|
|
||||||
# @abc.abstractmethod
|
|
||||||
# def render(self):
|
|
||||||
# raise NotImplementedError
|
|
||||||
|
|
||||||
# def set_image(self,image):
|
|
||||||
# self.image = image
|
|
||||||
|
|
||||||
# class Elements():
|
|
||||||
# def __init__(self, width:int, height:int):
|
|
||||||
# self.width = width
|
|
||||||
# self.height = height
|
|
||||||
# self.image = Image.new('RGBA', (self.width, self.height), (255, 255, 255, 0))
|
|
||||||
|
|
||||||
# def render(self):
|
|
||||||
# return self.image
|
|
||||||
|
|
||||||
# class Container(BaseContainer):
|
|
||||||
# def __init__(self, width:int, height:int,elements:List[Union['Container',Elements]] = []):
|
|
||||||
# super().__init__(width, height)
|
|
||||||
# self.elements = elements
|
|
||||||
|
|
||||||
# def render(self):
|
|
||||||
# for i in self.elements:
|
|
||||||
# image = i.render()
|
|
||||||
# self.image.paste(image, i.position)
|
|
||||||
# return self.image
|
|
||||||
|
|
||||||
Reference in New Issue
Block a user