53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
from struct import Struct
|
|
from typing import TYPE_CHECKING
|
|
|
|
from pydantic import BaseModel
|
|
from sqlmodel import (
|
|
VARBINARY,
|
|
Column,
|
|
Field,
|
|
Relationship,
|
|
SQLModel,
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
from .beatmap import Beatmap
|
|
FAILTIME_STRUCT = Struct("<100i")
|
|
|
|
|
|
class FailTime(SQLModel, table=True):
|
|
__tablename__: str = "failtime"
|
|
beatmap_id: int = Field(primary_key=True, foreign_key="beatmaps.id")
|
|
exit: bytes = Field(sa_column=Column(VARBINARY(400), nullable=False))
|
|
fail: bytes = Field(sa_column=Column(VARBINARY(400), nullable=False))
|
|
|
|
beatmap: "Beatmap" = Relationship(back_populates="failtimes")
|
|
|
|
@property
|
|
def exit_(self) -> list[int]:
|
|
return list(FAILTIME_STRUCT.unpack(self.exit))
|
|
|
|
@property
|
|
def fail_(self) -> list[int]:
|
|
return list(FAILTIME_STRUCT.unpack(self.fail))
|
|
|
|
@classmethod
|
|
def from_resp(cls, beatmap_id: int, failtime: "FailTimeResp") -> "FailTime":
|
|
return cls(
|
|
beatmap_id=beatmap_id,
|
|
exit=FAILTIME_STRUCT.pack(*failtime.exit),
|
|
fail=FAILTIME_STRUCT.pack(*failtime.fail),
|
|
)
|
|
|
|
|
|
class FailTimeResp(BaseModel):
|
|
exit: list[int] = Field(default_factory=lambda: list(FAILTIME_STRUCT.unpack(b"\x00" * 400)))
|
|
fail: list[int] = Field(default_factory=lambda: list(FAILTIME_STRUCT.unpack(b"\x00" * 400)))
|
|
|
|
@classmethod
|
|
def from_db(cls, failtime: FailTime) -> "FailTimeResp":
|
|
return cls(
|
|
exit=list(FAILTIME_STRUCT.unpack(failtime.exit)),
|
|
fail=list(FAILTIME_STRUCT.unpack(failtime.fail)),
|
|
)
|