Added support for psycopg3 along with psycopg2. #5011

This commit is contained in:
Khushboo Vashi
2023-02-15 11:31:29 +05:30
committed by Akshay Joshi
parent 7a4951f211
commit 5e0daccf76
635 changed files with 6500 additions and 1534 deletions

View File

@@ -35,13 +35,13 @@ def restore_schema(server, db_name, schema_name, sql_path):
)
old_isolation_level = connection.isolation_level
connection.set_isolation_level(0)
utils.set_isolation_level(connection, 0)
pg_cursor = connection.cursor()
with open(sql_path, 'r') as content_file:
sql = content_file.read()
pg_cursor.execute(sql)
connection.set_isolation_level(old_isolation_level)
utils.set_isolation_level(connection, old_isolation_level)
connection.commit()
SQL = """SELECT
@@ -95,10 +95,10 @@ def create_table(server, db_name, schema_id, table_name, query):
server['port'],
server['sslmode'])
old_isolation_level = connection.isolation_level
connection.set_isolation_level(0)
utils.set_isolation_level(connection, 0)
pg_cursor = connection.cursor()
pg_cursor.execute(query)
connection.set_isolation_level(old_isolation_level)
utils.set_isolation_level(connection, old_isolation_level)
connection.commit()
# Get 'oid' from newly created table
pg_cursor.execute("SELECT oid "