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).
77 lines
3.1 KiB
Python
77 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import JSON, Enum, ForeignKey, String, Text
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base, TimestampMixin, UUIDPrimaryKeyMixin
|
|
from app.models.enums import CompanyStatus
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.company_enrichment import CompanyEnrichment
|
|
from app.models.monitor_configuration import MonitorConfiguration
|
|
from app.models.notification_destination import NotificationDestinationCompany
|
|
|
|
|
|
class Company(Base, UUIDPrimaryKeyMixin, TimestampMixin):
|
|
__tablename__ = "companies"
|
|
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("users.id", ondelete="CASCADE"), index=True
|
|
)
|
|
name: Mapped[str] = mapped_column(String(200))
|
|
slug: Mapped[str] = mapped_column(String(220), index=True)
|
|
official_website: Mapped[str | None] = mapped_column(String(500), nullable=True)
|
|
description: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
monitoring_focus: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
industry: Mapped[str | None] = mapped_column(String(120), nullable=True)
|
|
country: Mapped[str | None] = mapped_column(String(120), nullable=True)
|
|
region: Mapped[str | None] = mapped_column(String(120), nullable=True)
|
|
headquarters: Mapped[str | None] = mapped_column(String(200), nullable=True)
|
|
# Best-effort discovered identifiers, e.g. {"ticker": "ACME", "linkedin_url": "..."}
|
|
# - see app/prompts/company_profile.py. Empty dict, never fabricated.
|
|
public_identifiers: Mapped[dict[str, str]] = mapped_column(JSON, default=dict)
|
|
status: Mapped[CompanyStatus] = mapped_column(
|
|
Enum(CompanyStatus, native_enum=False, length=20), default=CompanyStatus.ACTIVE
|
|
)
|
|
|
|
aliases: Mapped[list[CompanyAlias]] = relationship(
|
|
back_populates="company", cascade="all, delete-orphan"
|
|
)
|
|
competitors: Mapped[list[Competitor]] = relationship(
|
|
back_populates="company", cascade="all, delete-orphan"
|
|
)
|
|
monitor_configuration: Mapped[MonitorConfiguration | None] = relationship(
|
|
back_populates="company", cascade="all, delete-orphan", uselist=False
|
|
)
|
|
enrichment: Mapped[CompanyEnrichment | None] = relationship(
|
|
back_populates="company", cascade="all, delete-orphan", uselist=False
|
|
)
|
|
notification_links: Mapped[list[NotificationDestinationCompany]] = relationship(
|
|
cascade="all, delete-orphan"
|
|
)
|
|
|
|
|
|
class CompanyAlias(Base, UUIDPrimaryKeyMixin, TimestampMixin):
|
|
__tablename__ = "company_aliases"
|
|
|
|
company_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("companies.id", ondelete="CASCADE"), index=True
|
|
)
|
|
alias: Mapped[str] = mapped_column(String(200))
|
|
|
|
company: Mapped[Company] = relationship(back_populates="aliases")
|
|
|
|
|
|
class Competitor(Base, UUIDPrimaryKeyMixin, TimestampMixin):
|
|
__tablename__ = "competitors"
|
|
|
|
company_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("companies.id", ondelete="CASCADE"), index=True
|
|
)
|
|
name: Mapped[str] = mapped_column(String(200))
|
|
|
|
company: Mapped[Company] = relationship(back_populates="competitors")
|