Как получить данные SQL запроса?

Question

[enabl3]

Всем привет.
Подскажите пожалуйста, что я делаю не правильно?
Есть файл requests.sql, в нем пишу все запросы, в нем есть запрос на получение всех айдишников юзеров, выглядит так:

def users_in_db(message):
    conn = sqlite3.connect(db)
    cursor = conn.cursor()
    sql_users_in_db = """SELECT chat_id FROM users;"""
    cursor.execute(sql_users_in_db)
    users = cursor.fetchall()
    print(users)
    conn.close()

теперь в коммандах, файле handlers.py, хочу проверить есть ли юзер, написавший сообщение в бд

@dp.message_handler(commands=["users"])
async def process_start_command(message: Message):
    if str(message.chat.id) in str(users_in_db(message)):
        print("зареган")
    else:
        print("не зареган")

ответ в консоль приходит - список айдишников юзеров + не зареган. Если в коде напишу print(str(users_in_db(message))) и запущу, то в консоле будет список айдишников в базе + None.

Answer 1

[ScriptKiddo]

Ваша функция ничего не возвращает, поэтому и выводится всегда "не зареган".
Если у функции нет return, то результатом ее выполнения будет None

В итоге, if разворачивается в str(message.chat.id) in 'None', что всегда False, если только message.chat.id != 'None' или любому срезу строки.

Используйте контекстный менеджер with, так не придется закрывать курсор или подключение к БД вручную.
Возвращайте результат из функции, чтобы было с чем работать после того, как она выполнится.

Почитайте любой самоучитель по Python. Например, Лутца. Это отбросит множество таких вопросов и здорово поможет в изучении языка

def users_in_db(message):
    with sqlite3.connect(db) as conn:
        with conn.cursor() as cursor:
            sql_users_in_db = """SELECT chat_id FROM users WHERE chat_id = ?;"""
            cursor.execute(sql_users_in_db, message.chat.id)
            users = cursor.fetchall()
            print(users)
            if users:
                return True
    return False


@dp.message_handler(commands=["users"])
async def process_start_command(message: Message):
    if users_in_db(message):
        print("зареган")
    else:
        print("не зареган")

Comments

[enabl3]

Большое спасибо за ответ и рекомендацию. Обязательно учту.
Но как мне получить весь масив айдишников всех пользователей и почему оно по разоному работает, если тот же код выполнять с другого файла или же выполнять в хендлерах.
К примеру, если я хочу получить масив айдишников пользователей то в хендлерах я выполняю

conn = sqlite3.connect(db)
cursor = conn.cursor()
sql_users_in_db = """SELECT chat_id FROM users;"""
cursor.execute(sql_users_in_db)
conn.close()
users = cursor.fetchall()
print(users)

и с users я могу работать, так как возвращает список айдишников.
а если этот же код выполняется с другого файла, то users возвращает ничего.

[enabl3]

ScriptKiddo
разобрался

if users:
        return users

тогда возвращает список айдишников