Loading app/helpers/db.py +11 −14 Original line number Diff line number Diff line Loading @@ -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 Loading
app/helpers/db.py +11 −14 Original line number Diff line number Diff line Loading @@ -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