wip

klondike/base/generic_db.py

@@ -4,20 +4,33 @@
 
 import petl
 from polars import DataFrame
+from contextlib import contextmanager
 
 ##########
 
 
-class PolarBareDB:
+class KlondikeDB:
     """
     Abstract class with shared utilities across all database instances
     """
 
-    def __init__(self):
+    def __init__(self, **kwargs):
+        vars(self).update(kwargs=kwargs)
+
+        self.dialect = None
         self.__tempfiles = []
 
+    @contextmanager
+    def connection(self):
+        """Placeholder function"""
+        return
+
     def query(
-        self, sql: str, parameters: Optional[list] = None, batch_size: int = 250_000
+        self,
+        sql: str,
+        parameters: Optional[list] = None,
+        batch_size: int = 250_000,
+        return_values: bool = True,
     ) -> Optional[DataFrame]:
         """
         Leverages internal connection to query against Postgres instance
@@ -36,7 +49,7 @@ def query(
                 cursor.execute(query=sql, params=parameters)
 
                 # If no results, return without raising exeption
-                if not cursor.description:
+                if not cursor.description or not return_values:
                     return None
 
                 ###

klondike/bigquery/bigquery.py

@@ -4,7 +4,7 @@
 from google.cloud import bigquery
 from google.cloud.bigquery import dbapi
 
-from klondike.base.generic_db import PolarBareDB
+from klondike.base.generic_db import KlondikeDB
 import json
 import os
 import tempfile
@@ -20,7 +20,7 @@
 )
 
 
-class PolarBigQuery(PolarBareDB):
+class PolarBigQuery(KlondikeDB):
     """
     Establish and authenticate a connection to a BigQuery warehouse
     """

klondike/postgres/postgres.py

@@ -4,12 +4,12 @@
 
 import psycopg
 
-from klondike.base.generic_db import PolarBareDB
+from klondike.base.generic_db import KlondikeDB
 
 ##########
 
 
-class PolarPostgres(PolarBareDB):
+class PolarPostgres(KlondikeDB):
     """
     Establish and authenticate a connection to a Postgres instance.