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

Whitelist works as it should

Browse Source
This commit is contained in:
BarsTiger
2023-05-06 15:11:35 +03:00
parent 2789ef6461
commit c332247f1b
2 changed files with 29 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,5 +1,5 @@
from aiogram import types
from bot.db import db, DBTables, decrypt
from bot.db import db, DBTables
from bot.utils.cooldown import throttle
from bot.keyboards.set_model import get_set_model_keyboard
from bot.modules.api.models import get_models
@@ -9,8 +9,10 @@ from bot.utils.errorable_command import wrap_exception
@wrap_exception()
@throttle(cooldown=5*60, admin_ids=db[DBTables.config].get('admins'), by_id=False)
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')
if db[DBTables.config].get('whitelist') and \
(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()
if models is not None and len(models) > 0:
@@ -18,5 +20,5 @@ async def set_model_command(message: types.Message):
else:
await message.reply('❌ No models available')
return
await message.reply("Examples of models (with additional info): https://telegra.ph/Opisanie-raboty-modelej-05-03")
await message.reply("🪄 You can choose model from available:", reply_markup=get_set_model_keyboard(0))