feat: autocommit session

2023-06-25 12:22:06 +02:00 · 2023-06-25 12:22:06 +02:00 · 4a3d43bcc5
parent 915f3c2a4c
commit 4a3d43bcc5
1 changed files with 2 additions and 1 deletions
--- a/etl/helpers/data_io.py
+++ b/etl/helpers/data_io.py
@ -47,7 +47,8 @@ class WriteToPostgreSQL(beam.DoFn):
        self.table_key = table_key

    def setup(self):
-        self.connection = psycopg2.connect(**self.connection_details, autocommit=True)
+        self.connection = psycopg2.connect(**self.connection_details)
+        self.connection.autocommit = True

    def execute_insert(self, row, cursor):
        colnames = ",".join(row.keys())