Почему возникает ошибка AttributeError: 'Database' object has no attribute 'cursor'?

Question

[danya2854]

При выполнении функции add_queue, почему‑то вылазит ошибка: AttributeError: 'Database' object has no attribute 'cursor' из‑за чего это, и как исправить?

import telebot
from telebot import types 
from database import Database

db = Database('db.db') 
bot = telebot.TeleBot("Токен")

@bot.message_handler(content_types = ['text']) 
def bot_message(message): 
        if message.chat.type == 'private': 
                if message.text == ' Поиск собеседника': 
                    markup = types.ReplyKeyboardMarkup(resize_keyboard = True) 
                    item1 = types.KeyboardButton('❌ Остановить поиск') 
                    markup.add(item1) 
   
                    db.add_queue(message.chat.id)

                    bot.send_message(message.chat.id, '<b>Воспроизводиться поиск..</b>',reply_markup = markup, parse_mode='html')

import sqlite3

class Database:
    def __init__(self, database_file):
        self.connection = sqlite3.connect(database_file, check_same_thread = False)
        self.cursor = self.connection.cursor()
        
    def add_queue(self, chat_id):
        with self.connection:
            return self.cursor.execute("INSERT INTO `queue` (`chat_id`) VALUES (?)", (chat_id))
    
    def delete_queue(self, chat_id):
        with self.connection:
            return self.connection.cursor.execute("DELETE FROM `queue` WHERE `chat_id` = ?", (chat_id))

Comments

[Сергей Горностаев]

Где вы все только учитесь таким глупостям?..

[Vindicar]

Не, а кроме шуток... я что-то не соображу в чём дело. Думал, опечатка в cursor (русская с или что-то вроде), но нет.
Но вообще вот так вот с наскока check_same_thread = False делать не надо.

[danya2854]

Vindicar, Я делал всё как в видео, и у него всё работает, а у меня нет

[Сергей Горностаев]

Vindicar, если бы автор показал полный текст ошибки, было бы легче ткнуть пальцем в конкретное место. Но в целом весь класс Database - немыслимая ересь.

[Сергей Горностаев]

danya2854, читайте учебники и документацию, подавляющее большинство видео-уроков - низкокачественный шлак, записанный идиотами.

[danya2854]

Сергей Горностаев,

