Testing text2img command (will be replaced, only for tests); added exception handling, queue and database table with generated images
This commit is contained in:
31
bot/modules/api/status.py
Normal file
31
bot/modules/api/status.py
Normal file
@@ -0,0 +1,31 @@
|
||||
from bot.db import db, DBTables
|
||||
import aiohttp
|
||||
import asyncio
|
||||
import time
|
||||
|
||||
|
||||
async def job_exists(endpoint):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
r = await session.get(
|
||||
endpoint + "/sdapi/v1/progress",
|
||||
json={
|
||||
"skip_current_image": True,
|
||||
}
|
||||
)
|
||||
if r.status != 200:
|
||||
return None
|
||||
return (await r.json()).get('state').get('job_count') > 0
|
||||
|
||||
|
||||
async def wait_for_status(ignore_exceptions: bool = False):
|
||||
endpoint = db[DBTables.config].get('endpoint')
|
||||
try:
|
||||
while await job_exists(endpoint):
|
||||
while db[DBTables.cooldown].get('_last_time_status_checked', 0) + 5 > time.time():
|
||||
await asyncio.sleep(5)
|
||||
db[DBTables.cooldown]['_last_time_status_checked'] = time.time()
|
||||
return
|
||||
except Exception as e:
|
||||
if not ignore_exceptions:
|
||||
raise e
|
||||
return
|
||||
Reference in New Issue
Block a user