mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-15 15:45:27 +03:00
20 lines
628 B
Python
20 lines
628 B
Python
import sqlite_vec
|
|
from playhouse.sqliteq import SqliteQueueDatabase
|
|
|
|
|
|
class SqliteVecQueueDatabase(SqliteQueueDatabase):
|
|
def __init__(self, *args, load_vec_extension=False, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.load_vec_extension = load_vec_extension
|
|
|
|
def _connect(self, *args, **kwargs):
|
|
conn = super()._connect(*args, **kwargs)
|
|
if self.load_vec_extension:
|
|
self._load_vec_extension(conn)
|
|
return conn
|
|
|
|
def _load_vec_extension(self, conn):
|
|
conn.enable_load_extension(True)
|
|
sqlite_vec.load(conn)
|
|
conn.enable_load_extension(False)
|