comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-18 22:20:03 +00:00
Code Issues 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 9222ff6d81
commit 58ddf46c0a
4 changed files with 9 additions and 9 deletions
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 (
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",

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.
Returns: Number of rows updated

6
app/assets/scanner.py
View File

@@ -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

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,
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