Как использовать функцию в других функциях?

Question

[mikitos59]

Имею много хендлеров, где постоянно идет подключение к бд и проверка ячейки. Решил вынести отдельно функцию connect, чтобы не прописывать постоянно конект к бд. Как сделать, чтобы использовать эту функцию во всех хендлерах ? Спасибо за ответ)

def connect(message: types.Message):
    connect = sqlite3.connect('main.db')
    cursor = connect.cursor()
    cursor.execute('SELECT Bat FROM Users WHERE id = ?', (message.from_user.id))
    result = cursor.fetchone()
    if result is None:
        print('none')
        
        
@dp.message_handler(text='Hi')

    for row in result:
        if row == 'X':
            cursor.execute("SELECT `Glo` FROM `Scev` WHERE `test` = 'test'")
            row = cursor.fetchone()
            await message.answer(row[0], reply_markup=il_1)
    connect.close()

Comments

[Алан Гибизов]

Мне кажется, что придуманная функция работать не должна. Потому, что её имя connect, а внутри себя она тоже имеет имя connect, но при этом во-пепвых изнутри наружу внутренний connect явно не передаётся, а во-вторых внешний connect с внутренним, получается, конфликтует по имени.

Возможно, существует какая-то магия хендлеров, я с ней не знаком.

Answer 1

[Сергей Еремин]

Используй конструкцию with ... as... Она для этого и придумана...

Comments

[mikitos59]

Можно небольшой пример применения with as ?

[Сергей Еремин]

mikitos59,
Например, вот:
https://stackoverflow.com/questions/19522505/using...

class SQLite():
    def __init__(self, file='sqlite.db'):
        self.file=file
    def __enter__(self):
        self.conn = sqlite3.connect(self.file)
        self.conn.row_factory = sqlite3.Row
        return self.conn.cursor()
    def __exit__(self, type, value, traceback):
        self.conn.commit()
        self.conn.close()

with SQLite('test.db') as cur:
    print(cur.execute('select sqlite_version();').fetchall()[0][0])

Answer 2

[o5a]

Если цель - использовать одно соединение, можно сделать через класс.

class DataBase():
    def __init__(self, database_name):
        self.conn = sqlite3.connect(database_name)
        self.cursor = self.conn.cursor()

    def commit(self):
        self.conn.commit()

    def close(self):
        self.conn.close()

# создаем объект класса
db = DataBase('my_db.db')
cursor = db.cursor

# и в любой функции можем использовать
cursor.execute ...
db.commit()