mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
import datetime
|
|
from typing import Dict, List
|
|
|
|
from pytz import timezone
|
|
from simnet.models.genshin.chronicle.abyss import SpiralAbyss
|
|
from simnet.models.genshin.chronicle.img_theater import ImgTheaterData
|
|
from simnet.models.genshin.diary import Diary
|
|
|
|
from core.services.history_data.models import (
|
|
HistoryData,
|
|
HistoryDataTypeEnum,
|
|
HistoryDataAbyss,
|
|
HistoryDataLedger,
|
|
HistoryDataImgTheater,
|
|
)
|
|
from gram_core.base_service import BaseService
|
|
from gram_core.services.history_data.services import HistoryDataBaseServices
|
|
|
|
try:
|
|
import ujson as jsonlib
|
|
except ImportError:
|
|
import json as jsonlib
|
|
|
|
|
|
__all__ = (
|
|
"HistoryDataBaseServices",
|
|
"HistoryDataAbyssServices",
|
|
"HistoryDataLedgerServices",
|
|
"HistoryDataImgTheaterServices",
|
|
)
|
|
|
|
TZ = timezone("Asia/Shanghai")
|
|
|
|
|
|
def json_encoder(value):
|
|
if isinstance(value, datetime.datetime):
|
|
return value.astimezone(TZ).strftime("%Y-%m-%d %H:%M:%S")
|
|
return value
|
|
|
|
|
|
class HistoryDataAbyssServices(BaseService, HistoryDataBaseServices):
|
|
DATA_TYPE = HistoryDataTypeEnum.ABYSS.value
|
|
|
|
@staticmethod
|
|
def exists_data(data: HistoryData, old_data: List[HistoryData]) -> bool:
|
|
floor = data.data.get("floors")
|
|
return any(d.data.get("floors") == floor for d in old_data)
|
|
|
|
@staticmethod
|
|
def create(user_id: int, abyss_data: SpiralAbyss, character_data: Dict[int, int]):
|
|
data = HistoryDataAbyss(abyss_data=abyss_data, character_data=character_data)
|
|
json_data = data.json(by_alias=True, encoder=json_encoder)
|
|
return HistoryData(
|
|
user_id=user_id,
|
|
data_id=abyss_data.season,
|
|
time_created=datetime.datetime.now(),
|
|
type=HistoryDataAbyssServices.DATA_TYPE,
|
|
data=jsonlib.loads(json_data),
|
|
)
|
|
|
|
|
|
class HistoryDataLedgerServices(BaseService, HistoryDataBaseServices):
|
|
DATA_TYPE = HistoryDataTypeEnum.LEDGER.value
|
|
|
|
@staticmethod
|
|
def create(user_id: int, diary_data: Diary):
|
|
data = HistoryDataLedger(diary_data=diary_data)
|
|
json_data = data.json(by_alias=True, encoder=json_encoder)
|
|
return HistoryData(
|
|
user_id=user_id,
|
|
data_id=diary_data.data_id,
|
|
time_created=datetime.datetime.now(),
|
|
type=HistoryDataLedgerServices.DATA_TYPE,
|
|
data=jsonlib.loads(json_data),
|
|
)
|
|
|
|
|
|
class HistoryDataImgTheaterServices(BaseService, HistoryDataBaseServices):
|
|
DATA_TYPE = HistoryDataTypeEnum.ROLE_COMBAT.value
|
|
|
|
@staticmethod
|
|
def exists_data(data: HistoryData, old_data: List[HistoryData]) -> bool:
|
|
floor = data.data.get("detail", {}).get("rounds_data")
|
|
return any(d.data.get("detail", {}).get("rounds_data") == floor for d in old_data)
|
|
|
|
@staticmethod
|
|
def create(user_id: int, abyss_data: ImgTheaterData, character_data: Dict[int, int]):
|
|
data = HistoryDataImgTheater(abyss_data=abyss_data, character_data=character_data)
|
|
json_data = data.json(by_alias=True, encoder=json_encoder)
|
|
return HistoryData(
|
|
user_id=user_id,
|
|
data_id=abyss_data.schedule.id,
|
|
time_created=datetime.datetime.now(),
|
|
type=HistoryDataImgTheaterServices.DATA_TYPE,
|
|
data=jsonlib.loads(json_data),
|
|
)
|