From dbc97f8c5b1c940138da117e72b580c81fd019b2 Mon Sep 17 00:00:00 2001 From: CMHopeSunshine <277073121@qq.com> Date: Wed, 7 Sep 2022 18:12:10 +0800 Subject: [PATCH] =?UTF-8?q?:sparkles:=20`=E6=8A=BD=E5=8D=A1=E8=AE=B0?= =?UTF-8?q?=E5=BD=95`=E6=94=AF=E6=8C=81UIGF=E6=A0=87=E5=87=86json=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E7=9A=84=E5=AF=BC=E5=85=A5=E5=92=8C=E5=AF=BC=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .gitignore | 2 +- .../plugins/Paimon_Gacha_Log/__init__.py | 38 ++++++++- .../plugins/Paimon_Gacha_Log/data_source.py | 84 ++++++++++++++++++- 3 files changed, 117 insertions(+), 7 deletions(-) diff --git a/.gitignore b/.gitignore index 23ffc7d..3d14e24 100644 --- a/.gitignore +++ b/.gitignore @@ -140,7 +140,7 @@ dmypy.json # End of https://www.toptal.com/developers/gitignore/api/python - +*.out accounts data resources diff --git a/LittlePaimon/plugins/Paimon_Gacha_Log/__init__.py b/LittlePaimon/plugins/Paimon_Gacha_Log/__init__.py index 24ff6f8..52833e3 100644 --- a/LittlePaimon/plugins/Paimon_Gacha_Log/__init__.py +++ b/LittlePaimon/plugins/Paimon_Gacha_Log/__init__.py @@ -1,10 +1,11 @@ +from typing import Union from nonebot import on_command -from nonebot.adapters.onebot.v11 import MessageEvent +from nonebot.adapters.onebot.v11 import Bot, MessageEvent, GroupMessageEvent, PrivateMessageEvent from nonebot.plugin import PluginMetadata from LittlePaimon.utils import logger from LittlePaimon.utils.message import CommandPlayer -from .data_source import get_gacha_log_img, get_gacha_log_data +from .data_source import get_gacha_log_img, get_gacha_log_data, create_import_command, gacha_log_to_UIGF __plugin_meta__ = PluginMetadata( name="原神抽卡记录分析", @@ -30,6 +31,18 @@ show_log = on_command('查看抽卡记录', aliases={'抽卡记录'}, priority=1 'pm_usage': '查看抽卡记录(uid)', 'pm_priority': 2 }) +import_log = on_command('导入抽卡记录', aliases={'抽卡记录导入'}, priority=12, block=True, state={ + 'pm_name': '导入抽卡记录', + 'pm_description': '导入符合UIGF标准的抽卡记录json文件，发送命令后，于5分钟内上传json文件即可', + 'pm_usage': '导入抽卡记录', + 'pm_priority': 3 +}) +export_log = on_command('导出抽卡记录', aliases={'抽卡记录导出'}, priority=12, block=True, state={ + 'pm_name': '导出抽卡记录', + 'pm_description': '导出符合UIGF标准的抽卡记录json文件', + 'pm_usage': '导出抽卡记录', + 'pm_priority': 4 +}) running_update = [] running_show = [] @@ -66,3 +79,24 @@ async def _(event: MessageEvent, player=CommandPlayer(1)): logger.info('原神抽卡记录', f'➤➤绘制抽卡记录图片时出现错误：{e}') await update_log.send(f'绘制抽卡记录分析时出现错误：{e}') running_show.remove(f'{player[0].user_id}-{player[0].uid}') + + +@import_log.handle() +async def _(event: Union[GroupMessageEvent, PrivateMessageEvent]): + create_import_command(event.user_id) + await import_log.finish('请在5分钟内，上传或发送符合UIGF标准的抽卡记录json文件', at_sender=True) + + +@export_log.handle() +async def _(event: Union[GroupMessageEvent, PrivateMessageEvent], bot: Bot, player=CommandPlayer(1)): + state, msg, path = gacha_log_to_UIGF(player[0].user_id, player[0].uid) + if not state: + await export_log.finish(msg, at_sender=True) + else: + try: + if isinstance(event, GroupMessageEvent): + await bot.upload_group_file(group_id=event.group_id, file=str(path.absolute()), name=path.name) + else: + await bot.upload_private_file(user_id=event.user_id, file=str(path.absolute()), name=path.name) + except Exception as e: + await export_log.finish(f'上传文件失败，错误信息：{e}', at_sender=True) diff --git a/LittlePaimon/plugins/Paimon_Gacha_Log/data_source.py b/LittlePaimon/plugins/Paimon_Gacha_Log/data_source.py index 55b333b..04a08cb 100644 --- a/LittlePaimon/plugins/Paimon_Gacha_Log/data_source.py +++ b/LittlePaimon/plugins/Paimon_Gacha_Log/data_source.py @@ -1,14 +1,17 @@ import asyncio import time +from typing import Dict, Union, Tuple, Optional +from pathlib import Path import datetime -from typing import Dict, Union, Tuple -from LittlePaimon import DRIVER +from nonebot import on_notice +from nonebot.rule import Rule +from nonebot.adapters.onebot.v11 import GroupUploadNoticeEvent, NoticeEvent from LittlePaimon.database.models import PlayerInfo from LittlePaimon.config import GACHA_LOG from LittlePaimon.utils.api import get_authkey_by_stoken from LittlePaimon.utils import aiorequests, logger -from LittlePaimon.utils.files import load_json +from LittlePaimon.utils.files import load_json, save_json from .models import GachaItem, GachaLogInfo, GACHA_TYPE_LIST from .draw import draw_gacha_log @@ -70,6 +73,38 @@ def save_gacha_log_info(user_id: str, uid: str, info: GachaLogInfo): f.write(info.json(ensure_ascii=False, indent=4)) +def gacha_log_to_UIGF(user_id: str, uid: str) -> Tuple[bool, str, Optional[Path]]: + data, state = load_history_info(user_id, uid) + if not state: + return False, f'UID{uid}还没有抽卡记录数据，请先更新', None + logger.info('原神抽卡记录', '➤', {'用户': user_id, 'UID': uid}, '导出抽卡记录', True) + save_path = Path() / 'data' / 'LittlePaimon' / 'user_data' / 'gacha_log_data' / f'gacha_log_UIGF-{user_id}-{uid}.json' + uigf_dict = { + 'info': { + 'uid': uid, + 'lang': 'zh-cn', + 'export_time': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'), + 'uigf_version': 'v2.1' + }, + 'list': [] + } + for items in data.item_list.values(): + for item in items: + uigf_dict['list'].append({ + 'gacha_type': item.gacha_type, + 'item_id': '', + 'count': '1', + 'time': item.time.strftime('%Y-%m-%d %H:%M:%S'), + 'name': item.name, + 'item_type': item.item_type, + 'rank_type': item.rank_type, + 'id': item.id, + 'uigf_gacha_type': item.gacha_type + }) + save_json(uigf_dict, save_path) + return True, '导出成功', save_path + + async def get_gacha_log_data(user_id: str, uid: str): """ 使用authkey获取抽卡记录数据，并合并旧数据 @@ -130,8 +165,49 @@ async def get_gacha_log_img(user_id: str, uid: str, nickname: str): return f'UID{uid}还没有抽卡记录数据，请先更新' player_info = await PlayerInfo.get_or_none(user_id=user_id, uid=uid) if player_info: - return await draw_gacha_log(player_info.user_id, player_info.uid, player_info.nickname, player_info.signature, data) + return await draw_gacha_log(player_info.user_id, player_info.uid, player_info.nickname, player_info.signature, + data) else: return await draw_gacha_log(user_id, uid, nickname, None, data) +def create_import_command(user_id: int): + def file_rule(event: NoticeEvent): + if event.get_user_id() != str(user_id): + return False + if isinstance(event, GroupUploadNoticeEvent): + return True + return event.notice_type == 'offline_file' + + import_cmd = on_notice(priority=12, rule=Rule(file_rule), expire_time=datetime.timedelta(minutes=5), temp=True) + + @import_cmd.handle() + async def _(event: NoticeEvent): + event_data = event.dict() + file_name = event_data['file']['name'] + if not file_name.endswith('.json'): + await import_cmd.finish('文件格式错误，请上传json文件', at_sender=True) + file_url = event_data['file']['url'] + try: + data = await aiorequests.get(url=file_url) + data = data.json() + new_num = 0 + uid = data['info']['uid'] + logger.info('原神抽卡记录', '➤', {'用户': user_id, 'UID': uid}, '导入抽卡记录', True) + gacha_log, _ = load_history_info(event.get_user_id(), uid) + for item in data['list']: + pool_name = GACHA_TYPE_LIST[item['gacha_type']] + item_info = GachaItem.parse_obj(item) + if item_info not in gacha_log.item_list[pool_name]: + gacha_log.item_list[pool_name].append(item_info) + new_num += 1 + for i in gacha_log.item_list.values(): + i.sort(key=lambda x: (x.time, x.id)) + gacha_log.update_time = datetime.datetime.now() + save_gacha_log_info(event.get_user_id(), uid, gacha_log) + if new_num == 0: + await import_cmd.send(f'UID{uid}抽卡记录导入完成，本次没有新增数据', at_sender=True) + else: + await import_cmd.send(f'UID{uid}抽卡记录导入完成，共新增{new_num}条抽卡记录', at_sender=True) + except Exception as e: + await import_cmd.finish(f'导入抽卡记录时发生错误，错误信息：{e}', at_sender=True)