64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""Workspace models for VoIdea.
|
|
|
|
Personal workspace (1 per user, free) + Team workspaces (tariff-gated).
|
|
"""
|
|
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import Boolean, ForeignKey, Integer, String, Text
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.core.base import SQLBase, TimestampMixin, UUIDMixin
|
|
|
|
|
|
class Workspace(SQLBase, UUIDMixin, TimestampMixin):
|
|
__tablename__ = "workspaces"
|
|
|
|
name: Mapped[str] = mapped_column(
|
|
String(255), nullable=False
|
|
)
|
|
type: Mapped[str] = mapped_column(
|
|
String(20), nullable=False, default="personal", index=True
|
|
)
|
|
owner_id: Mapped[UUID] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("users.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
description: Mapped[Optional[str]] = mapped_column(
|
|
Text, nullable=True, default=""
|
|
)
|
|
|
|
owner = relationship("User", backref="owned_workspaces", lazy="selectin")
|
|
members = relationship(
|
|
"WorkspaceMembership", back_populates="workspace",
|
|
lazy="selectin", cascade="all, delete-orphan",
|
|
)
|
|
|
|
|
|
class WorkspaceMembership(SQLBase, UUIDMixin, TimestampMixin):
|
|
__tablename__ = "workspace_memberships"
|
|
|
|
workspace_id: Mapped[UUID] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("workspaces.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
user_id: Mapped[UUID] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("users.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
role: Mapped[str] = mapped_column(
|
|
String(20), nullable=False, default="member"
|
|
)
|
|
|
|
workspace = relationship(
|
|
"Workspace", back_populates="members", lazy="selectin"
|
|
)
|
|
user = relationship("User", backref="workspace_memberships", lazy="selectin")
|