Bläddra i källkod

use sqlitequeuedb

Blake Blackshear 4 år sedan
förälder
incheckning
630c2ee6f6
1 ändrade filer med 6 tillägg och 2 borttagningar
  1. 6 2
      frigate/app.py

+ 6 - 2
frigate/app.py

@@ -10,6 +10,7 @@ import signal
 import yaml
 from peewee_migrate import Router
 from playhouse.sqlite_ext import SqliteExtDatabase
+from playhouse.sqliteq import SqliteQueueDatabase
 
 from frigate.config import FrigateConfig
 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)
 
     def init_database(self):
-        self.db = SqliteExtDatabase(self.config.database.path)
+        migrate_db = SqliteExtDatabase(self.config.database.path)
 
         # Run migrations
         del(logging.getLogger('peewee_migrate').handlers[:])
-        router = Router(self.db)
+        router = Router(migrate_db)
         router.run()
 
+        migrate_db.close()
+
+        self.db = SqliteQueueDatabase(self.config.database.path)
         models = [Event]
         self.db.bind(models)