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): return backup_db_path, move log stmt to run_migrations

Browse Source
This commit is contained in:
psychedelicious 2023-12-11 11:29:06 +11:00
parent b3d5955bc7
commit 567f107a81
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/app/services/shared/sqlite/sqlite_migrator.py
View File

@ -190,6 +190,7 @@ class SQLiteMigrator:
if self._db_path: if self._db_path:
# We are using a file database. Create a copy of the database to run the migrations on. # We are using a file database. Create a copy of the database to run the migrations on.
temp_db_path = self._create_temp_db(self._db_path) temp_db_path = self._create_temp_db(self._db_path)
self._logger.info(f"Copied database to {temp_db_path} for migration")
temp_db_conn = sqlite3.connect(temp_db_path) temp_db_conn = sqlite3.connect(temp_db_path)
# We have to re-set this because we just created a new connection. # We have to re-set this because we just created a new connection.
if self._log_sql: if self._log_sql:
@ -198,11 +199,12 @@ class SQLiteMigrator:
self._run_migrations(temp_db_cursor) self._run_migrations(temp_db_cursor)
# Close the connections, copy the original database as a backup, and move the temp database to the # Close the connections, copy the original database as a backup, and move the temp database to the
# original database's path. # original database's path.
self._finalize_migration( backup_db_path = self._finalize_migration(
temp_db_conn=temp_db_conn, temp_db_conn=temp_db_conn,
temp_db_path=temp_db_path, temp_db_path=temp_db_path,
original_db_path=self._db_path, original_db_path=self._db_path,
) )
self._logger.info(f"Migration successful. Original DB backed up to {backup_db_path}")
else: else:
# We are using a memory database. No special backup or special handling needed. # We are using a memory database. No special backup or special handling needed.
self._run_migrations(self._cursor) self._run_migrations(self._cursor)
@ -296,14 +298,13 @@ class SQLiteMigrator:
"""Copies the current database to a new file for migration.""" """Copies the current database to a new file for migration."""
temp_db_path = get_temp_db_path(current_db_path) temp_db_path = get_temp_db_path(current_db_path)
shutil.copy2(current_db_path, temp_db_path) shutil.copy2(current_db_path, temp_db_path)
self._logger.info(f"Copied database to {temp_db_path} for migration")
return temp_db_path return temp_db_path
def _finalize_migration(self, temp_db_conn: sqlite3.Connection, temp_db_path: Path, original_db_path: Path) -> None: def _finalize_migration(self, temp_db_conn: sqlite3.Connection, temp_db_path: Path, original_db_path: Path) -> Path:
"""Closes connections, renames the original database as a backup and renames the migrated database to the original name.""" """Closes connections, renames the original database as a backup and renames the migrated database to the original name."""
self._conn.close() self._conn.close()
temp_db_conn.close() temp_db_conn.close()
backup_db_path = get_backup_db_path(original_db_path) backup_db_path = get_backup_db_path(original_db_path)
original_db_path.rename(backup_db_path) original_db_path.rename(backup_db_path)
temp_db_path.rename(original_db_path) temp_db_path.rename(original_db_path)
self._logger.info(f"Migration successful. Original DB backed up to {backup_db_path}") return backup_db_path