File size: 2,347 Bytes
7c6ffa6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | from __future__ import annotations
from datetime import datetime
from sqlalchemy import DateTime, ForeignKey, Index, Integer, String, func
from sqlalchemy.orm import Mapped, mapped_column
from app.core.database import Base
from app.utils.ids import prefixed_id
class UserPlan(Base):
__tablename__ = "user_plans"
id: Mapped[str] = mapped_column(
String(40),
primary_key=True,
default=lambda: prefixed_id("plan"),
)
user_id: Mapped[str] = mapped_column(
String(40),
ForeignKey("users.id"),
unique=True,
index=True,
nullable=False,
)
selected_plan: Mapped[str] = mapped_column(
String(60),
default="free",
nullable=False,
)
status: Mapped[str] = mapped_column(
String(40),
default="active",
nullable=False,
)
trial_started_at: Mapped[datetime | None] = mapped_column(
DateTime(timezone=True),
nullable=True,
)
trial_ends_at: Mapped[datetime | None] = mapped_column(
DateTime(timezone=True),
nullable=True,
)
# Tracks the start of the current billing period for monthly reset logic
period_start: Mapped[datetime | None] = mapped_column(
DateTime(timezone=True),
nullable=True,
)
monthly_video_limit: Mapped[int] = mapped_column(Integer, default=3, nullable=False)
monthly_video_used: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
monthly_generation_limit: Mapped[int] = mapped_column(Integer, default=30, nullable=False)
monthly_generation_used: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
created_at: Mapped[datetime] = mapped_column(
DateTime(timezone=True),
server_default=func.now(),
nullable=False,
)
updated_at: Mapped[datetime] = mapped_column(
DateTime(timezone=True),
server_default=func.now(),
onupdate=func.now(),
nullable=False,
)
# Supabase/Postgres best practices: explicit indexes for common query patterns
# (user lookups already have unique+index on user_id; add for plan filtering + period resets)
__table_args__ = (
Index("ix_user_plans_selected_plan_status", "selected_plan", "status"),
Index("ix_user_plans_period_start", "period_start"),
)
|