[FE] fix issue on fetching random jobs

This commit is contained in:
HappyZ 2023-05-27 23:35:45 -07:00
parent fa306254c9
commit 9b7d2bd89d
2 changed files with 12 additions and 3 deletions

View File

@ -129,11 +129,21 @@ class Database:
return result[0]
def get_random_jobs(self, limit_count=0) -> list:
query = f"SELECT {', '.join(ANONYMOUS_KEYS)} FROM {HISTORY_TABLE_NAME} WHERE {KEY_JOB_STATUS} = ? AND {KEY_IS_PRIVATE} = ? AND rowid IN (SELECT rowid FROM {HISTORY_TABLE_NAME} ORDER BY RANDOM() LIMIT ?) AND {KEY_JOB_TYPE} IN ({VALUE_JOB_IMG2IMG, VALUE_JOB_INPAINTING, VALUE_JOB_RESTORATION})"
query = f"SELECT {', '.join(ANONYMOUS_KEYS)} FROM {HISTORY_TABLE_NAME} WHERE rowid IN (SELECT rowid FROM {HISTORY_TABLE_NAME} WHERE {KEY_JOB_STATUS} = ? AND {KEY_IS_PRIVATE} = ? AND {KEY_JOB_TYPE} IN (?, ?, ?) ORDER BY RANDOM() LIMIT ?)"
# execute the query and return the results
c = self.get_cursor()
rows = c.execute(query, (VALUE_JOB_DONE, False, limit_count)).fetchall()
rows = c.execute(
query,
(
VALUE_JOB_DONE,
False,
VALUE_JOB_IMG2IMG,
VALUE_JOB_INPAINTING,
VALUE_JOB_TXT2IMG,
limit_count,
),
).fetchall()
jobs = []
for row in rows:

View File

@ -9,7 +9,6 @@ from utilities.config import Config
from utilities.logger import DummyLogger
from utilities.images import image_to_base64
from utilities.images import load_image
from utilities.images import save_image
from utilities.images import base64_to_image