Fixed PostgresToGCSOperator fail on empty resultset for use_server_si…

…de_cursor=True (#21307) Fixed issue #20007
apache · Feb 15, 2022 · 2eb1056 · 2eb1056
1 parent 00f0025
commit 2eb1056
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/providers/google/cloud/transfers/postgres_to_gcs.py b/airflow/providers/google/cloud/transfers/postgres_to_gcs.py
@@ -59,7 +59,8 @@ def description(self):
  """Fetch first row to initialize cursor description when using server side cursor."""
  if not self.initialized:
  element = self.cursor.fetchone()
- self.rows.append(element)
+ if element is not None:
+ self.rows.append(element)
  self.initialized = True
  return self.cursor.description