mirror of
https://github.com/hansjone/netx.git
synced 2026-10-09 06:40:45 +08:00
oliver
d26e4ca2fc
Brownfield tables still require task_id while ORM only wrote device_id; mirror both columns on write and relax the leftover constraint. Co-authored-by: Cursor <cursoragent@cursor.com>
170 lines
8.8 KiB
Python
170 lines
8.8 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from uuid import uuid4
|
|
|
|
from sqlalchemy import BigInteger, Boolean, DateTime, Float, ForeignKey, Integer, LargeBinary, String, Text, UniqueConstraint, event
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from ..db import Base
|
|
from ..timeutil import utcnow_naive
|
|
from ._types import JsonType as _JsonType
|
|
|
|
|
|
def _mirror_device_task_id(target: object) -> None:
|
|
"""Keep legacy task_id column aligned with device_id (brownfield NOT NULL)."""
|
|
did = str(getattr(target, "device_id", None) or getattr(target, "task_id", None) or "")
|
|
target.device_id = did # type: ignore[attr-defined]
|
|
target.task_id = did # type: ignore[attr-defined]
|
|
|
|
class PortTrafficDevice(Base):
|
|
"""Per-NE port traffic monitoring config (device-centric)."""
|
|
|
|
__tablename__ = "port_traffic_device"
|
|
__table_args__ = (
|
|
UniqueConstraint("source", "ne_id", name="uq_port_traffic_device_ne"),
|
|
)
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True, default=lambda: uuid4().hex)
|
|
source: Mapped[str] = mapped_column(String(32), default="managed", index=True)
|
|
ne_id: Mapped[str] = mapped_column(String(128), default="", index=True)
|
|
ne_name: Mapped[str] = mapped_column(String(256), default="")
|
|
ne_ip: Mapped[str] = mapped_column(String(128), default="")
|
|
vendor: Mapped[str] = mapped_column(String(64), default="")
|
|
note: Mapped[str] = mapped_column(String(256), default="") # optional remark
|
|
status: Mapped[str] = mapped_column(String(32), default="draft", index=True) # draft|running|paused|stopped
|
|
interval_sec: Mapped[int] = mapped_column(Integer, default=60)
|
|
retention_days: Mapped[int] = mapped_column(Integer, default=7)
|
|
concurrency: Mapped[int] = mapped_column(Integer, default=1)
|
|
collect_running: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
last_collect_started_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
last_collect_ended_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
last_error: Mapped[str] = mapped_column(String(1024), default="")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive, index=True)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive)
|
|
|
|
|
|
# Back-compat alias while callers migrate.
|
|
PortTrafficTask = PortTrafficDevice
|
|
|
|
|
|
class PortTrafficSeries(Base):
|
|
"""Logical port (business link) that survives physical NE/if replacement."""
|
|
|
|
__tablename__ = "port_traffic_series"
|
|
__table_args__ = (UniqueConstraint("device_id", "title", name="uq_port_traffic_series_title"),)
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True, default=lambda: uuid4().hex)
|
|
device_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
# Legacy column retained for brownfield DBs that still enforce NOT NULL task_id.
|
|
task_id: Mapped[str] = mapped_column(String(64), default="")
|
|
title: Mapped[str] = mapped_column(String(256), default="")
|
|
status: Mapped[str] = mapped_column(String(32), default="active", index=True) # active|disabled
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive)
|
|
|
|
|
|
class PortTrafficTarget(Base):
|
|
"""Monitored interface under a device monitoring config."""
|
|
|
|
__tablename__ = "port_traffic_target"
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True, default=lambda: uuid4().hex)
|
|
device_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
# Legacy column retained for brownfield DBs that still enforce NOT NULL task_id.
|
|
task_id: Mapped[str] = mapped_column(String(64), default="")
|
|
series_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
source: Mapped[str] = mapped_column(String(32), default="managed", index=True)
|
|
target_id: Mapped[str] = mapped_column(String(128), index=True) # NE id
|
|
ne_name: Mapped[str] = mapped_column(String(256), default="")
|
|
ne_ip: Mapped[str] = mapped_column(String(128), default="")
|
|
vendor: Mapped[str] = mapped_column(String(64), default="")
|
|
ifname: Mapped[str] = mapped_column(String(128), default="")
|
|
if_description: Mapped[str] = mapped_column(String(512), default="")
|
|
bw_bps: Mapped[int] = mapped_column(BigInteger, default=0)
|
|
status: Mapped[str] = mapped_column(String(32), default="active", index=True) # active|disabled|retired
|
|
last_error: Mapped[str] = mapped_column(String(1024), default="")
|
|
last_sample_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive)
|
|
|
|
|
|
@event.listens_for(PortTrafficSeries, "before_insert")
|
|
@event.listens_for(PortTrafficSeries, "before_update")
|
|
def _sync_series_task_id(mapper, connection, target) -> None: # noqa: ANN001
|
|
_mirror_device_task_id(target)
|
|
|
|
|
|
@event.listens_for(PortTrafficTarget, "before_insert")
|
|
@event.listens_for(PortTrafficTarget, "before_update")
|
|
def _sync_target_task_id(mapper, connection, target) -> None: # noqa: ANN001
|
|
_mirror_device_task_id(target)
|
|
|
|
|
|
class PortTrafficSample(Base):
|
|
"""Time-series sample for a monitored interface."""
|
|
|
|
__tablename__ = "port_traffic_sample"
|
|
__table_args__ = (UniqueConstraint("target_row_id", "ts", name="uq_port_traffic_sample_ts"),)
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True, default=lambda: uuid4().hex)
|
|
target_row_id: Mapped[str] = mapped_column(String(64), index=True)
|
|
series_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
ts: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive, index=True)
|
|
in_bps: Mapped[float] = mapped_column(Float, default=0.0)
|
|
out_bps: Mapped[float] = mapped_column(Float, default=0.0)
|
|
in_util_pct: Mapped[float] = mapped_column(Float, default=0.0)
|
|
out_util_pct: Mapped[float] = mapped_column(Float, default=0.0)
|
|
bw_bps: Mapped[int] = mapped_column(BigInteger, default=0)
|
|
rate_period_sec: Mapped[int] = mapped_column(Integer, default=0)
|
|
raw_ok: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
message: Mapped[str] = mapped_column(String(512), default="")
|
|
|
|
|
|
class PortTrafficEvent(Base):
|
|
"""Collect / ops log line for a monitored device (and optional interface)."""
|
|
|
|
__tablename__ = "port_traffic_event"
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True, default=lambda: uuid4().hex)
|
|
device_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
target_row_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
ifname: Mapped[str] = mapped_column(String(128), default="")
|
|
level: Mapped[str] = mapped_column(String(16), default="error", index=True) # info|warn|error
|
|
message: Mapped[str] = mapped_column(Text, default="")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive, index=True)
|
|
|
|
|
|
class PortTrafficBoard(Base):
|
|
"""Named multi-panel traffic ops wall (cutover / duty view)."""
|
|
|
|
__tablename__ = "port_traffic_board"
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True, default=lambda: uuid4().hex)
|
|
name: Mapped[str] = mapped_column(String(256), default="", index=True)
|
|
remark: Mapped[str] = mapped_column(String(1024), default="")
|
|
cols: Mapped[int] = mapped_column(Integer, default=2)
|
|
created_by: Mapped[str] = mapped_column(String(64), default="")
|
|
updated_by: Mapped[str] = mapped_column(String(64), default="")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive, index=True)
|
|
|
|
|
|
class PortTrafficPanel(Base):
|
|
"""One chart cell on a traffic board."""
|
|
|
|
__tablename__ = "port_traffic_panel"
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True, default=lambda: uuid4().hex)
|
|
board_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
title: Mapped[str] = mapped_column(String(256), default="")
|
|
target_id: Mapped[str] = mapped_column(String(64), default="", index=True)
|
|
range_hours: Mapped[int] = mapped_column(Integer, default=24)
|
|
baseline: Mapped[str] = mapped_column(String(16), default="off") # off|day|week|shift|custom
|
|
offset_hours: Mapped[int] = mapped_column(Integer, default=0)
|
|
ahead_hours: Mapped[int] = mapped_column(Integer, default=1) # extend chart past "now" for baseline peek
|
|
baseline_target_id: Mapped[str] = mapped_column(String(64), default="")
|
|
y_mode: Mapped[str] = mapped_column(String(16), default="auto") # auto|current|util
|
|
ord: Mapped[int] = mapped_column(Integer, default=0)
|
|
col_span: Mapped[int] = mapped_column(Integer, default=1)
|
|
row_span: Mapped[int] = mapped_column(Integer, default=1)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=utcnow_naive)
|