update

plugins/sign_in/__init__.py

@@ -1,10 +1,16 @@
-from .group_user_checkin import group_user_check_in, group_user_check, group_impression_rank
+from .group_user_checkin import group_user_check_in, group_user_check, group_impression_rank, impression_rank
 from nonebot.typing import T_State
 from nonebot.adapters.cqhttp import Bot, GroupMessageEvent
 from nonebot.adapters.cqhttp.permission import GROUP
-from util.utils import get_message_text
-from nonebot.plugin import MatcherGroup
-
+from utils.init_result import image
+from nonebot import on_command
+from utils.utils import get_message_text
+from pathlib import Path
+from configs.path_config import DATA_PATH
+try:
+    import ujson as json
+except ModuleNotFoundError:
+    import json
 
 __plugin_name__ = '签到'
 __plugin_usage__ = (
@@ -13,13 +19,23 @@ __plugin_usage__ = (
     '“我的签到” 来获取历史签到信息\n'
     '“好感度排行” 来查看当前好感度前十的伙伴\n'
     '/ 签到时有 3% 概率 * 2 /'
-
 )
 
+_file = Path(f'{DATA_PATH}/not_show_sign_rank_user.json')
+try:
+    data = json.load(open(_file, 'r', encoding='utf8'))
+except (FileNotFoundError, ValueError, TypeError):
+    data = {
+        '0': []
+    }
 
-sign_match_group = MatcherGroup(priority=5, permission=GROUP, block=True)
 
-sign = sign_match_group.on_command("签到")
+sign = on_command("签到", priority=5, permission=GROUP, block=True)
+my_sign = on_command(cmd="我的签到", aliases={'好感度'}, priority=5, permission=GROUP, block=True)
+sign_rank = on_command(cmd="积分排行", aliases={'好感度排行', '签到排行', '积分排行', '好感排行',
+                                            '好感度排名，签到排名，积分排名'},
+                       priority=5, permission=GROUP, block=True)
+total_sign_rank = on_command('签到总排行', aliases={'好感度总排行', '好感度总榜', '签到总榜'}, priority=5, block=True)
 
 
 @sign.handle()
@@ -30,9 +46,6 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
     )
 
 
-my_sign = sign_match_group.on_command(cmd="我的签到", aliases={'好感度'})
-
-
 @my_sign.handle()
 async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
     await my_sign.send(
@@ -40,12 +53,34 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
         at_sender=True,
     )
 
-sign_ranking = sign_match_group.on_command(cmd="积分排行", aliases={'好感度排行', '签到排行', '积分排行', '好感排行',
-                                                               '好感度排名，签到排名，积分排名'})
 
-
-@sign_ranking.handle()
+@sign_rank.handle()
 async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
-    await sign_ranking.send(
+    await sign_rank.send(
         await group_impression_rank(event.group_id)
-    )
+    )
+
+
+@total_sign_rank.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    msg = get_message_text(event.json())
+    if not msg:
+        await total_sign_rank.send('请稍等..正在整理数据...')
+        await total_sign_rank.send(image(b64=await impression_rank(0, data)))
+    elif msg in ['屏蔽我']:
+        if event.user_id in data['0']:
+            await total_sign_rank.finish('您已经在屏蔽名单中了，请勿重复添加！', at_sender=True)
+        data['0'].append(event.user_id)
+        await total_sign_rank.send('设置成功，您不会出现在签到总榜中！', at_sender=True)
+    elif msg in ['显示我']:
+        if event.user_id not in data['0']:
+            await total_sign_rank.finish('您不在屏蔽名单中！', at_sender=True)
+        data['0'].remove(event.user_id)
+        await total_sign_rank.send('设置成功，签到总榜将会显示您的头像名称以及好感度！', at_sender=True)
+    with open(_file, 'w', encoding='utf8') as f:
+        json.dump(data, f, ensure_ascii=False, indent=4)
+
+
+
+
+