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): tidy dependencies for migrations

Browse Source
This commit is contained in:
psychedelicious 2023-12-12 13:09:09 +11:00
parent 0cf7fe43af
commit a69f518c76
2 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/services/shared/sqlite_migrator/migrations/migration_2.py
View File

@ -140,7 +140,7 @@ migration_2 = Migration(
from_version=1,
to_version=2,
migrate_callback=migrate_callback,
dependencies={"image_files": image_files_dependency, "logger": logger_dependency},
dependencies={image_files_dependency.name: image_files_dependency, logger_dependency.name: logger_dependency},
)
"""
Database version 2.

18
invokeai/app/services/shared/sqlite_migrator/sqlite_migrator_common.py
View File

@ -33,10 +33,13 @@ class MigrationDependency:
self.dependency_type = dependency_type
self.value = None
def set(self, value: Any) -> None:
"""Sets the value of the dependency."""
def set_value(self, value: Any) -> None:
"""
Sets the value of the dependency.
If the value is not of the correct type, a TypeError is raised.
"""
if not isinstance(value, self.dependency_type):
raise ValueError(f"Dependency {self.name} must be of type {self.dependency_type}")
raise TypeError(f"Dependency {self.name} must be of type {self.dependency_type}")
self.value = value
@ -96,13 +99,16 @@ class Migration(BaseModel):
"""Provides a dependency for this migration."""
if name not in self.dependencies:
raise ValueError(f"{name} of type {type(value)} is not a dependency of this migration")
self.dependencies[name].set(value)
self.dependencies[name].set_value(value)
def run(self, cursor: sqlite3.Cursor) -> None:
"""Runs the migration."""
"""
Runs the migration.
If any dependencies are missing, a MigrationError is raised.
"""
missing_dependencies = [d.name for d in self.dependencies.values() if d.value is None]
if missing_dependencies:
raise ValueError(f"Missing migration dependencies: {', '.join(missing_dependencies)}")
raise MigrationError(f"Missing migration dependencies: {', '.join(missing_dependencies)}")
self.migrate_callback = partial(self.migrate_callback, **{d.name: d.value for d in self.dependencies.values()})
self.migrate_callback(cursor=cursor)