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

fix(db): check for graph_executions table before dropping

Browse Source 
This is needed to not fail tests; see comment in code.
This commit is contained in:
psychedelicious 2024-01-31 23:56:04 +11:00
parent 198e8c9d55
commit d532073f5b
1 changed files with 32 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
invokeai/app/services/shared/sqlite_migrator/migrations/migration_5.py
View File

@ -15,19 +15,43 @@ class Migration5Callback:
def _drop_graph_executions(self, cursor: sqlite3.Cursor) -> None:
"""Drops the `graph_executions` table."""
# Check if the table exists first. This shouldn't ever be an issue for users, but it is for tests.
#
# Unlike other database tables, whose schemas are managed by the migrator, `graph_executions`
# is created and managed by an instance of `SqliteItemStorage``. Specifically, it is created in the
# `__init__` of `SqliteItemStorage`.
#
# In tests, we don't always create an instance of `SqliteItemStorage`, so the table may not exist in
# the test memory db fixture.
#
# We _do_, however, run the migrator on the test memory db fixtures. So, we need to check if the
# table exists before dropping it, else unrelated tests will fail.
#
# This song and dance won't be necessary again, because we aren't using `SqliteItemStorage` for anything
# moving forward.
cursor.execute(
"""--sql
SELECT COUNT(*) FROM graph_executions;
"""
)
count = cast(int, cursor.fetchone()[0])
self._logger.info(f"Clearing {count} old sessions from database")
cursor.execute(
"""--sql
DROP TABLE IF EXISTS graph_executions;
SELECT name FROM sqlite_master WHERE type='table' AND name='graph_executions';
"""
)
if cursor.fetchone():
cursor.execute(
"""--sql
SELECT COUNT(*) FROM graph_executions;
"""
)
count = cast(int, cursor.fetchone()[0])
self._logger.info(f"Clearing {count} old sessions from database")
cursor.execute(
"""--sql
DROP TABLE IF EXISTS graph_executions;
"""
)
else:
self._logger.info("No 'graph_executions' table found.")
def build_migration_5(logger: Logger) -> Migration:
"""