mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-11-25 01:29:49 +00:00
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
import datetime
|
|
from typing import List
|
|
|
|
from pytz import timezone
|
|
from simnet.models.starrail.diary import StarRailDiary
|
|
from simnet.models.zzz.chronicle.challenge import ZZZChallenge
|
|
|
|
from core.services.history_data.models import (
|
|
HistoryData,
|
|
HistoryDataTypeEnum,
|
|
HistoryDataAbyss,
|
|
HistoryDataLedger,
|
|
)
|
|
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",
|
|
)
|
|
|
|
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:
|
|
floors = data.data.get("abyss_data", {}).get("all_floor_detail")
|
|
return any(d.data.get("abyss_data", {}).get("all_floor_detail") == floors for d in old_data)
|
|
|
|
@staticmethod
|
|
def create(user_id: int, abyss_data: ZZZChallenge):
|
|
data = HistoryDataAbyss(abyss_data=abyss_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: StarRailDiary):
|
|
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),
|
|
)
|