59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
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",
|
|
)
|