"""Site model -- physical location grouping for devices.""" import uuid from datetime import datetime from sqlalchemy import DateTime, Float, ForeignKey, String, Text, UniqueConstraint, func from sqlalchemy.dialects.postgresql import UUID from sqlalchemy.orm import Mapped, mapped_column, relationship from app.models.base import Base class Site(Base): __tablename__ = "sites" __table_args__ = (UniqueConstraint("tenant_id", "name", name="uq_sites_tenant_name"),) id: Mapped[uuid.UUID] = mapped_column( UUID(as_uuid=True), primary_key=True, default=uuid.uuid4, server_default=func.gen_random_uuid(), ) tenant_id: Mapped[uuid.UUID] = mapped_column( UUID(as_uuid=True), ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True, ) name: Mapped[str] = mapped_column(String(255), nullable=False) latitude: Mapped[float | None] = mapped_column(Float, nullable=True) longitude: Mapped[float | None] = mapped_column(Float, nullable=True) address: Mapped[str | None] = mapped_column(Text, nullable=True) elevation: Mapped[float | None] = mapped_column(Float, nullable=True) notes: Mapped[str | None] = mapped_column(Text, nullable=True) created_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), server_default=func.now(), nullable=False ) updated_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), server_default=func.now(), onupdate=func.now(), nullable=False ) # Relationships tenant: Mapped["Tenant"] = relationship("Tenant", back_populates="sites") # type: ignore[name-defined] devices: Mapped[list["Device"]] = relationship( # type: ignore[name-defined] "Device", back_populates="site", foreign_keys="[Device.site_id]" ) def __repr__(self) -> str: return f""