db-sdk 4.0.12__py3-none-any.whl

db_sdk/__init__.py

@@ -0,0 +1,19 @@
+import os
+from db_sdk.core.exceptions import ConfigurationError
+
+
+def load_config_from_env():
+    db_type = os.getenv("DB_TYPE")
+
+    if not db_type:
+        raise ConfigurationError("DB_TYPE environment variable not set.")
+
+    return {
+        "db_type": db_type,
+        "host": os.getenv("DB_HOST"),
+        "user": os.getenv("DB_USER"),
+        "password": os.getenv("DB_PASSWORD"),
+        "database": os.getenv("DB_NAME"),
+        "port": os.getenv("DB_PORT"),
+        "uri": os.getenv("DB_URI"),
+    }

db_sdk/connectors/mongodb.py

@@ -0,0 +1,15 @@
+from pymongo import MongoClient
+from ..core.base_connector import BaseConnector
+from ..core.exceptions import DatabaseConnectionError
+
+
+class MongoConnector(BaseConnector):
+
+    def connect(self):
+        try:
+            uri = self.config.get("uri")
+            client = MongoClient(uri)
+            return client
+
+        except Exception as e:
+            raise DatabaseConnectionError(f"MongoDB connection failed: {e}")

db_sdk/connectors/mysql.py

@@ -0,0 +1,20 @@
+import mysql.connector
+from ..core.base_connector import BaseConnector
+from ..core.exceptions import DatabaseConnectionError
+
+
+class MySQLConnector(BaseConnector):
+
+    def connect(self):
+        try:
+            connection = mysql.connector.connect(
+                host=self.config.get("host"),
+                user=self.config.get("user"),
+                password=self.config.get("password"),
+                database=self.config.get("database"),
+                port=self.config.get("port", 3306),
+            )
+            return connection
+
+        except mysql.connector.Error as e:
+            raise DatabaseConnectionError(f"MySQL connection failed: {e}")

db_sdk/connectors/postgres.py

@@ -0,0 +1,20 @@
+import psycopg2
+from ..core.base_connector import BaseConnector
+from ..core.exceptions import DatabaseConnectionError
+
+
+class PostgresConnector(BaseConnector):
+
+    def connect(self):
+        try:
+            connection = psycopg2.connect(
+                host=self.config.get("host"),
+                user=self.config.get("user"),
+                password=self.config.get("password"),
+                dbname=self.config.get("database"),
+                port=self.config.get("port", 5432),
+            )
+            return connection
+
+        except psycopg2.Error as e:
+            raise DatabaseConnectionError(f"PostgreSQL connection failed: {e}")

db_sdk/connectors/redis.py

@@ -0,0 +1,20 @@
+import redis
+from ..core.base_connector import BaseConnector
+from ..core.exceptions import DatabaseConnectionError
+
+
+class RedisConnector(BaseConnector):
+
+    def connect(self):
+        try:
+            connection = redis.Redis(
+                host=self.config.get("host"),
+                port=self.config.get("port", 6379),
+                password=self.config.get("password"),
+                decode_responses=True,
+            )
+            connection.ping()  # Test connection
+            return connection
+
+        except redis.RedisError as e:
+            raise DatabaseConnectionError(f"Redis connection failed: {e}")

db_sdk/core/base_connector.py

@@ -0,0 +1,11 @@
+class BaseConnector:
+    """
+    Base connector class.
+    All database connectors must inherit from this.
+    """
+
+    def __init__(self, config: dict):
+        self.config = config
+
+    def connect(self):
+        raise NotImplementedError("Connect method must be implemented by subclass.")

db_sdk/core/exceptions.py

@@ -0,0 +1,18 @@
+class UniversalDBException(Exception):
+    """Base exception for Universal DB SDK."""
+    pass
+
+
+class UnsupportedDatabaseError(UniversalDBException):
+    """Raised when unsupported database type is provided."""
+    pass
+
+
+class DatabaseConnectionError(UniversalDBException):
+    """Raised when connection fails."""
+    pass
+
+
+class ConfigurationError(UniversalDBException):
+    """Raised when configuration is invalid."""
+    pass

db_sdk/core/factory.py

