From 9a2440920e2381b2179416b55ce087bbbf6268ba Mon Sep 17 00:00:00 2001 From: Aymaru Date: Wed, 31 Mar 2021 10:44:18 -0600 Subject: [PATCH] Changed yield value --- src/crate/client/cursor.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/crate/client/cursor.py b/src/crate/client/cursor.py index 39475e173..26a9faad0 100644 --- 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):