chore: pre-commit run

Author: ehddnr301

db_utils/base_connector.py

@@ -1,6 +1,7 @@
 from abc import ABC, abstractmethod
 import pandas as pd
 
+
 class BaseConnector(ABC):
     """
     Abstract base class for database connectors.

db_utils/config.py

@@ -1,7 +1,8 @@
 from typing import Optional, Dict, TypedDict
 
+
 class DBConfig(TypedDict):
-	
+
     host: str
     port: Optional[int]
     user: Optional[str]

db_utils/databricks_connector.py

@@ -4,10 +4,12 @@
 from .config import DBConfig
 from .logger import logger
 
+
 class DatabricksConnector(BaseConnector):
     """
     Connect to Databricks SQL Warehouse and execute queries.
     """
+
     connection = None
 
     def __init__(self, config: DBConfig):
@@ -75,4 +77,4 @@ def close(self) -> None:
         if self.connection:
             self.connection.close()
             logger.error("Connection to Databricks closed.")
-        self.connection = None
+        self.connection = None

db_utils/duckdb_connector.py

@@ -4,6 +4,7 @@
 from .config import DBConfig
 from .logger import logger
 
+
 class DuckDBConnector(BaseConnector):
     """
     Connect to DuckDB and execute SQL queries.
@@ -56,4 +57,4 @@ def close(self) -> None:
         if self.connection:
             self.connection.close()
             logger.error("Connection to DuckDB closed.")
-        self.connection = None
+        self.connection = None

db_utils/mariadb_connector.py

@@ -4,11 +4,13 @@
 from .config import DBConfig
 from .logger import logger
 
+
 class MariaDBConnector(BaseConnector):
     """
     Connect to MariaDB and execute SQL queries.
     This class uses **mysql-connector-python** to connect to the MariaDB server.
     """
+
     connection = None
 
     def __init__(self, config: DBConfig):
@@ -35,7 +37,7 @@ def connect(self) -> None:
                 port=self.port,
                 user=self.user,
                 password=self.password,
-                database=self.database
+                database=self.database,
             )
             logger.info("Successfully connected to MariaDB.")
         except Exception as e:
@@ -71,4 +73,4 @@ def close(self) -> None:
         if self.connection:
             self.connection.close()
             print("Connection to MariaDB closed.")
-        self.connection = None
+        self.connection = None

db_utils/mysql_connector.py

@@ -4,10 +4,12 @@
 from .config import DBConfig
 from .logger import logger
 
+
 class MySQLConnector(BaseConnector):
     """
     Connect to MySQL and execute SQL queries.
     """
+
     connection = None
 
     def __init__(self, config: DBConfig):
@@ -34,7 +36,7 @@ def connect(self) -> None:
                 port=self.port,
                 user=self.user,
                 password=self.password,
-                database=self.database
+                database=self.database,
             )
             logger.info("Successfully connected to MySQL.")
         except Exception as e:

db_utils/oracle_connector.py

@@ -4,11 +4,12 @@
 from .config import DBConfig
 from .logger import logger
 
-class OracleConnector(BaseConnector):
 
+class OracleConnector(BaseConnector):
     """
     Connect to Oracle database and execute SQL queries.
     """
+
     connection = None
 
     def __init__(self, config: DBConfig):
@@ -63,4 +64,4 @@ def close(self) -> None:
         if self.connection:
             self.connection.close()
             logger.error("Connection to Oracle closed.")
-        self.connection = None
+        self.connection = None

db_utils/postgres_connector.py

@@ -4,6 +4,7 @@
 from .config import DBConfig
 from .logger import logger
 
+
 class PostgresConnector(BaseConnector):
     """
     Connect to PostgreSQL and execute SQL queries.
@@ -71,4 +72,4 @@ def close(self) -> None:
         if self.connection:
             self.connection.close()
             logger.info("Connection to PostgreSQL closed.")
-        self.connection = None
+        self.connection = None

db_utils/snowflake_connector.py

@@ -4,10 +4,12 @@
 from .config import DBConfig
 from .logger import logger
 
+
 class SnowflakeConnector(BaseConnector):
     """
     Connect to Snowflake database and execute SQL queries.
     """
+
     connection = None
 
     def __init__(self, config: DBConfig):
@@ -79,4 +81,4 @@ def close(self) -> None:
         if self.connection:
             self.connection.close()
             logger.info("Connection to Snowflake closed.")
-        self.connection = None
+        self.connection = None

db_utils/sqlite_connector.py

@@ -4,10 +4,12 @@
 from .config import DBConfig
 from .logger import logger
 
+
 class SQLiteConnector(BaseConnector):
     """
     Connect to SQLite and execute SQL queries.
     """
+
     connection = None
 
     def __init__(self, config: DBConfig):
@@ -65,4 +67,4 @@ def close(self) -> None:
         if self.connection:
             self.connection.close()
             logger.info("Connection to SQLite closed.")
-        self.connection = None
+        self.connection = None