Use process_sql for ownership.

iangow · Jan 11, 2024 · 1fc16df · 1fc16df
1 parent 9ba220d
commit 1fc16df
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/wrds2pg/wrds2pg.py b/wrds2pg/wrds2pg.py
@@ -752,12 +752,11 @@ def wrds_update(table_name, schema,
                 create_role(engine, schema)
 
             sql = f'ALTER TABLE "{schema}"."{alt_table_name}" OWNER TO {schema}'
-            with engine.connect() as conn:
-                conn.execute(text(sql))
+            process_sql(sql, engine)
 
             if not role_exists(engine, "%s_access" % schema):
                 create_role(engine, "%s_access" % schema)
-                
+
             sql = f'GRANT SELECT ON "{schema}"."{alt_table_name}"'
             sql += f' TO {schema}_access'
             res = process_sql(sql, engine)