Как сделать вайтлист для телеграм бота на python aiogram?

Question

[flameyolo]

Требуется сделать вайтлист для пользователей моего телеграм бота, чтобы без моего добавления юзера в разрешенный список бот отвечал ему, что он не находится в вайтлисте.

whitelist = []

@dp.message_handler(func=lambda message: message.chat.id not in whitelist)
def checker(message):
   bot.send_message(message.chat.id, "Вы не находитесь в Whitelist!")

Пытался что-то сделать на коленке, ничего не работает, выдает ошибку на моменте запуска бота.

Answer 1

[LXSTVAYNE]

Во-первых декорированная функция должна быть помечена как async. Во-вторых aiogram это асинхронная библиотека, значит все её сетевые функции должны быть await . Код работает, если всё это исправить.

Comments

[flameyolo]

Спасибо большое, проблема решена!

Код для потомков:

whitelist = []

@dp.message_handler(lambda message: message.chat.id not in whitelist)
async def checker(message):
   await message.answer("Вы не находитесь в Whitelist!")

Answer 2

[Владимир]

Для этого лучше создать бд, с колонной username(это будет message.from_user.username), и создать декоратор на команду какую ты захочешь(пример: /add).И чтобы добавить человека в Вайт лист нужно добавить его в бд.Ты делаешь переменную допустим user_add_in_db присваиваешь и в функции, где декоратор делаешь срез [4:] когда ты вводишь в боте /add {юзернейм чела}, ты добавляешь его в бд. И при отправке человеком команды, делаешь проверку если есть в бд то он в Вайт листе иначе его нет там. Если что-то непонятно, отпиши.

Comments

[flameyolo]

Я понял о чем ты говоришь, бот ограничивает пользователя, пока он не в вайтлисте, но он не дает это ему понять. Т.е. он не отправляет сообщение с тем, что человек не в вайтлисте, как это пофиксить?

И как вообще мне получить цифры из message.chat.id?

[Владимир]

flameyolo, возможно убрать лямбду из декоратора, и везде сделать if not message.chat.id in {запрос к бд}

[flameyolo]

Владимир, без лямбды даже не запускается

[flameyolo]

Владимир, я вообще убрал func= около лямбды, так оно работает, но не дает ответа, может быть знаешь решение? Буду благодарен

[Владимир]

flameyolo,

# Визуализация БД
db = {'user_id': 27747659}


@dp.message_handler()
async def checker(message):
if not message.chat.id in db['user_id']:
    await bot.send_message(message.chat.id,  'Ты не в Whitelist')
else:
    await bot.send_message(message.chat.id, 'Ты в Whitelist')

[Владимир]

flameyolo, я не заметил у тебя не было еще async перед def и await перед сообщением, попробуй с ними

[flameyolo]

Владимир, проблема уже была решена подобным способом

whitelist = []

@dp.message_handler(lambda message: message.chat.id not in whitelist)
async def checker(message):
   await message.answer("Вы не находитесь в Whitelist!")

Но все равно спасибо!

Answer 3

[Dim]

Вариант получше

from aiogram import types
from aiogram.dispatcher.middlewares import BaseMiddleware

class AccessMiddleware(BaseMiddleware):
    def __init__(self, allowed_users):
        super().__init__()
        self.allowed_users = allowed_users

    async def on_process_message(self, message: types.Message, data: dict):
        user_id = message.from_user.id
        if user_id not in self.allowed_users:
            await message.answer("У вас нет доступа к этому боту.")
            raise CancelHandler()  # Остановка дальнейшей обработки

# Добавьте это в bot.py
allowed_users = [123456789, 987654321]  # Замените на ваш список разрешённых ID
dp.middleware.setup(AccessMiddleware(allowed_users))