Spaces:
Running
Running
File size: 2,626 Bytes
8b3905d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 | """SQLAlchemy ORM models for PostgreSQL."""
from __future__ import annotations
import uuid
from datetime import datetime, timezone
from sqlalchemy import JSON, DateTime, Float, Integer, String, Text
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
class Base(DeclarativeBase):
pass
class EventRecord(Base):
__tablename__ = "events"
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
timestamp: Mapped[datetime] = mapped_column(DateTime(timezone=True))
event_type: Mapped[str] = mapped_column(String(128))
source_ip: Mapped[str | None] = mapped_column(String(64), nullable=True)
host: Mapped[str] = mapped_column(String(256), default="unknown")
severity: Mapped[str] = mapped_column(String(32))
payload: Mapped[dict] = mapped_column(JSON, default=dict)
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
class IncidentRecord(Base):
__tablename__ = "incidents"
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
title: Mapped[str] = mapped_column(String(512))
summary: Mapped[str] = mapped_column(Text)
graph: Mapped[dict] = mapped_column(JSON, default=dict)
risk_score: Mapped[float] = mapped_column(Float, default=0)
severity: Mapped[str] = mapped_column(String(32))
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
class AlertRecord(Base):
__tablename__ = "alerts"
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
channel: Mapped[str] = mapped_column(String(64))
title: Mapped[str] = mapped_column(String(512))
body: Mapped[str] = mapped_column(Text)
severity: Mapped[str] = mapped_column(String(32))
incident_id: Mapped[uuid.UUID | None] = mapped_column(UUID(as_uuid=True), nullable=True)
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
class MetricSnapshot(Base):
__tablename__ = "metric_snapshots"
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
threats_detected: Mapped[int] = mapped_column(Integer, default=0)
active_incidents: Mapped[int] = mapped_column(Integer, default=0)
blocked: Mapped[int] = mapped_column(Integer, default=0)
recorded_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
|