[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]
|
return result[0]
|
||||||
|
|
||||||
def get_random_jobs(self, limit_count=0) -> list:
|
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
|
# execute the query and return the results
|
||||||
c = self.get_cursor()
|
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 = []
|
jobs = []
|
||||||
for row in rows:
|
for row in rows:
|
||||||
|
|
|
||||||
|
|
@ -9,7 +9,6 @@ from utilities.config import Config
|
||||||
from utilities.logger import DummyLogger
|
from utilities.logger import DummyLogger
|
||||||
from utilities.images import image_to_base64
|
from utilities.images import image_to_base64
|
||||||
from utilities.images import load_image
|
from utilities.images import load_image
|
||||||
from utilities.images import save_image
|
|
||||||
from utilities.images import base64_to_image
|
from utilities.images import base64_to_image
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue