Вывод данных из бд в свой профиль telegram bot python?

Question

[Вова Ефанов]

class SQLighter:
	def __init__(self, database):
	    self.connection = sqlite3.connect(database, check_same_thread = False)
	    self.cursor = self.connection.cursor()
	    self.cursor.execute("CREATE TABLE IF NOT EXISTS stats (user_id INT, user_name TEXT, survived INT, infected INT)")
	def add_user(self, id, name, survived, infected):
	    #"""Добавляем значение"""
	    with self.connection:
	        return self.cursor.execute("""INSERT INTO `stats` (user_id, user_name, survived, infected) VALUES (?,?,?,?)""", (id, name, survived, infected))
	def get_nick(self, id):
	    #'''Эта функция позволяет извлекать нужное значение(в данном случае id) из таблицы stats'''
	    with self.connection:
	        return self.cursor.execute("SELECT user_id FROM stats WHERE id = ?", (id,)).fetchall()
	def set_fast(self, id):
	    #'''Эта функция позволяет устанавливать значение исходя из параметра id'''
	    with self.connection:
	        return self.cursor.execute("UPDATE stats SET infected=X WHERE id=?", (id))
	

sql_lighter = SQLighter("database.db")
sql_lighter.add_user(982543922, "Владимир", 1, 1)

Мой профиль:

bot.send_message(message.chat.id, f"Мой профиль\nМой ID: [{message.from_user.id}]\nМой Псевдоним: [{message.from_user.username}]\nМоё Ф.И: [{message.from_user.first_name} {message.from_user.last_name}]\nКоличество побед за выжившего: [{survived}]\nКоличество заражений за заражённого: [{infected}]")

Как мне из бд извлечь данные в свою профиль телеграмм бота?
Профиль - моя статистика

Comments

[Alexa2007]

а что не получилось именно

[Вова Ефанов]

Alexa2007, NameError: name 'survived' is not defined

[Вова Ефанов]

NoPressF, Якобы этой переменной не существует

[Alexa2007]

NoPressF, ну да, как и сказали выше, не сущуствует.. тебе ее надо как и get_nik выдернуть из базы

[Alexa2007]

NoPressF, а потом и с infected будет тоже самое

[1ksid]

NoPressF, Могу помочь с созданием игры. Если интересно напиши в тг @oneksid

Answer 1

[o5a]

Если удобнее, то можно получать данные в виде словаря (где ключи - названия столбцов). Тогда можно напрямую обращаться к нужному полю по имени.

class SQLighter:
  def __init__(self, database):
      self.connection = sqlite3.connect(database, check_same_thread = False)
      self.connection.row_factory = sqlite3.Row # этой строкой указываем, что результат получаем в виде словаря
      self.cursor = self.connection.cursor()
      self.cursor.execute("CREATE TABLE IF NOT EXISTS stats (user_id INT, user_name TEXT, survived INT, infected INT)")
...
  # и можно сделать такую функцию, возвращающую всю информацию пользователя
  def get_info(self, id):
      with self.connection:
          return self.cursor.execute("SELECT * FROM stats WHERE user_id=?", (id, )).fetchone()

sql_lighter = SQLighter("database.db")
sql_lighter.add_user(982543922, "Владимир", 1, 1)
# затем получаем весь словарь данных и можем использовать любые нужные поля из него
data = sql_lighter.get_info(982543922)
print(data)
print(data['user_name'])
print("Информация пользователя {}: имя: {}, survived: {}, infected: {}".format(data['user_id'], data['user_name'], data['survived'], data['infected']))

Comments

[Вова Ефанов]

А как сделать для каждого пользователя свои данные будут выводится в bot.send_message

[o5a]

NoPressF, так какой id передаете в get_info(id), по тому пользователю он и выдаст информацию.

[Вова Ефанов]

o5a, мне нужно писать каждый раз data = sql_lighter.get_info(id) ?

[o5a]

NoPressF, да, если нужно получить текущие данные по конкретному user_id.
Что значит "каждый раз"? Что вообще пытаетесь реализовать?

[Вова Ефанов]

o5a, Ну вот допустим я хочу каждому пользователю установить значение двух переменных в его профиле.

[o5a]

NoPressF, "установить" вообще не через SELECT делается, для этого UPDATE. Прочитайте хотя бы основы SQL, сразу понятнее будет, как это работает.
Типа такого https://proglib.io/p/sql-for-20-minutes

[Вова Ефанов]

o5a, Я если прочитаю то в итоге ничеоо не пойму

Answer 2

[tshipenchko]

Просто к слову, чем вам ORM не понравился)