Spaces:
Running
Running
File size: 2,142 Bytes
cfe45d5 | 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 | import enum
from datetime import datetime
from sqlalchemy import String, Enum, DateTime, Integer, Boolean
from sqlalchemy.orm import Mapped, mapped_column, relationship
from app.database import Base
class PlanTier(str, enum.Enum):
FREE = "free"
PRO = "pro"
class User(Base):
__tablename__ = "users"
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
email: Mapped[str] = mapped_column(String(320), unique=True, nullable=False, index=True)
name: Mapped[str] = mapped_column(String(255), nullable=False)
picture: Mapped[str | None] = mapped_column(String(2048), nullable=True)
google_id: Mapped[str] = mapped_column(String(255), unique=True, nullable=False, index=True)
# Subscription
plan: Mapped[PlanTier] = mapped_column(Enum(PlanTier), default=PlanTier.FREE)
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)
videos_used_this_period: Mapped[int] = mapped_column(Integer, default=0)
video_limit_bonus: Mapped[int] = mapped_column(Integer, default=0, server_default="0") # per-video credits purchased
period_start: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
# Relationships
projects = relationship("Project", back_populates="user", cascade="all, delete-orphan")
subscriptions = relationship("Subscription", back_populates="user", cascade="all, delete-orphan")
@property
def video_limit(self) -> int:
"""Max videos allowed in the current billing period."""
base = 1 if self.plan == PlanTier.FREE else 100 # Free: 1, Pro: 100/month
return base + (self.video_limit_bonus or 0)
@property
def can_create_video(self) -> bool:
return self.videos_used_this_period < self.video_limit
|