Как избежать повторения кода с открытием базы (sqlite3, Python)?

Question

[PesyCorm]

Привет! Подскажите, как я могу вынести логику открытия дб и проверки создана ли таблица?
Например, есть две функции: одна добавляет пользователя, другая возвращает список пользователей. И каждый раз мне нужно проверять есть ли таблица, создавать если нет

def add_user_into_db(login, password):

	with sqlite3.connect('users_base.db') as db:
		db_cursor = db.cursor()
		db_cursor.execute("CREATE TABLE IF NOT EXISTS users (login TEXT, password TEXT)")
		db.commit()

		добавить_пользователя()

def select_from_db(sel):

	with sqlite3.connect('users_base.db') as db:
		db_cursor = db.cursor()
		db_cursor.execute("CREATE TABLE IF NOT EXISTS users (login TEXT, password TEXT)")
		db.commit()

		список_пользователей()

Как я могу избежать повторения кода с открытием базы и проверкой таблицы?
Спасибо за ответ!

Comments

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

Я не большой спец, но мне кажется, во-первых, хорошо бы в принципе изо всех функций убрать открытие базы, а во все функции передавать объект базы как параметр; во-вторых, почему бы не вынести проверку наличия базы в отдельную функцию, и вызывать её из других твоих функций?

[PesyCorm]

Алан Гибизов, думал об этом же, но проблема в том, что открываю базу через with, а из него нельзя вернуть объект, можно только передать. Не могу сообразить, как можно реализовать такой способ, ведь в зависимости от того, хочу ли я добавить пользователя или просмотреть список, объект придется передавать в разные функции

[PesyCorm]

Алан Гибизов, а на счет того, чтобы вынести проверку наличия базы в отдельную функцию - отличный совет, спасибо! Если нет возможности полностью перенести логику со стартом работы с бд, то проверку точно можно не расписывать повторно

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

PesyCorm, я вижу это всё примерно вот так:

Код

import sqlite3


def test_db(db):
    try:
        db_cursor = db.cursor()
        db_cursor.execute("CREATE TABLE IF NOT EXISTS users (login TEXT, password TEXT)")
        db.commit()
        return True
    except DBError:  # тут надо правильную ошибку правильно обрабатывать, я не копался, какую именно и как
        return False


def add_user_into_db(db, login, password):
    if test_db(db):
        добавить_пользователя(login, password)


def select_from_db(db, sel):
    if test_db(db):
        return список_пользователей(sel)


if __name__ == '__main__':
    login = 'my_login'
    password = 'my_password'
    sel = 'string_for_select'
    with sqlite3.connect('users_base.db') as db:
        add_user_into_db(db=db, login=login, password=password)
        print(select_from_db(db=db, sel=sel))

Это тоже не лучший вариант, по-хорошему надо уж делать через pythonway и ООП, т.е. через ORM SQLAlchemy и всё такое.

[o5a]

PesyCorm, а зачем вообще так часто требуется проверка/создание таблицы. Создание таблицы стоит делать в какой-то одной функции инициализации базы в самом начале работы и всё.

Answer 1

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

PesyCorm, я вижу это всё примерно вот так:

Код

import sqlite3


def test_db(db):
    try:
        db_cursor = db.cursor()
        db_cursor.execute("CREATE TABLE IF NOT EXISTS users (login TEXT, password TEXT)")
        db.commit()
        return True
    except DBError:  # тут надо правильную ошибку правильно обрабатывать, я не копался, какую именно и как
        return False


def add_user_into_db(db, login, password):
    if test_db(db):
        добавить_пользователя(login, password)


def select_from_db(db, sel):
    if test_db(db):
        return список_пользователей(sel)


if __name__ == '__main__':
    login = 'my_login'
    password = 'my_password'
    sel = 'string_for_select'
    with sqlite3.connect('users_base.db') as db:
        add_user_into_db(db=db, login=login, password=password)
        print(select_from_db(db=db, sel=sel))

Это тоже не лучший вариант, по-хорошему надо уж делать через pythonway и ООП, т.е. через ORM SQLAlchemy и всё такое.

Comments

[PesyCorm]

Все же тогда оставлю with в самих функциях, т.к. они многократно вызываются из других файлов. А вот вынос проверки - это пригодится. Спасибо за помощь!