nonebot-plugin-tetris-stats/nonebot_plugin_tetris_stats/games/tetrio/api/models.py

from datetime import datetime
from typing import Literal

from nonebot_plugin_orm import Model
from sqlalchemy import DateTime, String
from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column

from ....db.models import PydanticType
from .schemas.base import SuccessModel
from .typedefs import Records, Summaries


class TETRIOHistoricalData(MappedAsDataclass, Model):
    id: Mapped[int] = mapped_column(init=False, primary_key=True)
    user_unique_identifier: Mapped[str] = mapped_column(String(24), index=True)
    api_type: Mapped[Literal['User Info', Records, Summaries]] = mapped_column(String(32), index=True)
    data: Mapped[SuccessModel] = mapped_column(PydanticType(get_model=[SuccessModel.__subclasses__], models=set()))
    update_time: Mapped[datetime] = mapped_column(DateTime, index=True)