[FE] fix issue on fetching random jobs
This commit is contained in:
parent
fa306254c9
commit
9b7d2bd89d
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue