from datetime import datetime, timezone from sqlalchemy import String, Boolean, DateTime, Text from sqlalchemy.orm import Mapped, mapped_column, relationship from app.core.database import Base class User(Base): __tablename__ = "users" id: Mapped[int] = mapped_column(primary_key=True) email: Mapped[str] = mapped_column(String(255), unique=True, index=True) name: Mapped[str] = mapped_column(String(255)) hashed_password: Mapped[str | None] = mapped_column(String(255), nullable=True) is_pro: Mapped[bool] = mapped_column(Boolean, default=False) created_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), default=lambda: datetime.now(timezone.utc) ) # Stripe stripe_customer_id: Mapped[str | None] = mapped_column(String(255), nullable=True, unique=True) stripe_subscription_id: Mapped[str | None] = mapped_column(String(255), nullable=True) # Personalization blocked_genres: Mapped[str | None] = mapped_column(Text, nullable=True) adventurousness: Mapped[int] = mapped_column(default=3, server_default="3") # Spotify OAuth spotify_id: Mapped[str | None] = mapped_column(String(255), nullable=True, unique=True) spotify_access_token: Mapped[str | None] = mapped_column(Text, nullable=True) spotify_refresh_token: Mapped[str | None] = mapped_column(Text, nullable=True) # Relationships playlists: Mapped[list["Playlist"]] = relationship(back_populates="user", cascade="all, delete-orphan") recommendations: Mapped[list["Recommendation"]] = relationship(back_populates="user", cascade="all, delete-orphan") from app.models.playlist import Playlist # noqa: E402 from app.models.recommendation import Recommendation # noqa: E402