Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-15 17:20:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor code

Browse Source
This commit is contained in:
hibiki
2021-07-30 21:21:51 +08:00
parent 2ad891aa1e
commit cc24822dca
165 changed files with 7815 additions and 8174 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/sign_in/__init__.py
+43 -38
View File
@@ -1,41 +1,53 @@
from .group_user_checkin import group_user_check_in, group_user_check, group_impression_rank, 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 utils.init_result import image
from utils.message_builder 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_name__ = "签到"
__plugin_usage__ = (
'用法：\n'
'对我说 “签到” 来签到\n'
'“我的签到” 来获取历史签到信息\n'
'“好感度排行” 来查看当前好感度前十的伙伴\n'
'/ 签到时有 3% 概率 * 2 /'
"用法：\n"
"对我说 “签到” 来签到\n"
"“我的签到” 来获取历史签到信息\n"
"“好感度排行” 来查看当前好感度前十的伙伴\n"
"/ 签到时有 3% 概率 * 2 /"
)
_file = Path(f'{DATA_PATH}/not_show_sign_rank_user.json')
_file = Path(f"{DATA_PATH}/not_show_sign_rank_user.json")
try:
data = json.load(open(_file, 'r', encoding='utf8'))
data = json.load(open(_file, "r", encoding="utf8"))
except (FileNotFoundError, ValueError, TypeError):
data = {
'0': []
}
data = {"0": []}
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)
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()
@@ -56,31 +68,24 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
@sign_rank.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
await sign_rank.send(
await group_impression_rank(event.group_id)
)
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("请稍等..正在整理数据...")
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:
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)