"""Maintenance window ORM model for scheduled maintenance periods. Maintenance windows allow operators to define time periods during which alerts are suppressed for specific devices (or all devices in a tenant). """ import uuid from datetime import datetime from sqlalchemy import Boolean, DateTime, ForeignKey, Text, VARCHAR, func from sqlalchemy.dialects.postgresql import JSONB, UUID from sqlalchemy.orm import Mapped, mapped_column from app.database import Base class MaintenanceWindow(Base): """Scheduled maintenance window with optional alert suppression. device_ids is a JSONB array of device UUID strings. An empty array means "all devices in tenant". """ __tablename__ = "maintenance_windows" 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, ) name: Mapped[str] = mapped_column(VARCHAR(200), nullable=False) device_ids: Mapped[list] = mapped_column( JSONB, nullable=False, server_default="'[]'::jsonb", ) start_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), nullable=False, ) end_at: Mapped[datetime] = mapped_column( DateTime(timezone=True), nullable=False, ) suppress_alerts: Mapped[bool] = mapped_column( Boolean, nullable=False, default=True, server_default="true", ) notes: Mapped[str | None] = mapped_column(Text, nullable=True) created_by: Mapped[uuid.UUID | None] = mapped_column( UUID(as_uuid=True), ForeignKey("users.id", ondelete="SET NULL"), 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(), nullable=False, ) def __repr__(self) -> str: return f""