PaiGram/core/cookies/models.py

import enum
from typing import Optional, Dict

from sqlmodel import SQLModel, Field, JSON, Enum, Column


class CookiesStatusEnum(int, enum.Enum):
    STATUS_SUCCESS = 0
    INVALID_COOKIES = 1
    TOO_MANY_REQUESTS = 2


class Cookies(SQLModel):
    __table_args__ = dict(mysql_charset="utf8mb4", mysql_collate="utf8mb4_general_ci")

    id: int = Field(primary_key=True)
    user_id: Optional[int] = Field(foreign_key="user.user_id")
    cookies: Optional[Dict[str, str]] = Field(sa_column=Column(JSON))
    status: Optional[CookiesStatusEnum] = Field(sa_column=Column(Enum(CookiesStatusEnum)))


class HyperionCookie(Cookies, table=True):
    __tablename__ = "mihoyo_cookies"


class HoyolabCookie(Cookies, table=True):
    __tablename__ = "hoyoverse_cookies"