mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
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 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__ = (
|
|
'用法：\n'
|
|
'对我说 “签到” 来签到\n'
|
|
'“我的签到” 来获取历史签到信息\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 = 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()
|
|
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
|
await sign.send(
|
|
await group_user_check_in(event.user_id, event.group_id),
|
|
at_sender=True,
|
|
)
|
|
|
|
|
|
@my_sign.handle()
|
|
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
|
await my_sign.send(
|
|
await group_user_check(event.user_id, event.group_id),
|
|
at_sender=True,
|
|
)
|
|
|
|
|
|
@sign_rank.handle()
|
|
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
|
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)
|
|
|
|
|
|
|
|
|
|
|