Changed yield value

crate · Jul 18, 2022 · 9a24409 · 9a24409
1 parent 024967a
commit 9a24409
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/crate/client/cursor.py b/src/crate/client/cursor.py
@@ -50,10 +50,14 @@ def execute(self, sql, parameters=None, bulk_parameters=None):
 
         self._result = self.connection.client.sql(sql, parameters,
                                                   bulk_parameters)
+
         if "rows" in self._result:
+            transformed_result = False
             if "col_types" in self._result:
+                transformed_result = True
                 self.rows = self.result_set_transformed()
-            if self.rows is None:
+
+            if not transformed_result:
                 self.rows = iter(self._result["rows"])
 
     def result_set_transformed(self):
@@ -64,7 +68,7 @@ def result_set_transformed(self):
                            self._result["col_types"]]
         for row in self._result["rows"]:
             gen_flags = (flag for flag in rows_to_convert)
-            yield self._transform_date_columns(row, gen_flags)
+            yield [t_row for t_row in self._transform_date_columns(row, gen_flags)]
 
     @staticmethod
     def _transform_date_columns(row, gen_flags):