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

Add whitelist to bot

Browse Source
This commit is contained in:
seel2304
2023-05-03 15:50:15 +00:00
parent c129c7bc60
commit 2dda780d3f
6 changed files with 86 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
bot/handlers/initialize/start.py
View File

@@ -18,6 +18,13 @@ async def start_command(message: types.Message):
await db[DBTables.config].write()
await message.reply(f'✅ Added {message.from_user.username} to admins. You can add other admins, '
f'check bot settings menu')
if ADMIN not in db[DBTables.config].get('whitelist'):
whitelist_ = db[DBTables.config].get('whitelist')
whitelist_.append(ADMIN)
db[DBTables.config]['whitelist'] = whitelist_
await db[DBTables.config].write()
await message.reply(f'✅ Added {message.from_user.username} to whitelist. You can add other users to whitelist, '
f'check bot settings menu')
if db[DBTables.config].get('enabled') is None:
db[DBTables.config]['enabled'] = True
await message.reply(f'✅ Generation is enabled now')