Skip to content

Commit 88c68e8

Browse files
tidy(app): workflow records get_many
1 parent 5073c7d commit 88c68e8

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

invokeai/app/services/workflow_records/workflow_records_sqlite.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -128,12 +128,15 @@ def get_many(
128128
# sanitize!
129129
assert order_by in WorkflowRecordOrderBy
130130
assert direction in SQLiteDirection
131+
132+
main_params: list[int | str] = []
133+
count_params: list[int | str] = []
134+
131135
if categories:
132136
assert all(c in WorkflowCategory for c in categories)
133-
count_query = "SELECT COUNT(*) FROM workflow_library WHERE category IN ({})".format(
134-
", ".join("?" for _ in categories)
135-
)
136-
main_query = """
137+
question_marks = ", ".join("?" for _ in categories)
138+
count_query = f"SELECT COUNT(*) FROM workflow_library WHERE category IN ({question_marks})"
139+
main_query = f"""
137140
SELECT
138141
workflow_id,
139142
category,
@@ -143,10 +146,10 @@ def get_many(
143146
updated_at,
144147
opened_at
145148
FROM workflow_library
146-
WHERE category IN ({})
147-
""".format(", ".join("?" for _ in categories))
148-
main_params: list[int | str] = [category.value for category in categories]
149-
count_params: list[int | str] = [category.value for category in categories]
149+
WHERE category IN ({question_marks})
150+
"""
151+
main_params.extend([category.value for category in categories])
152+
count_params.extend([category.value for category in categories])
150153
else:
151154
count_query = "SELECT COUNT(*) FROM workflow_library"
152155
main_query = """
@@ -160,8 +163,6 @@ def get_many(
160163
opened_at
161164
FROM workflow_library
162165
"""
163-
main_params: list[int | str] = []
164-
count_params: list[int | str] = []
165166

166167
stripped_query = query.strip() if query else None
167168
if stripped_query:

0 commit comments

Comments
 (0)