Как уменьшить вложенность, или делаю ли я что-то не так?

Question

[Илхомжон Матазимов]

Всем привет!
У меня есть говнокод, где я взаимодействую с БД, а именно PostgreSQL. Библиотека: Psycopg2.
Дело в том, что вложенность говнокода получается достаточно большим, и мне кажется, что я делаю что-то не так.
Вот код:

with psycopg2.connect('данные бд') as conn:
    with conn.cursor() as cursor:
        # для примера:
        cursor.execute('SELECT name FROM users WHERE id = %s', [id])
        result_name = [x[0] for x in cursor.fetchall()]
        with psycopg2.connect('данные бд') as conn: # получается вложенность
            with conn.cursor() as cursor: # без этой вложенности - получается ошибка
                cursor.execute('SELECT name FROM confirm WHERE name = %s', [result_name[0]])
                result_confirm = [x[0] for x in cursor.fetchall()]
                if result_confirm.__len__() == 0:
                    ...

Если вот эти вложенности не прописывать, то выходит ошибка: connection already is closed, поэтому приходится так засорять и так уже дерьмовый код.
Как это исправить? Или все так и должно быть?

Comments

[ayazer]

а почему бы не доставать все данные 1им запросом вообще?

[Илхомжон Матазимов]

ayazer, Эти запросы просто тут для примера, на самом деле же без разных запросов не обойтись

Answer 1

[Ivan Yakushenko]

Не совсем понятно на примере что именно не так, но такой вариант должен работать:

with psycopg2.connect('данные бд') as conn:
    with conn.cursor() as cursor:
        cursor.execute('SELECT name FROM users WHERE id = %s', [id])
        result_name = [x[0] for x in cursor.fetchall()]
    with conn.cursor() as cursor:
        cursor.execute('SELECT name FROM confirm WHERE name = %s', [result_name[0]])
        result_confirm = [x[0] for x in cursor.fetchall()]

Возможно вы где то выходите из блока, или закрываете соединение самостоятельно?
А вообще, я бы порекомендовал использовать SQLAlchemy, он простой как палка, но работать с БД намного проще.

Comments

[Илхомжон Матазимов]

Спасибо! Получилось чуть-чуть уменьшить вложенность, с Вашими указаниями.
Это предел уменьшения, или еще как-то можно уменьшить?

[Ivan Yakushenko]

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

[Илхомжон Матазимов]

Ivan Yakushenko, Спасибо огромное!