mirror of
https://github.com/PaiGramTeam/GramCore.git
synced 2024-11-21 21:58:04 +00:00
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
import enum
|
|
from datetime import datetime
|
|
from typing import Optional, Dict, Any
|
|
|
|
from sqlalchemy import func, BigInteger, JSON
|
|
from sqlmodel import Column, DateTime, Enum, Field, SQLModel, Integer
|
|
|
|
__all__ = ("GachaLogRank", "GachaLogTypeEnum", "GachaLogQueryTypeEnum")
|
|
|
|
|
|
class GachaLogTypeEnum(int, enum.Enum):
|
|
CHARACTER = 0 # 角色
|
|
WEAPON = 1 # 武器
|
|
DEFAULT = 2 # 常驻
|
|
DEFAULT_WEAPON = 3 # 常驻武器
|
|
HUN = 4 # 集录
|
|
PET = 5 # 邦布
|
|
|
|
|
|
class GachaLogQueryTypeEnum(str, enum.Enum):
|
|
TOTAL = "score_1"
|
|
FIVE_STAR_AVG = "score_2"
|
|
UP_STAR_AVG = "score_3"
|
|
NO_WARP = "score_4"
|
|
|
|
|
|
class GachaLogRank(SQLModel, table=True):
|
|
__tablename__ = "gacha_log_rank"
|
|
__table_args__ = dict(mysql_charset="utf8mb4", mysql_collate="utf8mb4_general_ci")
|
|
id: Optional[int] = Field(default=None, sa_column=Column(Integer(), primary_key=True, autoincrement=True))
|
|
player_id: int = Field(sa_column=Column(BigInteger(), primary_key=True))
|
|
type: GachaLogTypeEnum = Field(sa_column=Column(Enum(GachaLogTypeEnum), primary_key=True))
|
|
score_1: int = Field(sa_column=Column(BigInteger(), default=0))
|
|
"""总抽数"""
|
|
score_2: int = Field(sa_column=Column(BigInteger(), default=0))
|
|
"""五星平均"""
|
|
score_3: int = Field(sa_column=Column(BigInteger(), default=0))
|
|
"""up 平均"""
|
|
score_4: int = Field(sa_column=Column(BigInteger(), default=0))
|
|
"""小保底不歪概率"""
|
|
score_5: int = Field(sa_column=Column(BigInteger(), default=0))
|
|
"""保留字段"""
|
|
data: Optional[Dict[str, Any]] = Field(sa_column=Column(JSON))
|
|
time_created: Optional[datetime] = Field(
|
|
sa_column=Column(DateTime, server_default=func.now()) # pylint: disable=E1102
|
|
)
|
|
time_updated: Optional[datetime] = Field(sa_column=Column(DateTime, onupdate=func.now())) # pylint: disable=E1102
|
|
|
|
def update_by_new(self, new_ins: "GachaLogRank"):
|
|
self.score_1 = new_ins.score_1
|
|
self.score_2 = new_ins.score_2
|
|
self.score_3 = new_ins.score_3
|
|
self.score_4 = new_ins.score_4
|
|
self.score_5 = new_ins.score_5
|
|
self.data = new_ins.data
|
|
self.time_updated = datetime.now()
|