Commit fb276e7a authored by Anthony Jacob's avatar Anthony Jacob
Browse files

another way of trying

parent ed4fbfc1
Loading
Loading
Loading
Loading
+11 −14
Original line number Diff line number Diff line
@@ -6,24 +6,21 @@ from psycopg.rows import dict_row
def db_cursor(dict_results=False):
    try:
        conn = current_app.db_pool.getconn()
        cur = conn.cursor()
        cur = conn.cursor(row_factory=dict_row if dict_results else None)

        # --- Keepalive query ---
        cur.execute("SELECT 1;")
        # ----------------------
        if dict_results:
            cur = conn.cursor(row_factory=dict_row)

        yield conn, cur

        # with current_app.db_pool.connection() as conn:
        #     cur = conn.cursor()
        #     # --- Keepalive query ---
        #     cur.execute("SELECT 1;")
        #     with conn.cursor(row_factory=dict_row if dict_results else None) as cur:
        #         yield conn, cur
    except Exception as e:
        current_app.logger.error(f"Database error: {e}")
    finally:
        if cur:
            cur.close()
        if conn:
            current_app.db_pool.putconn(conn)
 No newline at end of file
        current_app.db_pool.putconn(conn, close=True)
        with current_app.db_pool.connection() as conn:
            cur = conn.cursor()
            # --- Keepalive query ---
            cur.execute("SELECT 1;")
            with conn.cursor(row_factory=dict_row if dict_results else None) as cur:
                yield conn, cur