66 lines
2.9 KiB
Python
66 lines
2.9 KiB
Python
"""BGP incident model for aggregated routing events."""
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import Column, DateTime, Float, ForeignKey, Index, Integer, JSON, String, Text
|
|
|
|
from app.core.enums import BGPStatus
|
|
from app.core.time import to_iso8601_utc
|
|
from app.db.session import Base
|
|
|
|
|
|
class BGPIncident(Base):
|
|
__tablename__ = "bgp_incidents"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
snapshot_id = Column(Integer, ForeignKey("data_snapshots.id"), nullable=True, index=True)
|
|
task_id = Column(Integer, ForeignKey("collection_tasks.id"), nullable=True, index=True)
|
|
source = Column(String(100), nullable=False, index=True)
|
|
incident_key = Column(String(255), nullable=False, index=True)
|
|
incident_type = Column(String(50), nullable=False, index=True)
|
|
title = Column(String(255), nullable=False)
|
|
summary = Column(Text, nullable=False)
|
|
severity = Column(String(20), nullable=False, index=True)
|
|
status = Column(String(20), nullable=False, default=BGPStatus.ACTIVE.value, index=True)
|
|
confidence = Column(Float, nullable=False, default=0.5)
|
|
started_at = Column(DateTime(timezone=True), nullable=False, default=datetime.utcnow, index=True)
|
|
ended_at = Column(DateTime(timezone=True), nullable=True)
|
|
affected_prefixes = Column(JSON, default=list)
|
|
affected_asns = Column(JSON, default=list)
|
|
affected_collectors = Column(JSON, default=list)
|
|
affected_regions = Column(JSON, default=list)
|
|
related_cables = Column(JSON, default=list)
|
|
related_ixps = Column(JSON, default=list)
|
|
evidence_refs = Column(JSON, default=list)
|
|
created_at = Column(DateTime(timezone=True), nullable=False, default=datetime.utcnow, index=True)
|
|
|
|
__table_args__ = (
|
|
Index("idx_bgp_incidents_source_created", "source", "created_at"),
|
|
Index("idx_bgp_incidents_type_status", "incident_type", "status"),
|
|
)
|
|
|
|
def to_dict(self) -> dict:
|
|
return {
|
|
"id": self.id,
|
|
"snapshot_id": self.snapshot_id,
|
|
"task_id": self.task_id,
|
|
"source": self.source,
|
|
"incident_key": self.incident_key,
|
|
"incident_type": self.incident_type,
|
|
"title": self.title,
|
|
"summary": self.summary,
|
|
"severity": self.severity,
|
|
"status": self.status,
|
|
"confidence": self.confidence,
|
|
"started_at": to_iso8601_utc(self.started_at),
|
|
"ended_at": to_iso8601_utc(self.ended_at),
|
|
"affected_prefixes": self.affected_prefixes or [],
|
|
"affected_asns": self.affected_asns or [],
|
|
"affected_collectors": self.affected_collectors or [],
|
|
"affected_regions": self.affected_regions or [],
|
|
"related_cables": self.related_cables or [],
|
|
"related_ixps": self.related_ixps or [],
|
|
"evidence_refs": self.evidence_refs or [],
|
|
"created_at": to_iso8601_utc(self.created_at),
|
|
}
|