saksham
1a4c80958f
FastAPI + Celery + Next.js + Postgres/Redis app with company monitoring, source collection, LLM-based change analysis, enrichment, and account security (Turnstile, escalating lockout, email verification).
41 lines
1.8 KiB
Python
41 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from sqlalchemy import JSON, Boolean, DateTime, Enum, ForeignKey, Integer, String
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base, TimestampMixin, UUIDPrimaryKeyMixin
|
|
from app.models.enums import MonitoringFrequency, SeverityLevel
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.company import Company
|
|
|
|
|
|
class MonitorConfiguration(Base, UUIDPrimaryKeyMixin, TimestampMixin):
|
|
__tablename__ = "monitor_configurations"
|
|
|
|
company_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("companies.id", ondelete="CASCADE"), unique=True, index=True
|
|
)
|
|
frequency_type: Mapped[MonitoringFrequency] = mapped_column(
|
|
Enum(MonitoringFrequency, native_enum=False, length=30),
|
|
default=MonitoringFrequency.WEEKLY,
|
|
)
|
|
# Only meaningful when frequency_type == CUSTOM: interval_minutes takes
|
|
# precedence if set, otherwise cron_expression is used.
|
|
interval_minutes: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
cron_expression: Mapped[str | None] = mapped_column(String(120), nullable=True)
|
|
timezone: Mapped[str] = mapped_column(String(64), default="America/New_York")
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
next_run: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
last_run: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
severity_threshold: Mapped[SeverityLevel] = mapped_column(
|
|
Enum(SeverityLevel, native_enum=False, length=20), default=SeverityLevel.MEDIUM
|
|
)
|
|
source_configuration: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict)
|
|
|
|
company: Mapped[Company] = relationship(back_populates="monitor_configuration")
|