From 3d0e81bd437d857426cab3d30ea5c9648ad50664 Mon Sep 17 00:00:00 2001
From: Brandon Rodriguez <brodriguez8774@gmail.com>
Date: Mon, 3 Oct 2022 10:43:08 -0400
Subject: [PATCH] Update record UPDATE query to return more useful result data
---
py_dbcn/connectors/core/records.py | 13 ++++++++++---
1 file changed, 10 insertions(+), 3 deletions(-)
diff --git a/py_dbcn/connectors/core/records.py b/py_dbcn/connectors/core/records.py
index 1bc484e..7fdc395 100644
--- a/py_dbcn/connectors/core/records.py
+++ b/py_dbcn/connectors/core/records.py
@@ -169,6 +169,7 @@ class BaseRecords:
# Replace original clause.
values_clause = updated_values_clause
+ orig_where_clause = where_clause
if len(where_clause) > 0:
where_clause = ' WHERE {0}'.format(where_clause)
@@ -177,9 +178,15 @@ class BaseRecords:
UPDATE {0}
SET {1}{2};
""".format(table_name, values_clause, where_clause)
- results = self._base.query.execute(query, display_query=display_query)
- if display_results:
- self._base.display.results('{0}'.format(results))
+ self._base.query.execute(query, display_query=display_query)
+
+ # Do a select to get the updated values as results.
+ results = self.select(
+ table_name,
+ where_clause=orig_where_clause,
+ display_query=False,
+ display_results=display_results,
+ )
return results
--
GitLab