58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
|
"""Tenant model for multi-tenancy support."""
|
||
|
|
|
||
|
|
from typing import TYPE_CHECKING
|
||
|
|
|
||
|
|
from sqlalchemy import String
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||
|
|
|
||
|
|
from app.models.base import Base, TimestampMixin, UUIDMixin
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from app.models.agent import Agent
|
||
|
|
from app.models.event import Event
|
||
|
|
from app.models.server import Server
|
||
|
|
from app.models.task import Task
|
||
|
|
|
||
|
|
|
||
|
|
class Tenant(UUIDMixin, TimestampMixin, Base):
|
||
|
|
"""
|
||
|
|
Tenant model representing a customer organization.
|
||
|
|
|
||
|
|
Each tenant has isolated servers, tasks, agents, and events.
|
||
|
|
"""
|
||
|
|
|
||
|
|
__tablename__ = "tenants"
|
||
|
|
|
||
|
|
name: Mapped[str] = mapped_column(
|
||
|
|
String(255),
|
||
|
|
unique=True,
|
||
|
|
nullable=False,
|
||
|
|
index=True,
|
||
|
|
)
|
||
|
|
domain: Mapped[str | None] = mapped_column(
|
||
|
|
String(255),
|
||
|
|
unique=True,
|
||
|
|
nullable=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Relationships
|
||
|
|
servers: Mapped[list["Server"]] = relationship(
|
||
|
|
back_populates="tenant",
|
||
|
|
lazy="selectin",
|
||
|
|
)
|
||
|
|
tasks: Mapped[list["Task"]] = relationship(
|
||
|
|
back_populates="tenant",
|
||
|
|
lazy="selectin",
|
||
|
|
)
|
||
|
|
agents: Mapped[list["Agent"]] = relationship(
|
||
|
|
back_populates="tenant",
|
||
|
|
lazy="selectin",
|
||
|
|
)
|
||
|
|
events: Mapped[list["Event"]] = relationship(
|
||
|
|
back_populates="tenant",
|
||
|
|
lazy="selectin",
|
||
|
|
)
|
||
|
|
|
||
|
|
def __repr__(self) -> str:
|
||
|
|
return f"<Tenant(id={self.id}, name={self.name})>"
|