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).
46 lines
2.2 KiB
Python
46 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from sqlalchemy import JSON, Boolean, DateTime, Enum, Float, ForeignKey, Integer, String
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.db.base import Base, TimestampMixin, UUIDPrimaryKeyMixin
|
|
from app.models.enums import MonitoringFrequency, SourceStatus, SourceType
|
|
|
|
|
|
class Source(Base, UUIDPrimaryKeyMixin, TimestampMixin):
|
|
__tablename__ = "sources"
|
|
|
|
company_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("companies.id", ondelete="CASCADE"), index=True
|
|
)
|
|
source_type: Mapped[SourceType] = mapped_column(Enum(SourceType, native_enum=False, length=20))
|
|
name: Mapped[str] = mapped_column(String(200))
|
|
base_url: Mapped[str | None] = mapped_column(String(500), nullable=True)
|
|
active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
status: Mapped[SourceStatus] = mapped_column(
|
|
Enum(SourceStatus, native_enum=False, length=20), default=SourceStatus.ACTIVE
|
|
)
|
|
trust_score: Mapped[float] = mapped_column(Float, default=0.7)
|
|
last_checked: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
last_successful_check: Mapped[datetime | None] = mapped_column(
|
|
DateTime(timezone=True), nullable=True
|
|
)
|
|
failure_count: Mapped[int] = mapped_column(Integer, default=0)
|
|
configuration_metadata: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict)
|
|
|
|
# Per-source check cadence override - NULL frequency_type means "inherit
|
|
# the company's default MonitorConfiguration cadence" (the behavior for
|
|
# every source before this feature existed, and still the default for
|
|
# any source that never sets an override). See app/tasks/scheduler.py
|
|
# and app/services/scheduling.py for how these combine into due-ness.
|
|
frequency_type: Mapped[MonitoringFrequency | None] = mapped_column(
|
|
Enum(MonitoringFrequency, native_enum=False, length=20), nullable=True
|
|
)
|
|
interval_minutes: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
cron_expression: Mapped[str | None] = mapped_column(String(120), nullable=True)
|
|
next_check: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|