|
1 | 1 | from datetime import datetime |
| 2 | +from sqlmodel import SQLModel, Field |
2 | 3 |
|
3 | | -from sqlalchemy import DateTime, String |
4 | | -from sqlalchemy.orm import Mapped, mapped_column |
5 | | - |
6 | | -from .database import Base |
7 | | - |
8 | | - |
9 | | -class TokenBlacklist(Base): |
10 | | - __tablename__ = "token_blacklist" |
11 | | - |
12 | | - id: Mapped[int] = mapped_column("id", autoincrement=True, nullable=False, unique=True, primary_key=True, init=False) |
13 | | - token: Mapped[str] = mapped_column(String, unique=True, index=True) |
14 | | - expires_at: Mapped[datetime] = mapped_column(DateTime) |
| 4 | +class TokenBlacklist(SQLModel, table=True): |
| 5 | + id: int = Field(default=None, primary_key=True, nullable=False) |
| 6 | + token: str = Field(index=True, nullable=False, unique=True) |
| 7 | + expires_at: datetime = Field(nullable=False) |
0 commit comments