35 lines
1.8 KiB
Python
35 lines
1.8 KiB
Python
from datetime import datetime, timezone
|
|
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import DateTime, Integer, String
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.db import Base
|
|
|
|
|
|
class Tenant(Base):
|
|
__tablename__ = "tenants"
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
|
name: Mapped[str] = mapped_column(String(100), nullable=False)
|
|
code: Mapped[str] = mapped_column(String(64), nullable=False, unique=True, index=True)
|
|
status: Mapped[str] = mapped_column(String(32), default="active")
|
|
plan_name: Mapped[str] = mapped_column(String(64), default="trial")
|
|
plan_code: Mapped[Optional[str]] = mapped_column(String(64), nullable=True, default="trial")
|
|
billing_status: Mapped[Optional[str]] = mapped_column(String(32), nullable=True, default="trialing")
|
|
trial_started_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
trial_ends_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
review_quota_monthly: Mapped[Optional[int]] = mapped_column(Integer, nullable=True, default=30)
|
|
generation_quota_monthly: Mapped[Optional[int]] = mapped_column(Integer, nullable=True, default=20)
|
|
knowledge_base_quota: Mapped[Optional[int]] = mapped_column(Integer, nullable=True, default=1)
|
|
storage_quota_mb: Mapped[Optional[int]] = mapped_column(Integer, nullable=True, default=2048)
|
|
max_users: Mapped[int] = mapped_column(Integer, default=20)
|
|
max_storage_mb: Mapped[int] = mapped_column(Integer, default=10240)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True),
|
|
default=lambda: datetime.now(timezone.utc),
|
|
onupdate=lambda: datetime.now(timezone.utc),
|
|
)
|