import uuid from typing import TYPE_CHECKING from sqlalchemy import ForeignKey, Integer, JSON, String, Text, UniqueConstraint from sqlalchemy import Uuid as UUID from sqlalchemy.orm import Mapped, mapped_column, relationship from src.models.base import Base, TimestampMixin, UUIDPrimaryKeyMixin if TYPE_CHECKING: from src.models.config_include import ConfigInclude from src.models.config_mount import ConfigMount from src.models.project import Project from src.models.tool_type import ToolType from src.models.user import User class ConfigProfile(UUIDPrimaryKeyMixin, TimestampMixin, Base): __tablename__ = "config_profiles" __table_args__ = ( UniqueConstraint("user_id", "name", name="uq_config_profiles_user_name"), ) user_id: Mapped[uuid.UUID] = mapped_column( UUID(), ForeignKey("users.id", ondelete="CASCADE"), nullable=False ) project_id: Mapped[uuid.UUID | None] = mapped_column( UUID(), ForeignKey("projects.id", ondelete="CASCADE"), nullable=True ) tool_type_id: Mapped[uuid.UUID | None] = mapped_column( UUID(), ForeignKey("tool_types.id", ondelete="CASCADE"), nullable=True ) name: Mapped[str] = mapped_column(String(255), nullable=False) description: Mapped[str | None] = mapped_column(Text, nullable=True) environment_variables: Mapped[dict[str, str] | None] = mapped_column( JSON, default=dict, nullable=True ) start_command: Mapped[str | None] = mapped_column(Text, nullable=True) working_directory: Mapped[str | None] = mapped_column(Text, nullable=True) port: Mapped[int | None] = mapped_column(Integer, nullable=True) is_default: Mapped[bool] = mapped_column(default=False, nullable=False) user: Mapped["User"] = relationship() project: Mapped["Project | None"] = relationship() tool_type: Mapped["ToolType | None"] = relationship() includes: Mapped[list["ConfigInclude"]] = relationship( "ConfigInclude", foreign_keys="ConfigInclude.profile_id", back_populates="profile", cascade="all, delete-orphan", order_by="ConfigInclude.order_index", ) mounts: Mapped[list["ConfigMount"]] = relationship( "ConfigMount", back_populates="profile", cascade="all, delete-orphan", order_by="ConfigMount.order_index", )