diff --git a/README.md b/README.md index 35d76bd..e089ad0 100644 --- a/README.md +++ b/README.md @@ -96,9 +96,17 @@ __！！注意！！__ ### 🚀进阶用法 -`/{时间类型（今日|年度）?}{B话榜|废话榜} {时间类型？} {ISO8601 格式时间} {群号}` +`/{时间类型（今日|年度）?}{B话榜|废话榜} {时间类型？} {ISO8601 格式时间 ?} {群号} {关键词}` -如：`/B话榜 历史 2024-01-01~2024-01-02 12345678` +如：`/B话榜 历史 2024-01-01~2024-01-02 12345678 女装` + +也可以 `/{时间类型（今日|年度）?}{B话榜|废话榜} {时间类型？} {ISO8601 格式时间 ?} -g {群号} -k {关键词}` + +以下调用方法均合法： + +`/今日B话榜 -g 12345678 -k 女装` +`/昨日B话榜 -k 女装` +`/本周B话榜 -g 12345678` ## 💪 目前支持的平台 @@ -158,14 +166,19 @@ __！！注意！！__ - [x] 私聊的查询（超级用户可以任意查询群聊的信息）一半完成 -- [ ] 特殊的储存方案优化消息统计 +- [x] 特殊的储存方案优化消息统计 -- [ ] 查询带某关键词的消息量 +- [x] 查询带某关键词的消息量 + +- [ ] 合并转发 待补充。..... -## 📖版本 - +## 📖版本日志 + +
+点我展开 + ### V1.0 - 看看群里群友能有多话痨 @@ -199,5 +212,11 @@ __！！注意！！__ - 优化代码，添加一些新的可配置项。 +### V2.4 + + - 添加一些新的可配置项。 + +
+ ### 👾题外话 ~~整个项目快被我写成屎山了~~ diff --git a/nonebot_plugin_dialectlist/__init__.py b/nonebot_plugin_dialectlist/__init__.py index 8af4786..58e4887 100644 --- a/nonebot_plugin_dialectlist/__init__.py +++ b/nonebot_plugin_dialectlist/__init__.py @@ -93,6 +93,7 @@ rank_cmd = on_alconna( str, ], Option("-g|--group_id", Args["group_id?", str]), + Option("-k|--keyword", Args["keyword?", str]), behaviors=[SameTime()], ), aliases={"废话榜"}, @@ -129,6 +130,7 @@ async def _group_message( type: Optional[str] = None, time: Optional[str] = None, group_id: Optional[str] = None, + keyword: Optional[str] = None, ): t1 = t.time() state["t1"] = t1 @@ -140,6 +142,8 @@ async def _group_message( if group_id: state["group_id"] = group_id + state["keyword"] = keyword + if not type: await rank_cmd.finish(__plugin_meta__.usage) @@ -226,7 +230,11 @@ async def handle_rank( if not id: await saa.Text("没有指定群哦").finish() + keyword = state["keyword"] + if plugin_config.counting_cache: + if keyword: + await saa.Text("已开启缓存~缓存不支持关键词查询哦").finish() t1 = t.time() raw_rank = await get_cache(start, stop, id) logger.debug(f"获取计数消息花费时间:{t.time() - t1}") @@ -242,11 +250,13 @@ async def handle_rank( time_stop=stop, exclude_id1s=plugin_config.excluded_people, ) - raw_rank = msg_counter(messages) + raw_rank = msg_counter(messages, keyword) logger.debug(f"获取计数消息花费时间:{t.time() - t1}") if not raw_rank: - await saa.Text("明明这个时间段都没有人说话怎么会有话痨榜呢？").finish() + await saa.Text( + "没有获取到排行榜数据哦，请确认时间范围和群号是否正确或者关键词是否存在~" + ).finish() rank = got_rank(raw_rank) ids = await persist_id2user_id([int(i[0]) for i in rank]) @@ -259,8 +269,13 @@ async def handle_rank( logger.debug(f"获取用户信息花费时间:{t.time() - t1}") string: str = "" - if plugin_config.show_text_rank: + + if keyword: + string += f"关于{keyword}的话痨榜结果：\n" + else: + string += "话痨榜：\n" + for i in rank2: logger.debug(i.user_name) for i in range(len(rank2)): diff --git a/nonebot_plugin_dialectlist/usage.py b/nonebot_plugin_dialectlist/usage.py index 63ea115..892f222 100644 --- a/nonebot_plugin_dialectlist/usage.py +++ b/nonebot_plugin_dialectlist/usage.py @@ -2,8 +2,42 @@ from inspect import cleandoc __usage__ = cleandoc( """ -/今日B话榜 查看今天群里谁B话最多。 -/昨日B话榜 顾名思义。 -以此类推，有本周，本月，上周，上月，年度排行榜。 +快速调用： +/今日B话榜 ————看看今天群友发了多少消息。 + +### 🎨一般用法 + +-`/B话榜` ————看看有史以来（机器人存在以来）群友们发了多少消息！ （好像没写） + +-`/今日B话榜` ————看看今天的群友发了多少消息！ + +-`/昨日B话榜` ————看看昨天的群友发了多少消息！ + +-`/前日B话榜` ————看看前天的群友发了多少消息！ + +-`/本周B话榜` ————看看本周的群友发了多少消息！ + +-`/上周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` + """ ) diff --git a/nonebot_plugin_dialectlist/utils.py b/nonebot_plugin_dialectlist/utils.py index 63a0e13..b85b974 100644 --- a/nonebot_plugin_dialectlist/utils.py +++ b/nonebot_plugin_dialectlist/utils.py @@ -3,7 +3,7 @@ import httpx import asyncio import unicodedata -from typing import Dict, List +from typing import Dict, List, Optional from sqlalchemy import or_, select from sqlalchemy.sql import ColumnElement @@ -77,7 +77,9 @@ async def persist_id2group_id(ids: List[str]) -> List[str]: 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: @@ -92,6 +94,10 @@ def msg_counter(msg_list: List[MessageRecord]) -> Dict[str, int]: logger.info("wow , there are {} msgs to count !!!".format(msg_len)) for i in msg_list: + logger.debug(f"processing msg {i.plain_text}") + if keyword: + if keyword not in i.plain_text: + continue try: lst[str(i.session_persist_id)] += 1 except KeyError: diff --git a/tests/render.py b/tests/render.py deleted file mode 100644 index aecbd25..0000000 --- a/tests/render.py +++ /dev/null @@ -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 - \ No newline at end of file