40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
import uuid
|
||
|
|
from datetime import datetime
|
||
|
|
from sqlalchemy import String, Boolean, DateTime, ForeignKey, func
|
||
|
|
from sqlalchemy import Uuid
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||
|
|
from app.models.base import Base, UUIDMixin, TimestampMixin
|
||
|
|
|
||
|
|
|
||
|
|
class User(Base, UUIDMixin, TimestampMixin):
|
||
|
|
__tablename__ = "users"
|
||
|
|
|
||
|
|
email: Mapped[str] = mapped_column(String(255), unique=True, nullable=False, index=True)
|
||
|
|
hashed_password: Mapped[str] = mapped_column(String(255), nullable=False)
|
||
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
||
|
|
|
||
|
|
refresh_tokens: Mapped[list["RefreshToken"]] = relationship(
|
||
|
|
"RefreshToken", back_populates="user", cascade="all, delete-orphan"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class RefreshToken(Base, UUIDMixin):
|
||
|
|
__tablename__ = "refresh_tokens"
|
||
|
|
|
||
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
||
|
|
Uuid(as_uuid=True),
|
||
|
|
ForeignKey("users.id", ondelete="CASCADE"),
|
||
|
|
nullable=False,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
token_hash: Mapped[str] = mapped_column(String(255), nullable=False)
|
||
|
|
expires_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
||
|
|
revoked: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
|
||
|
|
created_at: Mapped[datetime] = mapped_column(
|
||
|
|
DateTime(timezone=True),
|
||
|
|
server_default=func.now(),
|
||
|
|
nullable=False,
|
||
|
|
)
|
||
|
|
|
||
|
|
user: Mapped["User"] = relationship("User", back_populates="refresh_tokens")
|