Spaces:
Running
Running
| from __future__ import annotations | |
| from datetime import datetime | |
| from uuid import uuid4 | |
| from sqlalchemy import ( | |
| JSON, | |
| DateTime, | |
| ForeignKey, | |
| Index, | |
| Integer, | |
| String, | |
| UniqueConstraint, | |
| ) | |
| from sqlalchemy.orm import Mapped, mapped_column | |
| from app.security.models import Base, utcnow | |
| class Team(Base): | |
| __tablename__ = "teams" | |
| __table_args__ = ( | |
| Index("ix_teams_workspace_id", "workspace_id"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| name: Mapped[str] = mapped_column(String(255), nullable=False) | |
| created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow) | |
| updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow, onupdate=utcnow) | |
| class TeamMember(Base): | |
| __tablename__ = "team_members" | |
| __table_args__ = ( | |
| UniqueConstraint("team_id", "user_id", name="uq_team_member"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| team_id: Mapped[str] = mapped_column(String(36), ForeignKey("teams.id", ondelete="CASCADE"), nullable=False) | |
| user_id: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"), nullable=False) | |
| role: Mapped[str] = mapped_column(String(50), nullable=False) | |
| created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow) | |
| class ProjectCollaborator(Base): | |
| __tablename__ = "project_collaborators" | |
| __table_args__ = ( | |
| UniqueConstraint("project_id", "user_id", name="uq_project_collaborator"), | |
| Index("ix_project_collaborators_workspace", "workspace_id"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| project_id: Mapped[str] = mapped_column(String(36), ForeignKey("projects.id", ondelete="CASCADE"), nullable=False) | |
| user_id: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"), nullable=False) | |
| role: Mapped[str] = mapped_column(String(50), nullable=False) | |
| created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow) | |
| class CollaborationActivity(Base): | |
| __tablename__ = "collaboration_activities" | |
| __table_args__ = ( | |
| Index("ix_collaboration_activities_workspace_created", "workspace_id", "created_at"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| user_id: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"), nullable=False) | |
| action: Mapped[str] = mapped_column(String(100), nullable=False) | |
| entity_id: Mapped[str] = mapped_column(String(36), nullable=False) | |
| entity_type: Mapped[str] = mapped_column(String(64), nullable=False) | |
| metadata_json: Mapped[dict[str, object]] = mapped_column("metadata", JSON, nullable=False, default=dict) | |
| created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow) | |
| class Invitation(Base): | |
| __tablename__ = "invitations" | |
| __table_args__ = ( | |
| Index("ix_invitations_workspace_id", "workspace_id"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| email: Mapped[str] = mapped_column(String(255), nullable=False) | |
| role: Mapped[str] = mapped_column(String(50), nullable=False) | |
| status: Mapped[str] = mapped_column(String(20), nullable=False, default="pending") | |
| token_hash: Mapped[str | None] = mapped_column(String(255), nullable=True) | |
| expires_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False) | |
| class ApprovalWorkflow(Base): | |
| __tablename__ = "approval_workflows" | |
| __table_args__ = ( | |
| Index("ix_approval_workflows_project_id", "project_id"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| project_id: Mapped[str] = mapped_column(String(36), ForeignKey("projects.id", ondelete="CASCADE"), nullable=False) | |
| name: Mapped[str] = mapped_column(String(255), nullable=False) | |
| created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow) | |
| class ApprovalRequest(Base): | |
| __tablename__ = "approval_requests" | |
| __table_args__ = ( | |
| Index("ix_approval_requests_workflow_id", "workflow_id"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| workflow_id: Mapped[str] = mapped_column(String(36), ForeignKey("approval_workflows.id", ondelete="CASCADE"), nullable=False) | |
| project_id: Mapped[str] = mapped_column(String(36), ForeignKey("projects.id", ondelete="CASCADE"), nullable=False) | |
| editor_revision: Mapped[int] = mapped_column(Integer, nullable=False, default=1) | |
| status: Mapped[str] = mapped_column(String(20), nullable=False, default="pending") | |
| created_by: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"), nullable=False) | |
| created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow) | |
| class ReviewComment(Base): | |
| __tablename__ = "review_comments" | |
| __table_args__ = ( | |
| Index("ix_review_comments_request_id", "request_id"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column(String(36), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False) | |
| request_id: Mapped[str] = mapped_column(String(36), ForeignKey("approval_requests.id", ondelete="CASCADE"), nullable=False) | |
| user_id: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"), nullable=False) | |
| content: Mapped[str] = mapped_column(String(2000), nullable=False) | |
| created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow) | |