@dj_igoran

UnboundLocalError: local variable '' referenced before assignment, что делать?

Не работает бот, при команде "/start" работает до первого ввода пользователя.

Вот ошибка:

INFO:aiogram.dispatcher.dispatcher:Start polling.
ERROR:asyncio:Task exception was never retrieved
future: exception=UnboundLocalError("local variable 'signup' referenced before assignment")>
Traceback (most recent call last):
File "C:\Users\zamov\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\dispatcher.py", line 414, in _process_polling_updates
for responses in itertools.chain.from_iterable(await self.process_updates(updates, fast)):
File "C:\Users\zamov\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\dispatcher.py", line 236, in process_updates
return await asyncio.gather(*tasks)
File "C:\Users\zamov\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\handler.py", line 116, in notify
response = await handler_obj.handler(*args, **partial_data)
File "C:\Users\zamov\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\dispatcher.py", line 257, in process_update
return await self.message_handlers.notify(update.message)
File "C:\Users\zamov\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\handler.py", line 116, in notify
response = await handler_obj.handler(*args, **partial_data)
File "c:\Users\zamov\OneDrive\Desktop\telebot\main.py", line 31, in bot_message
if db.get_singup(message.from_user.id) == "setnickname":
File "c:\Users\zamov\OneDrive\Desktop\telebot\db.py", line 26, in get_singup
return signup
UnboundLocalError: local variable 'signup' referenced before assignment

Код db.py:

import sqlite3

class Database:
    def __init__(self, db_file):
        self.connection = sqlite3.connect(db_file)
        self.cursor = self.connection.cursor()

    def add_user(self, user_id):
        with self.connection:
            return self.cursor.execute("INSERT INTO 'users' ('user_id') VALUES (?)", (user_id,))

    def user_exists(self, user_id):
        with self.connection:
            result = self.cursor.execute("SELECT * FROM 'users' WHERE 'user_id' = ?", (user_id,)).fetchall()
            return bool(len(result))

    def set_nickname(self, user_id, nickname):
        with self.connection:
            return self.cursor.execute("UPDATE 'users' SET 'nickname' = ? WHERE 'user_id' = ?", (nickname, user_id,))

    def get_singup(self, user_id):
        with self.connection:
            result = self.cursor.execute("SELECT 'signup' FROM 'users' WHERE 'user_id' = ?", (user_id,)).fetchall()
            for row in result:
               signup = str(row[0])
            return signup

    def set_singup(self, user_id, signup):
        with self.connection:
            return self.cursor.execute("UPDATE 'users' SET 'signup' = ? WHERE 'user_id' = ?", (signup, user_id,))

Код main.py:

import logging
from aiogram import Bot, Dispatcher, executor, types
import markups as nav
from db import Database

TOKEN = "2091008426:AAFgzkWhLouPSTWs2Yh8hZI0-KEmhaSDcmo"

logging.basicConfig(level=logging.INFO)

bot = Bot(token=TOKEN)
dp = Dispatcher(bot)

db = Database('database.db')

@dp.message_handler(commands=['start'])
async def start(message: types.Message):
    if(not db.user_exists(message.from_user.id)):
        db.add_user(message.from_user.id)
        await bot.send_message(message.from_user.id, "Введите никнейм.")
    else:
        await bot.send_message(message.from_user.id, "Вы уже зарегестрированы.", reply_markup=nav.mainMenu)


@dp.message_handler()
async def bot_message(message: types.Message):
    if message.chat.type == 'private':
        if message.text == 'ПРОФИЛЬ':
            pass

        else:
            if db.get_singup(message.from_user.id) == "setnickname":
                if(len(message.text) > 20):
                    await bot.send_message(message.from_user.id, "Количество символов не должно превышать 20.")
                elif '@' in message.text or '*' in message.text:
                    await bot.send_message(message.from_user.id, "Вы ввели запрещенный символ.")
            else:
                db.set_nickname(message.from_user.id, message.text)
                db.set_singup(message.from_user.id, "done")
                await bot.send_message(message.from_user.id, "Регистрация завершена!", reply_markup=nav.mainMenu)
    else:
        await bot.send_message(message.from_user.id, "Неизвестная команда.", reply_markup=nav.mainMenu)




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


Подскажите пожалуйста, что с этим делать?
(Объявить signup global не помогло)
  • Вопрос задан
  • 837 просмотров
Решения вопроса 1
sergey-gornostaev
@sergey-gornostaev Куратор тега Python
Седой и строгий
Заменить fetchall на fetchone и убрать цикл перед return signup.
Ответ написан
Пригласить эксперта
Ответы на вопрос 1
@dorolsik
Если работаете со SQLiteStudio попробуйте убрать кавычки у названий таблиц. Мне помогло.
Ответ написан
Комментировать
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы