From f3af1fb472576266542453fb20e0fd9a62699def Mon Sep 17 00:00:00 2001
From: Brandon Rodriguez <brodriguez8774@gmail.com>
Date: Mon, 21 Nov 2022 08:19:32 -0500
Subject: [PATCH] Update base query function
---
py_dbcn/connectors/core/query.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/py_dbcn/connectors/core/query.py b/py_dbcn/connectors/core/query.py
index 6d65b13..1fdffc4 100644
--- a/py_dbcn/connectors/core/query.py
+++ b/py_dbcn/connectors/core/query.py
@@ -31,7 +31,7 @@ class BaseQuery:
# Define provided direct parent object.
self._parent = parent
- def execute(self, query, display_query=True):
+ def execute(self, query, data=None, display_query=True):
"""Core function to execute database queries.
:param query: Query to execute.
@@ -40,12 +40,18 @@ class BaseQuery:
if display_query:
self._base.display.query(query)
+ if isinstance(data, str):
+ data = [data]
+
# Create connection and execute query.
cursor = self._base._connection.cursor()
# Improve query speed if PostgreSQL (supposedly. Needs more research).
if self._base._config.db_type == 'PostgreSQL':
query = cursor.mogrify(query)
- cursor.execute(query)
+ if data is not None:
+ cursor.execute(query, data)
+ else:
+ cursor.execute(query)
# Get results.
results = self._fetch_results(cursor)
--
GitLab