Телеграм бот. SQlite3 как выводить id?

Question

[Liperr]

Нужно, чтобы при каждом нажатии пользователя команды /start, его id вносился в таблицу, если его там нет. На данном этапе в таблицу записывается ''built-in function id'' и делается это каждый раз(нужно только, если айди нет). Что добавить, а что убрать? Буду благодарен, если ответ будет подробным, с sqlite работаю впервые.

import telebot
import sqlite3
from telebot import types

bot = telebot.TeleBot('1084130875:AAGB5EfXIMo')
id = str(id)

@bot.message_handler(commands=['start'])
def start(message):
	markup = types.ReplyKeyboardMarkup(resize_keyboard=True, row_width=2)
	btn1 = types.KeyboardButton('Поиск')
	btn2 = types.KeyboardButton('Связь')
	btn3 = types.KeyboardButton('Добавить')
	markup.add(btn1, btn2, btn3)

	send_message = f"<b>Привет {message.from_user.first_name}!</b>"
	bot.send_message(message.chat.id, send_message, parse_mode='html', reply_markup=markup)

	connection = sqlite3.connect("database.db", check_same_thread = True)
	cursor = connection.cursor()

	cursor.execute("CREATE TABLE IF NOT EXISTS users (id)")
	cursor.execute("INSERT OR IGNORE INTO users (id) VALUES (:id)", {"id": id})

	connection.commit()
	connection.close()

Answer 1

[Александр]

для начала сделать импорты:

import sqlite3
from sqlite3 import Error
from time import sleep, ctime

вызов всех SQL выражений сделать через функцию:

def post_sql_query(sql_query):
    with sqlite3.connect('my.db') as connection:
        cursor = connection.cursor()
        try:
            cursor.execute(sql_query)
        except Error:
            pass
        result = cursor.fetchall()
        return result

далее нужно создать таблицу в sqlite и в качестве primary key (именно он на уникальность определит) сделать id user:

def create_tables():
    users_query = '''CREATE TABLE IF NOT EXISTS USERS 
                        (user_id INTEGER PRIMARY KEY NOT NULL,
                        username TEXT,
                        first_name TEXT,
                        last_name TEXT,
                        reg_date TEXT);'''
    post_sql_query(users_query)

и функцию регистрации пользователя:

def register_user(user, username, first_name, last_name):
    user_check_query = f'SELECT * FROM USERS WHERE user_id = {user};'
    user_check_data = post_sql_query(user_check_query)
    if not user_check_data:
        insert_to_db_query = f'INSERT INTO USERS (user_id, username, first_name,  last_name, reg_date) VALUES ({user}, "{username}", "{first_name}", "{last_name}", "{ctime()}");'
        post_sql_query(insert_to_db_query )

а дальше вызывать её из хендлера:

create_tables()  # вызываем функцию создания таблицы users


@bot.message_handler(commands=['start'])
def start(message):
    register_user(message.from_user.id, message.from_user.username,
                  message.from_user.first_name, message.from_user.last_name)
    bot.send_message(message.from_user.id, f'Welcome  {message.from_user.first_name}' )

Думаю принцип понятен, далее можно со всеми данными так работать и да код так себе, но должен работать.
Достаточно подробно?)

Comments

[Liperr]

Да, очень подробно:) Но не работает. Таблица создалась, но новые пользователи не заносятся(

[Александр]

Liperr, тест на внимательность не пройден.

insert_to_db_query = f'INSERT INTO USERS (user_id, username, first_name,  last_name, reg_date) VALUES ({user}, "{username}", "{first_name}", "{last_name}", "{ctime()}");'

надо правильно обернуть ctime

[Liperr]

Александр Шабельский, спасибо большое, всё работает!))

[Victoriast0408]

Александр, а как сделать update таблицы, если я добавила столбец Description, ловлю вводимый юзером текст description = Message.text и хочу добавить этот текст в колонку Description?

[Александр]

Victoriast0408, нужно уточнить что именно Вам не понятно, можете сформулировать по пунктам?

[Victoriast0408]

Александр,
1. Когда юзер нажимает на команду /start, бот приветствует его и записывает его user_id, first_name, last_name в базу данных. Рекомендованный Вами код работает отлично, именно так как нужно.
2. Дальше, бот предлагает юзеру перейти по команде /begin
3. Когда юзер нажимает на команду /begin, бот выводит сообщение с просьбой ввести текст (это описание товара, которое необходимо также внести в базу данных). Для этого текста в бд предусмотрена колонка description.
4. Юзер вводит текст
5. Бот отлавливает введённый юзером текст (Message.text сохраняется в переменную description).
6. Теперь мой вопрос: как дополнить базу данных содержимым этой переменой description (а именно записать введённый юзером текст в бд в колонку description)?
Заранее спасибо)

[Victoriast0408]

Александр,
Мой неработающий вариант (текст в колонку description не записывается):

def add_description(description)
       update_db_desc = f‘ UPDATE ITEMS SET    (description) = {description} WHERE person_id = person_id;‘
        post_sql_query(update_db_desc)

И вот скрин по боту:


** person_id это порядковый номер в таблице

Где я допустила ошибку?)

[Александр]

Victoriast0408, завтра постучите в ЛС , посмотрим