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

7
invokeai/app/api/dependencies.py
View File

@ -2,6 +2,9 @@
from logging import Logger
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_migrator import SQLiteMigrator
from invokeai.backend.util.logging import InvokeAILogger
from invokeai.version.invokeai_version import __version__
@ -69,6 +72,10 @@ class ApiDependencies:
output_folder = config.output_path
db = SqliteDatabase(config, logger)
migrator = SQLiteMigrator(conn=db.conn, database=db.database, lock=db.lock, logger=logger)
migrator.register_migration(migration_1)
migrator.register_migration(migration_2)
migrator.run_migrations()
configuration = config
logger = logger