@@ -0,0 +1,30 @@
+from db_sdk.connectors.mysql import MySQLConnector
+from db_sdk.connectors.postgres import PostgresConnector
+from db_sdk.connectors.mongodb import MongoConnector
+from db_sdk.connectors.redis import RedisConnector
+# from db_sdk.core.exceptions import UnsupportedDatabaseError
+from .exceptions import UnsupportedDatabaseError
+
+
+class ConnectionFactory:
+
+    CONNECTOR_MAPPING = {
+        "mysql": MySQLConnector,
+        "postgres": PostgresConnector,
+        "mongodb": MongoConnector,
+        "redis": RedisConnector,
+    }
+
+    @staticmethod
+    def create_connection(db_type: str, config: dict):
+        db_type = db_type.lower()
+
+        if db_type not in ConnectionFactory.CONNECTOR_MAPPING:
+            raise UnsupportedDatabaseError(
+                f"Unsupported database type: {db_type}"
+            )
+
+        connector_class = ConnectionFactory.CONNECTOR_MAPPING[db_type]
+        connector = connector_class(config)
+
+        return connector.connect()

db_sdk/test_connection.py

@@ -0,0 +1,15 @@
+from db_sdk.core.factory import ConnectionFactory
+
+config = {
+    "host": "192.168.137.9",
+    "user": "pwf",
+    "password": "JqYVmy5uPTzG",
+    "database": "pwf",
+    "port": 3306
+}
+
+try:
+    connection = ConnectionFactory.create_connection("mysql", config)
+    print("✅ Database connected successfully!")
+except Exception as e:
+    print("❌ Connection failed:", e)

db_sdk-4.0.12.dist-info/METADATA

@@ -0,0 +1,17 @@
+Metadata-Version: 2.4
+Name: db_sdk
+Version: 4.0.12
+Summary: A simple database connection SDK
+Author: Neha
+Requires-Python: >=3.8
+Requires-Dist: pymongo>=4.0.0
+Requires-Dist: mysql-connector-python>=8.0
+Requires-Dist: psycopg2-binary>=2.9.0
+Requires-Dist: redis>=4.0.0
+Requires-Dist: pytest>=7.0.0
+Requires-Dist: pytest-cov>=4.0.0
+Requires-Dist: mongomock>=4.1.2
+Dynamic: author
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

db_sdk-4.0.12.dist-info/RECORD

@@ -0,0 +1,17 @@
+db_sdk/__init__.py,sha256=8oYWCzSeU8jAHKIdO03T92VMmBHjgjLc7UZENoxrJR4,509
+db_sdk/test_connection.py,sha256=iEnbWKy5JE9PQx_Ltyrd-B46295uGzzCH-EyDRhqEDc,371
+db_sdk/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+db_sdk/config/settings.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+db_sdk/connectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+db_sdk/connectors/mongodb.py,sha256=0qp1pZme06h3U0NNzt_Q8ffBV1v6aLxuY2SFzTjUW3Y,423
+db_sdk/connectors/mysql.py,sha256=M8UkfrfIbCW4PwytQYuisi4QvxHx49BrfbOHYyOuQx8,665
+db_sdk/connectors/postgres.py,sha256=0XXKbHdQJWPX95pfaU14Jn7eEnJVjOeovYA55w0alb0,650
+db_sdk/connectors/redis.py,sha256=WwVqKvJkk9DPwQg0dPy5SJmQh4T23_fELGy0uApsyoE,626
+db_sdk/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+db_sdk/core/base_connector.py,sha256=IoOpxqUPgNDsOK2om755qF0Mz-zrQIplXBngGeMQsLk,291
+db_sdk/core/exceptions.py,sha256=R9q4_FkxaZhyf24Cwmg8pfs0dh5sFoXf_Ih88bdr2kA,431
+db_sdk/core/factory.py,sha256=hrydt3J3S2YrYgoGo1iBjG_E8dFvvbX_QwhkbdSrFpo,960
+db_sdk-4.0.12.dist-info/METADATA,sha256=zurWB8bG8HdzH1zCXoV6KaDwNoShma4rvguuy1kdQ7k,443
+db_sdk-4.0.12.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+db_sdk-4.0.12.dist-info/top_level.txt,sha256=sWeCT6R-hslk8L6onXix41smZ45S8k6LmYDErKvC8xA,7
+db_sdk-4.0.12.dist-info/RECORD,,

db_sdk-4.0.12.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

db_sdk-4.0.12.dist-info/top_level.txt

@@ -0,0 +1 @@
+db_sdk