from __future__ import annotations import uuid from sqlalchemy import Boolean, ForeignKey, JSON, String, Text, UniqueConstraint from sqlalchemy.orm import Mapped, mapped_column from govoplan_core.db.base import Base, TimestampMixin def new_uuid() -> str: return str(uuid.uuid4()) class GovernanceTemplate(Base, TimestampMixin): __tablename__ = "governance_templates" __table_args__ = (UniqueConstraint("kind", "slug", name="uq_governance_templates_kind_slug"),) id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid) kind: Mapped[str] = mapped_column(String(20), nullable=False, index=True) slug: Mapped[str] = mapped_column(String(100), nullable=False) name: Mapped[str] = mapped_column(String(255), nullable=False) description: Mapped[str | None] = mapped_column(Text) permissions: Mapped[list[str]] = mapped_column(JSON, default=list, nullable=False) is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) class GovernanceTemplateAssignment(Base, TimestampMixin): __tablename__ = "governance_template_assignments" __table_args__ = (UniqueConstraint("template_id", "tenant_id", name="uq_governance_template_tenant"),) id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid) template_id: Mapped[str] = mapped_column(ForeignKey("governance_templates.id", ondelete="CASCADE"), nullable=False, index=True) tenant_id: Mapped[str] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True) mode: Mapped[str] = mapped_column(String(20), default="available", nullable=False) __all__ = ["GovernanceTemplate", "GovernanceTemplateAssignment", "new_uuid"]