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

Key error 'enabled' fixed

Browse Source
This commit is contained in:
BarsTiger
2023-05-19 17:03:06 +03:00
parent 21fd67c15a
commit 9d6428656b
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bot/handlers/admin/on_off.py
View File

@@ -14,4 +14,4 @@ async def on_off_call(message: types.Message, is_command=None):
db[DBTables.config]['enabled'] = False if message.text.lower() == 'off' else True db[DBTables.config]['enabled'] = False if message.text.lower() == 'off' else True
await db[DBTables.config].write() await db[DBTables.config].write()
await message.reply(f"Generation enabled: {'💚 (yes)' if db[DBTables.config]['enabled'] else '💔 (no)'}") await message.reply(f"Generation enabled: {'💚 (yes)' if db[DBTables.config].get('enabled') else '💔 (no)'}")

2
bot/handlers/txt2img/txt2img.py
View File

@@ -16,7 +16,7 @@ from bot.utils.errorable_command import wrap_exception
@throttle(cooldown=30, admin_ids=db[DBTables.config].get('admins')) @throttle(cooldown=30, admin_ids=db[DBTables.config].get('admins'))
async def generate_command(message: types.Message): async def generate_command(message: types.Message):
temp_message = await message.reply("⏳ Enqueued...") temp_message = await message.reply("⏳ Enqueued...")
if not db[DBTables.config]['enabled']: if not db[DBTables.config].get('enabled'):
await message.reply('💔 Generation is disabled by admins now. Try again later') await message.reply('💔 Generation is disabled by admins now. Try again later')
await temp_message.delete() await temp_message.delete()
return return