wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-06 09:47:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Rename bulk_set_needs_verify to bulk_update_needs_verify for readability

Browse Source 
Amp-Thread-ID: https://ampcode.com/threads/T-019c3167-e8f1-7409-904b-5fc0edaeef37
Co-authored-by: Amp <amp@ampcode.com>
This commit is contained in:
Luke Mino-Altherr 2026-02-05 21:25:58 -08:00
parent a51bbd0b25
commit 82885b40b9
4 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/assets/database/queries/__init__.py
View File

@ -25,7 +25,7 @@ from app.assets.database.queries.asset_info import (
from app.assets.database.queries.cache_state import ( from app.assets.database.queries.cache_state import (
CacheStateRow, CacheStateRow,
bulk_insert_cache_states_ignore_conflicts, bulk_insert_cache_states_ignore_conflicts,
bulk_set_needs_verify, bulk_update_needs_verify,
delete_assets_by_ids, delete_assets_by_ids,
delete_cache_states_by_ids, delete_cache_states_by_ids,
delete_orphaned_seed_asset, delete_orphaned_seed_asset,
@ -65,7 +65,7 @@ __all__ = [
"bulk_insert_assets", "bulk_insert_assets",
"bulk_insert_cache_states_ignore_conflicts", "bulk_insert_cache_states_ignore_conflicts",
"bulk_insert_tags_and_meta", "bulk_insert_tags_and_meta",
"bulk_set_needs_verify", "bulk_update_needs_verify",
"delete_asset_info_by_id", "delete_asset_info_by_id",
"delete_assets_by_ids", "delete_assets_by_ids",
"delete_cache_states_by_ids", "delete_cache_states_by_ids",

2
app/assets/database/queries/cache_state.py
View File

@ -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. """Set needs_verify flag for multiple cache states.
Returns: Number of rows updated Returns: Number of rows updated

6
app/assets/scanner.py
View File

@ -7,7 +7,7 @@ from typing import Literal, TypedDict
import folder_paths import folder_paths
from app.assets.database.queries import ( from app.assets.database.queries import (
add_missing_tag_for_asset_id, add_missing_tag_for_asset_id,
bulk_set_needs_verify, bulk_update_needs_verify,
delete_cache_states_by_ids, delete_cache_states_by_ids,
delete_orphaned_seed_asset, delete_orphaned_seed_asset,
ensure_tags_exist, ensure_tags_exist,
@ -195,8 +195,8 @@ def sync_cache_states_with_filesystem(
survivors.add(os.path.abspath(s["fp"])) survivors.add(os.path.abspath(s["fp"]))
delete_cache_states_by_ids(session, stale_state_ids) delete_cache_states_by_ids(session, stale_state_ids)
bulk_set_needs_verify(session, to_set_verify, value=True) bulk_update_needs_verify(session, to_set_verify, value=True)
bulk_set_needs_verify(session, to_clear_verify, value=False) bulk_update_needs_verify(session, to_clear_verify, value=False)
return survivors if collect_existing_paths else None return survivors if collect_existing_paths else None

6
tests-unit/assets_test/queries/test_cache_state.py
View File

@ -9,7 +9,7 @@ from app.assets.database.queries import (
get_unreferenced_unhashed_asset_ids, get_unreferenced_unhashed_asset_ids,
delete_assets_by_ids, delete_assets_by_ids,
get_cache_states_for_prefixes, get_cache_states_for_prefixes,
bulk_set_needs_verify, bulk_update_needs_verify,
delete_cache_states_by_ids, delete_cache_states_by_ids,
delete_orphaned_seed_asset, delete_orphaned_seed_asset,
bulk_insert_cache_states_ignore_conflicts, bulk_insert_cache_states_ignore_conflicts,
@ -336,7 +336,7 @@ class TestBulkSetNeedsVerify:
state2 = _make_cache_state(session, asset, "/path2.bin", needs_verify=False) state2 = _make_cache_state(session, asset, "/path2.bin", needs_verify=False)
session.commit() 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() session.commit()
assert updated == 2 assert updated == 2
@ -346,7 +346,7 @@ class TestBulkSetNeedsVerify:
assert state2.needs_verify is True assert state2.needs_verify is True
def test_empty_list_updates_nothing(self, session: Session): 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 assert updated == 0