Как заставить бота искать в базе SQLite и выдавать ответ в чат?

Question

[Денис Алешин]

пытаюсь написать Телеграмм Бота для удобства поиска информации, создал базу sqlite в ней таблица с двумя столбцами. Нужно что бы бот любое сообщение пользователя искал в базе и при совпадении выдавал результат как сообщение в чат... Пробовал по разному, потратил честных 5 дней что бы понять как это сделать, но видимо тут совсем всё не просто, пишу первый раз.

import telebot
import config
import sqlite3


bot: telebot = telebot.TeleBot(config.TOKEN)
conn = sqlite3.connect('shops.db')
pur = conn.cursor()


@bot.message_handler(commands=['start'])
def message(message):
    bot.reply_to(message, "Привет")



cur = conn.cursor()
r = cur.execute('SELECT about FROM data WHERE shop =?', {message.text}).fetchone()
print(r)
conn.commit()

bot.polling(none_stop=True)<code lang="python">

</code>

Comments

[User]

cur.execute(...)
r = cur.fetchall()

При извлечении данных коммит не нужен

[Денис Алешин]

User, спасибо за ответ, а можешь чуть подробнее... плиз

[Никита Булгак]

Денис Алешин, ну он тебе предлагает вывести все данные из базы в массив , а дальше проверяй по условию и выводи потом. Если нужна помощь пиши в Telegram : @"Keffmen"

Answer 1

[WhiteMarlin]

import telebot
import config
import sqlite3


bot = telebot.TeleBot(config.TOKEN)
conn = sqlite3.connect('shops.db')

def get_row(text):
    cur = conn.cursor()
    data = cur.execute("SELECT about FROM data WHERE shop = ?", (text,)).fetchone()
    cur.close()
    return data

@bot.message_handler(commands=['start'])
def message(message):
    bot.send_message(message.chat.id, f"Привет, {message.chat.first_name}!")
    

@bot.message_handler(func=lambda message: True)
def send_data(message)
    data = get_row(message.text) # text - по которому выбираете запись в db
    bot.reply_to(message.chat.id, data[0]) # 0 индекс столбца, обычно 0 - это id. Замените на нужный индекс столбика

bot.polling(none_stop=True)

Вообще-то так никто не делает, это плохая затея. Вам нужно изучить SQL и безопасность. Но в качестве примера, как делать выборку из базы и отправлять в чат-бот - сойдет.

Comments

[Денис Алешин]

Спасибо за ответ! Я попробовал погуглить что за id столбца... И как я понимаю это не про таблицу в бд?

[WhiteMarlin]

Денис Алешин, id это название столбца в таблице базы данных. В нем хранится уникальный номер записи (строки). Если вы его не создали, то его нет в таблице, но его нужно обязательно создавать. Посмотрите структуру любой базы данных - везде есть id.

[Денис Алешин]

WhiteMarlin, Если вставляю в [] имя столбца из таблицы то ошибка...
"SQLite objects created in a thread can only be used in that same thread. The object was created in thread id 8516 and this is thread id 3816."

[Денис Алешин]

WhiteMarlin, То есть, правильно лия понимаю, просто добавить ещё один столбец с именем "id" и заполнить его просто числами? 1, 2, 3, итд?

Но тогда всё равно не очень понятно, какое значение вписать в

bot.reply_to(message.chat.id, data[0])

[Денис Алешин]

Собственно создал дополнительный столбец с именем "id", всё равно выдаёт ошибку после первого обращения к боту...(Запускается Бот нормально)



Текст ошибки
SQLite objects created in a thread can only be used in that same thread. The object was created in thread id 3744 and this is thread id 9164.

Answer 2

[Николай Савельев]

data = cur.execute("SELECT about FROM data WHERE shop = ?", (text,)).fetchone()

SQL Injection found

Comments

[WhiteMarlin]

Поэтому я и написал, что так никто не делает. Это открытые двери для SQL Injection.

[Денис Алешин]

WhiteMarlin, Я это исправлю, мне хотя бы как то реализовать поиск и вывод инфы в чат, потом доделаю. Спасибо