administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-04 18:25:22 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add other fields

Browse Source
This commit is contained in:
Nick Mowen 2023-04-20 15:59:00 -06:00
parent 98b154f5ec
commit b2a8f8330a
3 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/events.py
View File

@ -3,15 +3,14 @@ import logging
import os
import queue
import threading
import time
from pathlib import Path
from peewee import fn
from frigate.config import EventsConfig, FrigateConfig, RecordConfig
from frigate.config import EventsConfig, FrigateConfig
from frigate.const import CLIPS_DIR
from frigate.models import Event
from frigate.timeline import InputTypeEnum
from frigate.timeline import TimelineSourceEnum
from frigate.types import CameraMetricsTypes
from multiprocessing.queues import Queue
@ -79,7 +78,7 @@ class EventProcessor(threading.Thread):
self.timeline_queue.put(
(
camera,
InputTypeEnum.tracked_object,
TimelineSourceEnum.tracked_object,
event_type,
self.events_in_process.get(event_data["id"]),
event_data,

6
frigate/timeline.py
View File

@ -14,9 +14,9 @@ from multiprocessing.synchronize import Event as MpEvent
logger = logging.getLogger(__name__)
class InputTypeEnum(str, Enum):
class TimelineSourceEnum(str, Enum):
# api = "api"
# audio = "audio"
# external = "external"
tracked_object = "tracked_object"
@ -42,7 +42,7 @@ class TimelineProcessor(threading.Thread):
except queue.Empty:
continue
if input_type == InputTypeEnum.object:
if input_type == TimelineSourceEnum.object:
self.handle_object_detection(
camera, event_type, prev_event_data, event_data
)

2
migrations/013_create_timeline_table.py
View File

@ -37,7 +37,7 @@ SQL = pw.SQL
def migrate(migrator, database, fake=False, **kwargs):
migrator.sql(
'CREATE TABLE IF NOT EXISTS "timeline" ("timestamp" DATETIME NOT NULL, "camera" VARCHAR(20) NOT NULL, "source" VARCHAR(20) NOT NULL, "source_id" VARCHAR(30) NOT NULL, "class_type" VARCHAR(50) NOT NULL, "data" JSON)'
'CREATE TABLE IF NOT EXISTS "timeline" ("timestamp" DATETIME NOT NULL, "camera" VARCHAR(20) NOT NULL, "source" VARCHAR(20) NOT NULL, "source_id" VARCHAR(30), "class_type" VARCHAR(50) NOT NULL, "data" JSON)'
)
migrator.sql('CREATE INDEX IF NOT EXISTS "timeline_camera" ON "timeline" ("camera")')
migrator.sql('CREATE INDEX IF NOT EXISTS "timeline_source" ON "timeline" ("source")')