nodes/ComfyAssets-ComfyUI_PromptManager
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(api): exclude protected prompts from pruning

Browse Source
This commit is contained in:
Vito Sansevero
2025-12-09 09:29:34 -08:00
parent 2e3079c01b
commit da239bb7e1
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py/api.py
+8 -4
View File
@@ -3205,15 +3205,19 @@ class PromptManagerAPI:
}
# Prune orphaned prompts (prompts with no linked images)
# Note: Prompts with the __protected__ tag are excluded from orphan cleanup
# This allows users to add prompts manually without images that won't be deleted
if 'prune_orphaned_prompts' in operations:
try:
with self.db.model.get_connection() as conn:
# Find prompts that have no images linked to them
# Exclude prompts with the __protected__ tag
cursor = conn.execute("""
SELECT p.id
FROM prompts p
LEFT JOIN generated_images gi ON p.id = gi.prompt_id
SELECT p.id
FROM prompts p
LEFT JOIN generated_images gi ON p.id = gi.prompt_id
WHERE gi.prompt_id IS NULL
AND (p.tags IS NULL OR p.tags NOT LIKE '%"__protected__"%')
""")
orphaned_prompts = cursor.fetchall()
orphaned_count = len(orphaned_prompts)
@@ -3231,7 +3235,7 @@ class PromptManagerAPI:
results['prune_orphaned_prompts'] = {
'success': True,
'removed_count': removed_count,
'message': f'Removed {removed_count} orphaned prompts (prompts with no linked images)'
'message': f'Removed {removed_count} orphaned prompts (prompts with no linked images, excluding protected prompts)'
}
except Exception as e:
results['prune_orphaned_prompts'] = {