From 316c2044f0eb62ac15d2dd805755d756b779d50c Mon Sep 17 00:00:00 2001 From: ShiXui Date: Sun, 11 Sep 2022 14:00:49 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=92=A1QQ=E7=BE=A4=E8=AF=9D=E7=97=A8?= =?UTF-8?q?=E6=8E=92=E8=A1=8C=E5=8F=AF=E8=A7=86=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nonebot_plugin_dialectlist/__init__.py | 6 ++- nonebot_plugin_dialectlist/config.py | 5 ++- nonebot_plugin_dialectlist/qqDBRecorder.py | 44 +++++++++++++++++++--- 3 files changed, 46 insertions(+), 9 deletions(-) diff --git a/nonebot_plugin_dialectlist/__init__.py b/nonebot_plugin_dialectlist/__init__.py index c26082e..eda6212 100644 --- a/nonebot_plugin_dialectlist/__init__.py +++ b/nonebot_plugin_dialectlist/__init__.py @@ -1,5 +1,6 @@ import re import time +import asyncio from typing import List, Tuple, Union from datetime import datetime, timedelta @@ -184,6 +185,7 @@ async def handle_message( guild_id = event.guild_id msg = await get_guild_message_records(guild_id=str(guild_id),bot=bot) - msg += plugin_config.dialectlist_string_suffix_format.format(timecost=time.time()-st) - await rankings.finish(msg) + await rankings.send(msg) + await asyncio.sleep(1) #让图片先发出来 + await rankings.finish(plugin_config.dialectlist_string_suffix_format.format(timecost=time.time()-st-1)) \ No newline at end of file diff --git a/nonebot_plugin_dialectlist/config.py b/nonebot_plugin_dialectlist/config.py index 517d37b..284b380 100644 --- a/nonebot_plugin_dialectlist/config.py +++ b/nonebot_plugin_dialectlist/config.py @@ -9,10 +9,13 @@ class Config(BaseModel, extra=Extra.ignore): timezone: Optional[str] dialectlist_string_format: str = '第{index}名：\n{nickname},{chatdatanum}条消息\n' - dialectlist_string_suffix_format: str = '\n你们的职业是水群吗？————MYX\n计算花费时间:{timecost}秒' + dialectlist_string_suffix_format: str = '你们的职业是水群吗？————MYX\n计算花费时间:{timecost}秒' dialectlist_path:str = os.path.dirname(__file__) + dialectlist_image_path: Path = Path(dialectlist_path)/'image.png' + dialectlist_imageSvg_path: Path = Path(dialectlist_path)/'image.svg' dialectlist_json_path:Path = Path(dialectlist_path)/'qqguild.json' dialectlist_get_num:int = 10 + dialectlist_visualization:bool = True global_config = get_driver().config plugin_config = Config.parse_obj(global_config) \ No newline at end of file diff --git a/nonebot_plugin_dialectlist/qqDBRecorder.py b/nonebot_plugin_dialectlist/qqDBRecorder.py index 96da136..ed3abab 100644 --- a/nonebot_plugin_dialectlist/qqDBRecorder.py +++ b/nonebot_plugin_dialectlist/qqDBRecorder.py @@ -1,13 +1,18 @@ import time +import pygal +import unicodedata from datetime import datetime from sqlmodel import select, or_ from typing_extensions import Literal from typing import Iterable, List, Optional, Dict +from pygal.style import Style +style=Style(font_family="SimHei",) + from nonebot.log import logger from nonebot.adapters import Bot -from nonebot.adapters.onebot.v11 import Message +from nonebot.adapters.onebot.v11 import Message,MessageSegment from nonebot.adapters.onebot.v11.exception import ActionFailed from nonebot_plugin_datastore import create_session @@ -16,6 +21,17 @@ from nonebot_plugin_chatrecorder.model import MessageRecord from .config import plugin_config +def remove_control_characters(string:str) -> str: + """将字符串中的控制符去除 + + Args: + string (str): 需要去除的字符串 + + Returns: + (str): 经过处理的字符串 + """ + return "".join(ch for ch in string if unicodedata.category(ch)[0]!="C") + async def get_message_records( user_ids: Optional[Iterable[str]] = None, group_ids: Optional[Iterable[str]] = None, @@ -86,7 +102,7 @@ async def msg_counter( got_num:int=10, )->Message: ''' - 计算出结果并返回可以直接发送的字符串 + 计算出结果并返回可以直接发送的字符串和图片 ''' st = time.time() @@ -111,7 +127,7 @@ async def msg_counter( try: maxkey = max(lst, key=lst.get) # type: ignore except ValueError: - ranking.append(("null",0)) + ranking.append(["null",0]) continue logger.debug('searching number {} from group {}'.format(str(maxkey),str(gid))) @@ -125,7 +141,7 @@ async def msg_counter( no_cache=True ) nickname:str = member_info['nickname']if not member_info['card'] else member_info['card'] - ranking.append([nickname.strip(),lst.pop(maxkey)]) + ranking.append([remove_control_characters(nickname).strip(),lst.pop(maxkey)]) except ActionFailed as e: @@ -135,14 +151,30 @@ async def msg_counter( logger.debug('loaded list:\n{}'.format(ranking)) + + if plugin_config.dialectlist_visualization: + + view = pygal.Pie(inner_radius=0.6,style=style) + view.title = '消息圆环图' + for i in ranking: + view.add(str(i[0]),int(i[1])) + try: + png: bytes = view.render_to_png()# type: ignore + process_msg = Message(MessageSegment.image(png)) + except OSError: + logger.error('GTK+(GIMP Toolkit) is not installed, the svg can not be transformed to png') + plugin_config.dialectlist_visualization = False + process_msg = Message('无法发送可视化图片，请检查是否安装GTK+，详细安装教程可见github\nhttps://github.com/tschoonj/GTK-for-Windows-Runtime-Environment-Installer \n若不想安装这个软件，再次使用指令会转换为发送字符串而不是发送图片') + else: + process_msg = '' out:str = '' for i in range(got_num): index = i+1 - nickname,chatdatanum = str(ranking[i]) + nickname,chatdatanum = ranking[i] str_example = plugin_config.dialectlist_string_format.format(index=index,nickname=nickname,chatdatanum=chatdatanum) out = out + str_example logger.debug(out) logger.info('spent {} seconds to count from {} msg'.format(time.time()-st,msg_len)) - return Message(out) + return Message(out)+process_msg