Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

use sqlitequeuedb

Browse Source
This commit is contained in:
Blake Blackshear 2021-01-24 06:53:01 -06:00
parent 600477c487
commit 630c2ee6f6
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/app.py
View File

@ -10,6 +10,7 @@ import signal
import yaml import yaml
from peewee_migrate import Router from peewee_migrate import Router
from playhouse.sqlite_ext import SqliteExtDatabase from playhouse.sqlite_ext import SqliteExtDatabase
from playhouse.sqliteq import SqliteQueueDatabase
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.const import RECORD_DIR, CLIPS_DIR, CACHE_DIR from frigate.const import RECORD_DIR, CLIPS_DIR, CACHE_DIR
@ -117,13 +118,16 @@ class FrigateApp():
self.detected_frames_queue = mp.Queue(maxsize=len(self.config.cameras.keys())*2) self.detected_frames_queue = mp.Queue(maxsize=len(self.config.cameras.keys())*2)
def init_database(self): def init_database(self):
self.db = SqliteExtDatabase(self.config.database.path) migrate_db = SqliteExtDatabase(self.config.database.path)
# Run migrations # Run migrations
del(logging.getLogger('peewee_migrate').handlers[:]) del(logging.getLogger('peewee_migrate').handlers[:])
router = Router(self.db) router = Router(migrate_db)
router.run() router.run()
migrate_db.close()
self.db = SqliteQueueDatabase(self.config.database.path)
models = [Event] models = [Event]
self.db.bind(models) self.db.bind(models)