diff --git a/app/assets/database/queries/__init__.py b/app/assets/database/queries/__init__.py index ea526e549..a44ed04b9 100644 --- a/app/assets/database/queries/__init__.py +++ b/app/assets/database/queries/__init__.py @@ -25,7 +25,7 @@ from app.assets.database.queries.asset_info import ( from app.assets.database.queries.cache_state import ( CacheStateRow, bulk_insert_cache_states_ignore_conflicts, - bulk_set_needs_verify, + bulk_update_needs_verify, delete_assets_by_ids, delete_cache_states_by_ids, delete_orphaned_seed_asset, @@ -65,7 +65,7 @@ __all__ = [ "bulk_insert_assets", "bulk_insert_cache_states_ignore_conflicts", "bulk_insert_tags_and_meta", - "bulk_set_needs_verify", + "bulk_update_needs_verify", "delete_asset_info_by_id", "delete_assets_by_ids", "delete_cache_states_by_ids", diff --git a/app/assets/database/queries/cache_state.py b/app/assets/database/queries/cache_state.py index 9ecd8d7f4..39d79e16c 100644 --- a/app/assets/database/queries/cache_state.py +++ b/app/assets/database/queries/cache_state.py @@ -246,7 +246,7 @@ def get_cache_states_for_prefixes( ] -def bulk_set_needs_verify(session: Session, state_ids: list[int], value: bool) -> int: +def bulk_update_needs_verify(session: Session, state_ids: list[int], value: bool) -> int: """Set needs_verify flag for multiple cache states. Returns: Number of rows updated diff --git a/app/assets/scanner.py b/app/assets/scanner.py index c6b68dced..e0d5202f1 100644 --- a/app/assets/scanner.py +++ b/app/assets/scanner.py @@ -7,7 +7,7 @@ from typing import Literal, TypedDict import folder_paths from app.assets.database.queries import ( add_missing_tag_for_asset_id, - bulk_set_needs_verify, + bulk_update_needs_verify, delete_cache_states_by_ids, delete_orphaned_seed_asset, ensure_tags_exist, @@ -195,8 +195,8 @@ def sync_cache_states_with_filesystem( survivors.add(os.path.abspath(s["fp"])) delete_cache_states_by_ids(session, stale_state_ids) - bulk_set_needs_verify(session, to_set_verify, value=True) - bulk_set_needs_verify(session, to_clear_verify, value=False) + bulk_update_needs_verify(session, to_set_verify, value=True) + bulk_update_needs_verify(session, to_clear_verify, value=False) return survivors if collect_existing_paths else None diff --git a/tests-unit/assets_test/queries/test_cache_state.py b/tests-unit/assets_test/queries/test_cache_state.py index 46603195f..7af81db38 100644 --- a/tests-unit/assets_test/queries/test_cache_state.py +++ b/tests-unit/assets_test/queries/test_cache_state.py @@ -9,7 +9,7 @@ from app.assets.database.queries import ( get_unreferenced_unhashed_asset_ids, delete_assets_by_ids, get_cache_states_for_prefixes, - bulk_set_needs_verify, + bulk_update_needs_verify, delete_cache_states_by_ids, delete_orphaned_seed_asset, bulk_insert_cache_states_ignore_conflicts, @@ -336,7 +336,7 @@ class TestBulkSetNeedsVerify: state2 = _make_cache_state(session, asset, "/path2.bin", needs_verify=False) session.commit() - updated = bulk_set_needs_verify(session, [state1.id, state2.id], True) + updated = bulk_update_needs_verify(session, [state1.id, state2.id], True) session.commit() assert updated == 2 @@ -346,7 +346,7 @@ class TestBulkSetNeedsVerify: assert state2.needs_verify is True def test_empty_list_updates_nothing(self, session: Session): - updated = bulk_set_needs_verify(session, [], True) + updated = bulk_update_needs_verify(session, [], True) assert updated == 0