Traceback (most recent call last):
  File "D:\Анонимчатботтг\bot.py", line 35, in <module>
    bot.polling(none_stop=True)
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 1198, in polling
    self.__threaded_polling(non_stop=non_stop, interval=interval, timeout=timeout, long_polling_timeout=long_polling_timeout,
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 1273, in __threaded_polling
    raise e
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 1235, in __threaded_polling
    self.worker_pool.raise_exceptions()
  File "D:\Python\lib\site-packages\telebot\util.py", line 150, in raise_exceptions
    raise self.exception_info
  File "D:\Python\lib\site-packages\telebot\util.py", line 93, in run
    task(*args, **kwargs)
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 8945, in _run_middlewares_and_handler
    result = handler['function'](message)
  File "D:\Анонимчатботтг\bot.py", line 23, in bot_message
    db.add_queue(message.chat.id)
  File "D:\Анонимчатботтг\database.py", line 10, in add_queue
    return self.cursor.execute("INSERT INTO `queue` (`chat_id`) VALUES (?)", (chat_id))
AttributeError: 'Database' object has no attribute 'cursor'

[Vindicar]

Сергей Горностаев, ну вот меня это и смущает, что self.connection у него есть, а self.cursor - нет.
danya2854, скажи, пожалуйста, а у тебя точно класс Database выглядит, как показано? В нём ничего не изменилось?

Вообще, у тебя прилично в нём косяков:
1. Как я уже писал, нужно быть крайне осторожным с потоками, если используешь sqlite. Зачем добавил check_same_thread = False? Ответ "так было в видео" не принимается.
2. В delete_queue() с какого-то перепугу self.connection.cursor.execute вместо self.cursor.execute.
3. Потерял запятые в обоих методах - нужно (chat_id,), а не (chat_id). Смысл очень разный.

Ну и да - видео-уроки ругают не зря. Косяков в них часто много, объяснений "почему именно так", а не "делай раз, делай два" - часто недостаточно. Да и пойди разберись, какую именно версию библиотеки использовал автор. Лучше найти гитхаб этой библиотеки и поискать там папку с официальными примерами. Пользы будет куда как больше.

Я вот сейчас проверил код на таком скрипте - всё работает:

import sqlite3

class Database:
    def __init__(self, database_file):
        self.connection = sqlite3.connect(database_file, check_same_thread = False)
        self.cursor = self.connection.cursor()
        self.cursor.execute('CREATE TABLE IF NOT EXISTS queue (chat_id INT PRIMARY KEY)');  # создаём таблицу, если её не было
        
    def add_queue(self, chat_id):
        with self.connection:
            return self.cursor.execute("INSERT INTO `queue` (`chat_id`) VALUES (?)", (chat_id,))
    
    def delete_queue(self, chat_id):
        with self.connection:
            return self.connection.cursor.execute("DELETE FROM `queue` WHERE `chat_id` = ?", (chat_id,))


if __name__ == '__main__':
    d = Database(':memory:')
    d.add_queue(123)
    d.cursor.execute('SELECT * FROM queue')
    print(d.cursor.fetchall())

[danya2854]

Vindicar, Я изначально делал self.cursor.execute, но ничего не работало. Вот в общем-то то видео которое я смотрел
https://www.youtube.com/watch?v=_035pNI3-xY

[danya2854]

Vindicar, А зачем там

if __name__ == '__main__':
    d = Database(':memory:')
    d.add_queue(123)
    d.cursor.execute('SELECT * FROM queue')
    print(d.cursor.fetchall())

[danya2854]

Vindicar, Я использовал твой код, но всё ещё ошибка:

Traceback (most recent call last):
  File "D:\Анонимчатботтг\bot.py", line 35, in <module>
    bot.polling(none_stop=True)
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 1198, in polling
    self.__threaded_polling(non_stop=non_stop, interval=interval, timeout=timeout, long_polling_timeout=long_polling_timeout,
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 1273, in __threaded_polling
    raise e
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 1235, in __threaded_polling
    self.worker_pool.raise_exceptions()
  File "D:\Python\lib\site-packages\telebot\util.py", line 150, in raise_exceptions
    raise self.exception_info
  File "D:\Python\lib\site-packages\telebot\util.py", line 93, in run
    task(*args, **kwargs)
  File "D:\Python\lib\site-packages\telebot\__init__.py", line 8945, in _run_middlewares_and_handler
    result = handler['function'](message)
  File "D:\Анонимчатботтг\bot.py", line 23, in bot_message
    db.add_queue(message.chat.id)
  File "D:\Анонимчатботтг\database.py", line 10, in add_queue
    return self.cursor.execute("INSERT INTO `queue` (`chat_id`) VALUES (?)", (chat_id))
AttributeError: 'Database' object has no attribute 'cursor'
>>>

[Vindicar]

danya2854, ну так у меня бота нет, а проверить твой класс надо. Вот я и вызвал его методы сам.
А ты весь код покажи, в его нынешнем виде.
Кроме того, почему ты используешь polling()? Этот метод устарел, документация советует использовать infinity_polling().

[danya2854]

Vindicar, это основной файл

import telebot
from telebot import types
from database import Database

db = Database('db.db')
bot = telebot.TeleBot("Токен")

@bot.message_handler(commands=['start'])
def start(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
    item1 = types.KeyboardButton(' Поиск собеседника')
    markup.add(item1)
    bot.send_message(message.chat.id, f'<b>{message.from_user.first_name}</b>, добро пожаловать в нашего замечательного бота!\nНажмите на кнопку поиск собеседника или \nнапишите "/search" чтобы начать поиск',parse_mode='html', reply_markup = markup)

@bot.message_handler(content_types = ['text'])
def bot_message(message):
    if message.chat.type == 'private':
        if message.text == ' Поиск собеседника':
            markup = types.ReplyKeyboardMarkup(resize_keyboard = True)
            item1 = types.KeyboardButton('❌ Остановить поиск')
            markup.add(item1)
            
            db.add_queue(message.chat.id)
            
            bot.send_message(message.chat.id, '<b>Воспроизводиться поиск..</b>',reply_markup = markup, parse_mode='html')
            
        elif message.text == '❌ Остановить поиск':
            db.delete_queue(message.chat.id)
            bot.send_message(message.chat.id, '❌ Поиск остановлен.', parse_mode='html')

@bot.message_handler(commands=['search'])
def search(message):
    bot.send_message(message.chat.id, '<b>Воспроизводиться поиск..</b>',parse_mode='html')

bot.polling(none_stop=True)

[danya2854]

Vindicar, Это database

import sqlite3

class Database:
    def __init__(self, database_file):
        self.connection = sqlite3.connect(database_file, check_same_thread = False)
        self.cursor = self.connection.cursor()
        
    def add_queue(self, chat_id):
        with self.connection:
            return self.cursor.execute("INSERT INTO `queue` (`chat_id`) VALUES (?)", (chat_id,))
    
    def delete_queue(self, chat_id):
        with self.connection:
            return self.connection.cursor.execute("DELETE FROM `queue` WHERE `chat_id` = ?", (chat_id,))

[Vindicar]

danya2854, ну-ка, для отладки... вставь после db = Database('db.db') строку print(db.cursor), что выведет?

[danya2854]

Vindicar, ошибка

Traceback (most recent call last):
  File "D:\Анонимчатботтг\bot.py", line 6, in <module>
    print(db.cursor)
AttributeError: 'Database' object has no attribute 'cursor'
>>>

[Vindicar]

danya2854, окей, что-то не сходится. Проверь, у тебя database.py в проекте точно тот, который ты показываешь тут? Других файлов с таким именем нет? Файл целиком приводишь, ниже в нём ничего нет? Потому что всё выглядит так, словно это не тот класс Database, а какой-то другой.
Также попробуй вставить в конец конструктора Database.__init__() строку print('__init__:', self.cursor) и проверь, отработает ли она.

[danya2854]

Vindicar, Ошибка

>>> %Run bot.py
Traceback (most recent call last):
  File "D:\Анонимчатботтг\bot.py", line 5, in <module>
    db = Database('db.db')
  File "D:\Анонимчатботтг\database.py", line 7, in __init__
    print('__init__:', self.cursor)
AttributeError: 'Database' object has no attribute 'cursor'

[danya2854]

@Vindicar, Есть дискорд или телеграм?

[Vindicar]

danya2854, т.е. у тебя вот такой код?

class Database:
    def __init__(self, database_file):
        self.connection = sqlite3.connect(database_file, check_same_thread = False)
        self.cursor = self.connection.cursor()
        print('__init__:', self.cursor)  # и на этом print() вылетает ошибка?

Если так, то я пас. Такого не может быть.

[o5a]

Vindicar, действительно мистика.

danya2854, а что выводит, если прямо перед той строкой
print('__init__:', self.cursor)
добавить
print(dir(self))

Answer 1

[EugeneBel]

def add_queue(self, chat_id):
        with self.connection:
            return self.cursor.execute("INSERT INTO `queue` (`chat_id`) VALUES (?)", (chat_id))

После первого срабатывания connection закрывается из-за использования конструкции with.
Соответственно и cursor падает.
Строчку вообще нужно убрать. А connection закрывать вручную в конце.
И еще надо сделать cursor.commit() для сохранения изменений.

Comments

[danya2854]

А как? Можно написать в коментарии?