feat(edge): Event model + severity mapping (§7.3 payload)
This commit is contained in:
parent
02ca524bf6
commit
1020304dca
2 changed files with 86 additions and 0 deletions
64
edge/events.py
Normal file
64
edge/events.py
Normal file
|
|
@ -0,0 +1,64 @@
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
|
class EventType(str, Enum):
|
||||||
|
FALL = "fall"
|
||||||
|
NO_MOTION = "no_motion"
|
||||||
|
IMMOBILITY = "immobility"
|
||||||
|
GHOST = "ghost"
|
||||||
|
ZONE_WARNING = "zone_warning"
|
||||||
|
ZONE_RESTRICTED = "zone_restricted"
|
||||||
|
PINCH = "pinch"
|
||||||
|
MONITORING_DEGRADED = "monitoring_degraded"
|
||||||
|
DEVICE_FAULT = "device_fault"
|
||||||
|
|
||||||
|
|
||||||
|
class Severity(str, Enum):
|
||||||
|
CRITICAL = "critical"
|
||||||
|
HIGH = "high"
|
||||||
|
WARNING = "warning"
|
||||||
|
INFO = "info"
|
||||||
|
|
||||||
|
|
||||||
|
_SEVERITY = {
|
||||||
|
EventType.NO_MOTION: Severity.CRITICAL,
|
||||||
|
EventType.GHOST: Severity.CRITICAL,
|
||||||
|
EventType.FALL: Severity.HIGH,
|
||||||
|
EventType.ZONE_RESTRICTED: Severity.HIGH,
|
||||||
|
EventType.PINCH: Severity.HIGH,
|
||||||
|
EventType.IMMOBILITY: Severity.WARNING,
|
||||||
|
EventType.ZONE_WARNING: Severity.WARNING,
|
||||||
|
EventType.MONITORING_DEGRADED: Severity.WARNING,
|
||||||
|
EventType.DEVICE_FAULT: Severity.WARNING,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def severity_for(event_type: EventType) -> Severity:
|
||||||
|
return _SEVERITY.get(event_type, Severity.INFO)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True)
|
||||||
|
class Event:
|
||||||
|
event_type: EventType
|
||||||
|
severity: Severity
|
||||||
|
camera_id: str
|
||||||
|
track_id: int | None
|
||||||
|
zone_id: str | None
|
||||||
|
occurred_at: float
|
||||||
|
duration_s: float
|
||||||
|
metrics: dict
|
||||||
|
spatial: tuple | None # (x, y) 유령사고 등 공간 기준
|
||||||
|
|
||||||
|
def to_payload(self, device_id: str) -> dict:
|
||||||
|
return {
|
||||||
|
"device_id": device_id,
|
||||||
|
"camera_id": self.camera_id,
|
||||||
|
"event_type": self.event_type.value,
|
||||||
|
"severity": self.severity.value,
|
||||||
|
"track_id": self.track_id,
|
||||||
|
"zone_id": self.zone_id,
|
||||||
|
"occurred_at": self.occurred_at,
|
||||||
|
"duration_s": round(self.duration_s, 2),
|
||||||
|
"metrics": self.metrics,
|
||||||
|
}
|
||||||
22
tests/test_events.py
Normal file
22
tests/test_events.py
Normal file
|
|
@ -0,0 +1,22 @@
|
||||||
|
from edge.events import EventType, Severity, Event, severity_for
|
||||||
|
|
||||||
|
|
||||||
|
def test_severity_mapping():
|
||||||
|
assert severity_for(EventType.NO_MOTION) == Severity.CRITICAL
|
||||||
|
assert severity_for(EventType.ZONE_RESTRICTED) == Severity.HIGH
|
||||||
|
assert severity_for(EventType.FALL) == Severity.HIGH
|
||||||
|
assert severity_for(EventType.IMMOBILITY) == Severity.WARNING
|
||||||
|
assert severity_for(EventType.ZONE_WARNING) == Severity.WARNING
|
||||||
|
|
||||||
|
|
||||||
|
def test_event_to_payload_shape():
|
||||||
|
e = Event(EventType.FALL, Severity.HIGH, camera_id="cam-01", track_id=7,
|
||||||
|
zone_id=None, occurred_at=1000.0, duration_s=3.2,
|
||||||
|
metrics={"torso_angle": 78.0}, spatial=None)
|
||||||
|
p = e.to_payload(device_id="pi-01")
|
||||||
|
assert p["device_id"] == "pi-01"
|
||||||
|
assert p["event_type"] == "fall"
|
||||||
|
assert p["severity"] == "high"
|
||||||
|
assert p["track_id"] == 7
|
||||||
|
assert p["metrics"]["torso_angle"] == 78.0
|
||||||
|
assert "snapshots" not in p # snapshots는 Alerter가 붙임
|
||||||
Loading…
Reference in a new issue