[AIRFLOW-533] Set autocommit via set_autocommit

Delegate setting autocommit in insert_rows to set_autocommit Closes apache#1813 from thyming/fix-insert-rows- autocommit
dmartin11 · Nov 15, 2016 · f34b292 · f34b292
1 parent 12e48b4
commit f34b292
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airflow/hooks/dbapi_hook.py b/airflow/hooks/dbapi_hook.py
@@ -203,10 +203,10 @@ def insert_rows(self, table, rows, target_fields=None, commit_every=1000):
         else:
             target_fields = ''
         conn = self.get_conn()
-        cur = conn.cursor()
         if self.supports_autocommit:
-            cur.execute('SET autocommit = 0')
+            self.set_autocommit(conn, False)
         conn.commit()
+        cur = conn.cursor()
         i = 0
         for row in rows:
             i += 1