Как боту aiogram вывести число из бд sqlite?

Question

[newbie63]

В бд у каждого пользователя есть свое число. Я написал код на aiogram, но бот выдает одно и то же число(у каждого пользователя свое число). Как сделать так чтобы бот сопостовлял айди пользователя и айди из бд и отправлял число именно данного пользователя

from aiogram import Bot, types
from aiogram.types import ChatActions
import asyncio
import logging
from aiogram.dispatcher import Dispatcher
from aiogram.utils import executor
from config import TOKEN
import sqlite3

class SQLither:

    def __init__(self, database) -> object:
        self.conn = sqlite3.connect(database)
        self.c = self.conn.cursor()

    def exists_user(self, user_id):
        return bool(self.c.execute("SELECT * FROM users WHERE user_id=?", (user_id,)).fetchone())

    def add_to_db(self, user_id, first_name, last_name, username):
        self.c.execute("INSERT INTO users(user_id, first_name, last_name, username) VALUES(?,?,?,?,?)", (user_id, first_name, last_name, username, number))
        self.conn.commit()

bot = Bot(token=TOKEN)
dp = Dispatcher(bot)
db = SQLither("db.db")


@dp.message_handler(commands=['start'])
async def process_hi1_command(message: types.Message):
    user_id = message.from_user.id
    first_name = message.from_user.first_name
    last_name = message.from_user.last_name
    username = message.from_user.username
    if not db.exists_user(user_id):
        db.add_to_db(user_id, first_name, last_name, username)
    await message.answer('Добро пожаловать!')


@dp.message_handler(commands=['number'])
async def number(message: types.Message):
    conn = sqlite3.connect('db.db')
    cur = conn.cursor()
    cur.execute(f"SELECT * FROM users WHERE user_id")
    result = cur.fetchall()
    user_id = message.from_user.id
    await bot.send_message(message.from_user.id, f'Ваше число: = {[list(result[0])[5]][0]}')


if __name__ == "__main__":
    executor.start_polling(dp, skip_updates=True)

Answer 1

[o5a]

В функции exists_user как раз правильно составлен запрос на выбор конкретного пользователя (с использованием WHERE).
Аналогично должен быть составлен и запрос в функции number. В текущем виде тот запрос выбирает всех подряд.
Для передачи самого номера пользователя в команду нужно достать его из параметров:

async def number(message: types.Message):
    num = message.get_args()

Comments

[newbie63]

Нужно просто добавить еще один запрос только теперь с number? Заранее извините за глупые вопросы, не могу разобраться

cur = conn.cursor()
    cur.execute("SELECT * FROM users WHERE number")

[o5a]

newbie63, в запросе после WHERE указывается название поля таблицы, по которому собираетесь фильтровать. Вы писали

В бд у каждого пользователя есть свое число.

Вот это поле с числом, по которому собираетесь выбирать, и нужно указать в запросе.
Предполагаю, что речь про поле user_id, значит его и писать, а сам аргумент команды указывать как значение:

num = message.get_args()
cur.execute("SELECT * FROM users WHERE user_id = ?", (num, ))

Тогда команда
/number 12345
выдаст информацию о пользователе с user_id 12345

[newbie63]

o5a, Мне нужно чтобы бот отвечал на команду /number без дописывания к ней user_id

@dp.message_handler(commands=['number'])
async def number(message: types.Message):
    conn = sqlite3.connect('db.db')
    cur = conn.cursor()
    number = message.get_args()
    nomer = cur.execute("SELECT * FROM users WHERE user_id = ?", (number, ))
    await bot.send_message(message.from_user.id, f'Ваше число: = {nomer}')

На команду, бот выдает: Ваше число: =

[o5a]

newbie63, Вы сами понимаете, про какое "число пользователя" тут говорите? Что вы под этим числом понимаете? Вы создаете таблицу с 4 полями: user_id, first_name, last_name, username, вот из них и выбирайте.
Чтобы выводила по user_id самого пользователя? Тогда его и подставляйте.

@dp.message_handler(commands=['number'])
async def number(message: types.Message):
    conn = sqlite3.connect('db.db')
    cur = conn.cursor()
    user_id = message.from_user.id
    result = cur.execute("SELECT * FROM users WHERE user_id = ?", (user_id, )).fetchone()
    print(result) # вот и смотрите, что из этого хотите брать
    nomer = result[0] # например user_id
    await bot.send_message(message.from_user.id, f'Ваше число: = {nomer}')

Если речь про user_id, то зачем вообще его в базе искать, если он итак есть в message.from_user.id
В общем, определитесь, что за "число пользователя" ищете.

[newbie63]

o5a, Спасибо огромное, все получилось
P.S у меня в таблице не 4 поля, а 5. Последнее поле с этим числом

Answer 2

[UberPool]

О каком числе идет речь?
Вы записываете telegram_id,если вам нужно записать что-то другое просто добавьте запись.
По примеру вашей архитектуры:

user_id, first_name, last_name, username, <ваше число>

.
И тогда просто выводите это число пользователю,сопоставив telegram_id в базе данных.