mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-12-04 12:40:09 +00:00
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
import ujson
|
|
from abc import abstractmethod
|
|
from pathlib import Path
|
|
from typing import Optional, Tuple, TYPE_CHECKING
|
|
|
|
from modules.gacha_log.error import GachaLogNotFound
|
|
from modules.gacha_log.models import UIGFModel, UIGFInfo, ImportType, UIGFListInfo, UIGFItem
|
|
|
|
if TYPE_CHECKING:
|
|
from modules.gacha_log.models import GachaLogInfo
|
|
|
|
|
|
class GachaLogUigfConverter:
|
|
"""抽卡记录导出为 uigf 标准"""
|
|
|
|
gacha_log_path: Path
|
|
|
|
@staticmethod
|
|
@abstractmethod
|
|
async def save_json(path, data):
|
|
"""保存json文件"""
|
|
|
|
@abstractmethod
|
|
async def load_history_info(
|
|
self, user_id: str, uid: str, only_status: bool = False
|
|
) -> Tuple[Optional["GachaLogInfo"], bool]:
|
|
"""读取历史抽卡记录数据
|
|
:param user_id: 用户id
|
|
:param uid: 原神uid
|
|
:param only_status: 是否只读取状态
|
|
:return: 抽卡记录数据
|
|
"""
|
|
|
|
async def gacha_log_to_uigf(self, user_id: str, uid: str) -> Optional[Path]:
|
|
"""抽卡记录转换为 UIGF 格式
|
|
:param user_id: 用户ID
|
|
:param uid: 游戏UID
|
|
:return: 转换是否成功、转换信息、UIGF文件目录
|
|
"""
|
|
data, state = await self.load_history_info(user_id, uid)
|
|
if not state:
|
|
raise GachaLogNotFound
|
|
save_path = self.gacha_log_path / f"{user_id}-{uid}-uigf.json"
|
|
i = UIGFInfo(export_app=ImportType.PaiGram.value, export_app_version="v4")
|
|
list_info = UIGFListInfo(uid=int(uid), list=[])
|
|
info = UIGFModel(info=i, hk4e=[], hkrpg=[], nap=[list_info])
|
|
for items in data.item_list.values():
|
|
for item in items:
|
|
list_info.list.append(
|
|
UIGFItem(
|
|
id=item.id,
|
|
name=item.name,
|
|
gacha_id=item.gacha_id,
|
|
gacha_type=item.gacha_type,
|
|
item_id=item.item_id,
|
|
item_type=item.item_type,
|
|
rank_type=item.rank_type,
|
|
time=item.time.strftime("%Y-%m-%d %H:%M:%S"),
|
|
uigf_gacha_type=item.gacha_type,
|
|
)
|
|
)
|
|
await self.save_json(save_path, ujson.loads(info.model_dump_json()))
|
|
return save_path
|