From 75dd80155bd5d1af45dd845aed87b248eb430d2d Mon Sep 17 00:00:00 2001 From: Chenric <91937041+ChenXu233@users.noreply.github.com> Date: Fri, 2 Aug 2024 23:36:58 +0800 Subject: [PATCH] =?UTF-8?q?:zap:=20:construction:=20=E6=8F=90=E9=80=9F?= =?UTF-8?q?=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nonebot_plugin_dialectlist/__init__.py | 28 +++++++++++++++---- nonebot_plugin_dialectlist/model.py | 8 ++++-- .../template/css/style.css | 2 +- 3 files changed, 28 insertions(+), 10 deletions(-) diff --git a/nonebot_plugin_dialectlist/__init__.py b/nonebot_plugin_dialectlist/__init__.py index 43c25b3..1555a0d 100644 --- a/nonebot_plugin_dialectlist/__init__.py +++ b/nonebot_plugin_dialectlist/__init__.py @@ -52,6 +52,7 @@ from .utils import ( msg_counter, get_rank_image, persist_id2user_id, + # get_user_info2, ) __plugin_meta__ = PluginMetadata( @@ -192,6 +193,8 @@ async def _group_message( state["stop"] = state["start"] + timedelta(days=1) except ValueError: await rank_cmd.finish("请输入正确的日期,不然我没法理解呢!") + + logger.debug(f"命令解析花费时间:{t.time() - t1}") @rank_cmd.got( @@ -213,6 +216,7 @@ async def handle_rank( start: datetime = Arg(), stop: datetime = Arg(), ): + t1 = t.time() if id := state["group_id"]: id = str(id) logger.debug(f"group_id: {id}") @@ -223,6 +227,9 @@ async def handle_rank( if not id: await saa.Text("没有指定群哦").finish() + logger.debug(f"所属群聊解析花费时间:{t.time() - t1}") + t1 = t.time() + messages = await get_message_records( id2s=[id], id_type=SessionIdType.GROUP, @@ -233,11 +240,16 @@ async def handle_rank( time_stop=stop, exclude_id1s=plugin_config.excluded_people, ) - + + logger.debug(f"获取群聊消息花费时间:{t.time() - t1}") + t1 = t.time() + if not messages: await saa.Text("明明这个时间段都没有人说话怎么会有话痨榜呢?").finish() rank = got_rank(msg_counter(messages)) + logger.debug(f"群聊消息计数花费时间:{t.time() - t1}") + t1 = t.time() logger.debug(rank) rank2: List[UserRankInfo] = [] ids = await persist_id2user_id([int(i[0]) for i in rank]) @@ -292,6 +304,8 @@ async def handle_rank( string += str_example msg = saa.Text(string) + logger.debug(f"群聊消息渲染文字花费时间:{t.time() - t1}") + t1 = t.time() if plugin_config.visualization: image = await get_rank_image(rank2) @@ -302,11 +316,13 @@ async def handle_rank( suffix = saa.Text(plugin_config.string_suffix.format(timecost=timecost)) msg += suffix + logger.debug(f"群聊消息渲染图片花费时间:{t.time() - t1}") + await msg.finish(reply=True) -@scheduler.scheduled_job( - "dialectlist", day="*/2", id="xxx", args=[1], kwargs={"arg2": 2} -) -async def __(): - pass +# @scheduler.scheduled_job( +# "dialectlist", day="*/2", id="xxx", args=[1], kwargs={"arg2": 2} +# ) +# async def __(): +# pass diff --git a/nonebot_plugin_dialectlist/model.py b/nonebot_plugin_dialectlist/model.py index 9fa9106..f3dab5c 100644 --- a/nonebot_plugin_dialectlist/model.py +++ b/nonebot_plugin_dialectlist/model.py @@ -1,4 +1,5 @@ from typing import Union +from pydantic import BaseModel from nonebot_plugin_orm import Model from nonebot_plugin_userinfo import UserInfo @@ -10,6 +11,7 @@ class UserRankInfo(UserInfo): user_index: Union[int, str] user_avatar_bytes: bytes - -class MsgData(Model): - __table_args__ = {"extend_existing": True} +class MsgCountDayData(BaseModel): + user_id:str +# class MsgCountData(Model): +# __tablename__ = 'dialectlist_msg_data' diff --git a/nonebot_plugin_dialectlist/template/css/style.css b/nonebot_plugin_dialectlist/template/css/style.css index 206c320..77b62f6 100644 --- a/nonebot_plugin_dialectlist/template/css/style.css +++ b/nonebot_plugin_dialectlist/template/css/style.css @@ -1,5 +1,5 @@ body { - background-image: url('https://image.anosu.top/pixiv/direct?size=regular'); + background-image: url('https://api.yimian.xyz/img'); background-attachment: scroll; margin: 0; background-size: cover;