Skip to content
Snippets Groups Projects
Commit dd56ba5f authored by Bastien Abadie's avatar Bastien Abadie Committed by Erwan Rouchet
Browse files

Ensure django database wrappers always have a connection before creating a cursor

parent 22ca1d12
No related branches found
No related tags found
1 merge request!2272Ensure django database wrappers always have a connection before creating a cursor
......@@ -51,7 +51,13 @@ def run_pg_query(query, source_db):
Run a single Postgresql query and split the results into chunks.
When a name is given to a cursor, psycopg2 uses a server-side cursor; we just use a random string as a name.
"""
with connections[source_db].create_cursor(name=str(uuid.uuid4())) as pg_cursor:
db = connections[source_db]
# Make sure a connection is open and available for export databases
if source_db != "default" and db.connection is None:
db.connect()
with db.create_cursor(name=str(uuid.uuid4())) as pg_cursor:
pg_cursor.itersize = BATCH_SIZE
pg_cursor.execute(query)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment