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