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

fix: use system-installed Black in pre-commit to match CI

Browse Source 
Avoids version mismatch between pre-commit's pinned Black and CI's
latest Black. All hooks now use local system tools for consistency.
This commit is contained in:
Vito Sansevero
2026-02-08 05:30:31 -08:00
parent be205f3dcc
commit bb34b9cf3b
4 changed files with 44 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files
database/models.py
+22 -44
View File
@@ -68,8 +68,7 @@ class PromptModel:
- prompts table: Stores prompt text and metadata
- generated_images table: Links generated images to their source prompts
"""
conn.execute(
"""
conn.execute("""
CREATE TABLE IF NOT EXISTS prompts (
id INTEGER PRIMARY KEY AUTOINCREMENT,
text TEXT NOT NULL,
@@ -81,12 +80,10 @@ class PromptModel:
notes TEXT,
hash TEXT UNIQUE
)
"""
)
""")
# Create images table for gallery functionality
conn.execute(
"""
conn.execute("""
CREATE TABLE IF NOT EXISTS generated_images (
id INTEGER PRIMARY KEY AUTOINCREMENT,
prompt_id INTEGER NOT NULL,
@@ -103,20 +100,16 @@ class PromptModel:
FOREIGN KEY (prompt_id) REFERENCES prompts(id) ON DELETE CASCADE,
UNIQUE(prompt_id, filename)
)
"""
)
""")
# Create normalized tag tables (junction table pattern)
conn.execute(
"""
conn.execute("""
CREATE TABLE IF NOT EXISTS tags (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL UNIQUE
)
"""
)
conn.execute(
"""
""")
conn.execute("""
CREATE TABLE IF NOT EXISTS prompt_tags (
prompt_id INTEGER NOT NULL,
tag_id INTEGER NOT NULL,
@@ -124,8 +117,7 @@ class PromptModel:
FOREIGN KEY (prompt_id) REFERENCES prompts(id) ON DELETE CASCADE,
FOREIGN KEY (tag_id) REFERENCES tags(id) ON DELETE CASCADE
)
"""
)
""")
# Add unique constraint to existing databases (migration)
self._migrate_add_unique_constraint(conn)
@@ -208,8 +200,7 @@ class PromptModel:
self.logger.info("Migrating database: removing workflow_name column")
# Create new table without workflow_name
conn.execute(
"""
conn.execute("""
CREATE TABLE prompts_new (
id INTEGER PRIMARY KEY AUTOINCREMENT,
text TEXT NOT NULL,
@@ -221,17 +212,14 @@ class PromptModel:
notes TEXT,
hash TEXT UNIQUE
)
"""
)
""")
# Copy data from old table to new table
conn.execute(
"""
conn.execute("""
INSERT INTO prompts_new (id, text, created_at, updated_at, category, tags, rating, notes, hash)
SELECT id, text, created_at, updated_at, category, tags, rating, notes, hash
FROM prompts
"""
)
""")
# Drop old table and rename new one
conn.execute("DROP TABLE prompts")
@@ -264,8 +252,7 @@ class PromptModel:
)
# Create new table with correct types
conn.execute(
"""
conn.execute("""
CREATE TABLE generated_images_new (
id INTEGER PRIMARY KEY AUTOINCREMENT,
prompt_id INTEGER NOT NULL,
@@ -281,12 +268,10 @@ class PromptModel:
parameters TEXT,
FOREIGN KEY (prompt_id) REFERENCES prompts(id) ON DELETE CASCADE
)
"""
)
""")
# Copy data, converting prompt_id from TEXT to INTEGER
conn.execute(
"""
conn.execute("""
INSERT INTO generated_images_new
(id, prompt_id, image_path, filename, generation_time, file_size,
width, height, format, workflow_data, prompt_metadata, parameters)
@@ -295,8 +280,7 @@ class PromptModel:
FROM generated_images
WHERE prompt_id != '' AND prompt_id IS NOT NULL
AND CAST(prompt_id AS INTEGER) IN (SELECT id FROM prompts)
"""
)
""")
# Drop old table and rename new one
conn.execute("DROP TABLE generated_images")
@@ -345,15 +329,13 @@ class PromptModel:
)
# First, remove duplicates keeping only the most recent (highest id)
conn.execute(
"""
conn.execute("""
DELETE FROM generated_images
WHERE id NOT IN (
SELECT MAX(id) FROM generated_images
GROUP BY prompt_id, filename
)
"""
)
""")
duplicates_removed = conn.total_changes
if duplicates_removed > 0:
@@ -362,8 +344,7 @@ class PromptModel:
)
# Create new table with UNIQUE constraint
conn.execute(
"""
conn.execute("""
CREATE TABLE IF NOT EXISTS generated_images_new (
id INTEGER PRIMARY KEY AUTOINCREMENT,
prompt_id INTEGER NOT NULL,
@@ -380,20 +361,17 @@ class PromptModel:
FOREIGN KEY (prompt_id) REFERENCES prompts(id) ON DELETE CASCADE,
UNIQUE(prompt_id, filename)
)
"""
)
""")
# Copy data
conn.execute(
"""
conn.execute("""
INSERT INTO generated_images_new
(id, prompt_id, image_path, filename, generation_time, file_size,
width, height, format, workflow_data, prompt_metadata, parameters)
SELECT id, prompt_id, image_path, filename, generation_time, file_size,
width, height, format, workflow_data, prompt_metadata, parameters
FROM generated_images
"""
)
""")
# Drop old table and rename new one
conn.execute("DROP TABLE generated_images")