tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update set_model.py

Browse Source 
Whitelisted set_model
Chanched cooldown to 5 minutes
This commit is contained in:
vasmarfas
2023-05-03 21:39:16 +03:00
committed by GitHub
parent 8ec4ddcc0a
commit 1653e184a8
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bot/handlers/txt2img/set_model.py
View File

@@ -7,8 +7,11 @@ from bot.utils.errorable_command import wrap_exception
@wrap_exception() @wrap_exception()
@throttle(cooldown=30*60, admin_ids=db[DBTables.config].get('admins'), by_id=False) @throttle(cooldown=5*60, admin_ids=db[DBTables.config].get('admins'), by_id=False)
async def set_model_command(message: types.Message): async def set_model_command(message: types.Message):
if (message.chat.id not in db[DBTables.config]['whitelist'] and message.from_id not in db[DBTables.config]['whitelist']):
await message.reply('❌You are not on the white list, access denied. Contact admin @kilisauros for details')
return
models = await get_models() models = await get_models()
if models is not None and len(models) > 0: if models is not None and len(models) > 0:
db[DBTables.config]['models'] = models db[DBTables.config]['models'] = models