Как мне сделать проверку key в базе данных?

Question

[kado123]

В моем телеграм боте есть база данных с данными таблицами.


А также код:

@bot.message_handler(commands=['getfile'])
def gf_cmd(message):
    cur.execute(f"SELECT key FROM users")
    result = cur.fetchone()
    if result == 1:
        bot.send_message(message.chat.id, "Вот твои файлы: ...")
    if result == 0: 
        bot.send_message(message.chat.id, "У вас нет доступа...")

key - в базе данных является доступом для файлов в боте, но при вводе команды /getfile бот никак не реагирует. Как сделать так, чтоб он проверял это значение и если оно 0 - доступа нет, если 1 - доступ есть.

Comments

[Алан Гибизов]

Вы разделите проблему на части. Убедитесь, что хэндлер сам по себе корректно перехватывает команду - без всей этой логики внутре. Пусть просто напечатает ченьть.
Потом отдельно убедитесь, что селект корректно возвращает запрошенное.
Потом отдельно убедитесь, что логика корректно обрабатывает то, что вернул селект.
А то как понять, что у вас из этого косячит?

[kado123]

Алан Гибизов, хендлер нормально обрабатывает команду, скорее всего проблема в селекте

[Алан Гибизов]

kado123, а вы посмотрите, какие варианты возвращает селект в result. Почему-то мне кажется, что там например строка может быть, а вы её с int сравниваете. Или ещё какой косяк подобного рода.

[kado123]

Алан Гибизов, оказывается он возвращает не 0 или 1, а 0, и 1,

[kado123]

Алан Гибизов, можно как то это починить?

[kado123]

Алан Гибизов, Починил.

[Алан Гибизов]

kado123, в таком вот аксепте :)

Answer 1

[kado123]

Проблема решена, вместо 0, select отправлял 0, .

Answer 2

[Jack444]

У вас запрос не дописан, нужно же ключ конкретного пользователя проверять тогда запрос будет выглядеть так
f'SELECT key FROM users WHERE chatid={message.chat.id} LIMIT 1'

Comments

[kado123]

нужно проверять всех у кого ключ = 1

[kado123]

к сожалению не работает

f'SELECT key FROM users WHERE chatid={message.chat.id} LIMIT 1

[Jack444]

kado123, я не знаю через какой драйвер у вас идет соеденение с бд но обычно в result прилетает либо dict либо list, может проверку нужно сделать както так

if result['key']:
    bot.send_message(message.chat.id, "Вот твои файлы: ...")
else: 
    bot.send_message(message.chat.id, "У вас нет доступа...")

или так

if result[0]:
    bot.send_message(message.chat.id, "Вот твои файлы: ...")
else: 
    bot.send_message(message.chat.id, "У вас нет доступа...")

[Jack444]

kado123, можно ещё так упростить

@bot.message_handler(commands=['getfile'])
def gf_cmd(message):
    cur.execute(f'SELECT TRUE FROM users WHERE chatid={message.chat.id} AND key=1 LIMIT 1')
    result = cur.fetchone()
    if result:
        bot.send_message(message.chat.id, "Вот твои файлы: ...")
    else: 
        bot.send_message(message.chat.id, "У вас нет доступа...")