From ab282c6a5d4c5b3f138be351308af02a78e03d4e Mon Sep 17 00:00:00 2001
From: Brandon Rodriguez <brodriguez8774@gmail.com>
Date: Mon, 10 Oct 2022 13:14:24 -0400
Subject: [PATCH] Cleanup of leftovers from last commit

---
 py_dbcn/connectors/core/query.py   | 13 ++++++++++++-
 py_dbcn/connectors/core/records.py |  2 --
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/py_dbcn/connectors/core/query.py b/py_dbcn/connectors/core/query.py
index 642f7ee..6d65b13 100644
--- a/py_dbcn/connectors/core/query.py
+++ b/py_dbcn/connectors/core/query.py
@@ -42,6 +42,9 @@ class BaseQuery:
 
         # 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)
 
         # Get results.
@@ -57,12 +60,20 @@ class BaseQuery:
         return results
 
     def execute_many(self, query, data, display_query=True):
-        """"""
+        """Execute method to run multiple queries in one call.
+
+        :param query: Query to execute.
+        :param data: One or more sets of data to pass into query.
+        :param display_query: Optional bool indicating if query should output to console or not. Defaults to True.
+        """
         if display_query:
             self._base.display.query(query)
 
         # 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.executemany(query, data)
 
         # Get results.
diff --git a/py_dbcn/connectors/core/records.py b/py_dbcn/connectors/core/records.py
index d3f570c..5a9d79d 100644
--- a/py_dbcn/connectors/core/records.py
+++ b/py_dbcn/connectors/core/records.py
@@ -341,8 +341,6 @@ class BaseRecords:
         #     display_results=display_results,
         # )
 
-        self.select(table_name)
-
         return results
 
     def delete(self, table_name, where_clause, display_query=True, display_results=True):
-- 
GitLab