Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(db): move migrator out of SqliteDatabase

Browse Source
This commit is contained in:
psychedelicious
2023-12-10 15:26:38 +11:00
parent a2dc780188
commit c382329e8c
2 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/services/shared/sqlite/sqlite_database.py
View File

@ -4,10 +4,7 @@ from logging import Logger
from pathlib import Path
from invokeai.app.services.config import InvokeAIAppConfig
from invokeai.app.services.shared.sqlite.migrations.migration_1 import migration_1
from invokeai.app.services.shared.sqlite.migrations.migration_2 import migration_2
from invokeai.app.services.shared.sqlite.sqlite_common import sqlite_memory
from invokeai.app.services.shared.sqlite.sqlite_migrator import SQLiteMigrator
class SqliteDatabase:
@ -33,11 +30,6 @@ class SqliteDatabase:
self.conn.execute("PRAGMA foreign_keys = ON;")
migrator = SQLiteMigrator(conn=self.conn, database=self.database, lock=self.lock, logger=self._logger)
migrator.register_migration(migration_1)
migrator.register_migration(migration_2)
migrator.run_migrations()
def clean(self) -> None:
with self.lock:
try: