db-sdk 4.0.12__tar.gz

db_sdk-4.0.12/PKG-INFO

@@ -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/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-4.0.12/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-4.0.12/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-4.0.12/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-4.0.12/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-4.0.12/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-4.0.12/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-4.0.12/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-4.0.12/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/db_sdk.egg-info/PKG-INFO

@@ -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/db_sdk.egg-info/SOURCES.txt

@@ -0,0 +1,22 @@
+README.md
+setup.py
+db_sdk/__init__.py
+db_sdk/test_connection.py
+db_sdk.egg-info/PKG-INFO
+db_sdk.egg-info/SOURCES.txt
+db_sdk.egg-info/dependency_links.txt
+db_sdk.egg-info/requires.txt
+db_sdk.egg-info/top_level.txt
+db_sdk/config/__init__.py
+db_sdk/config/settings.py
+db_sdk/connectors/__init__.py
+db_sdk/connectors/mongodb.py
+db_sdk/connectors/mysql.py
+db_sdk/connectors/postgres.py
+db_sdk/connectors/redis.py
+db_sdk/core/__init__.py
+db_sdk/core/base_connector.py
+db_sdk/core/exceptions.py
+db_sdk/core/factory.py
+tests/test_connection.py
+tests/test_connection_1.py

db_sdk-4.0.12/db_sdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

db_sdk-4.0.12/db_sdk.egg-info/requires.txt

@@ -0,0 +1,7 @@
+pymongo>=4.0.0
+mysql-connector-python>=8.0
+psycopg2-binary>=2.9.0
+redis>=4.0.0
+pytest>=7.0.0
+pytest-cov>=4.0.0
+mongomock>=4.1.2

db_sdk-4.0.12/db_sdk.egg-info/top_level.txt

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

db_sdk-4.0.12/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

db_sdk-4.0.12/setup.py

@@ -0,0 +1,21 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="db_sdk",
+    version="4.0.12",
+    packages=find_packages(),
+    install_requires=[
+        "pymongo>=4.0.0",
+            "mysql-connector-python>=8.0",
+            "psycopg2-binary>=2.9.0",
+            "redis>=4.0.0",
+            "pytest>=7.0.0",
+            "pytest-cov>=4.0.0",
+            "mongomock>=4.1.2",
+    ],
+    author="Neha",
+    description="A simple database connection SDK",
+    python_requires=">=3.8",
+)
+
+# Comment: This setup.py file defines the package metadata and dependencies for the auth-sdk. It specifies that the package requires Python 3.8 or higher and depends on bcrypt, PyJWT, and python-dotenv. The package is named "auth-sdk" and is currently at version 0.1.0.

db_sdk-4.0.12/tests/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/tests/test_connection_1.py

@@ -0,0 +1,37 @@
+from unittest.mock import patch, MagicMock
+from db_sdk.connectors.mongodb import MongoConnector
+
+def test_factory_creates_mongo_connection():
+    print("\n--- Starting test ---")
+    
+    # Step 1: Create a fake MongoDB client
+    mock_client = MagicMock()
+    print(f"Mock client created: {mock_client}")
+
+    # Step 2: Patch MongoClient so no real connection is made
+    with patch("db_sdk.connectors.mongodb.MongoClient", return_value=mock_client) as mock_mongo:
+        print(f"MongoClient is now patched: {mock_mongo}")
+
+        # Step 3: Create the connector with a fake URI
+        config = {"uri": "mongodb://fake-host:9999"}
+        connector = MongoConnector(config)
+        print(f"Connector created: {connector}")
+        print(f"Connector config: {connector.config}")
+
+        # Step 4: Call connect() - this triggers MongoClient internally
+        client = connector.connect()
+        print(f"connect() was called, returned: {client}")
+
+        # Step 5: Verify MongoClient was called with the fake URI
+        mock_mongo.assert_called_once_with("mongodb://fake-host:9999")
+        print(f"MongoClient was called with URI: {mock_mongo.call_args}")
+
+        # Step 6: Verify the returned client is our mock
+        assert client is mock_client
+        print(f"Returned client matches mock_client: {client is mock_client}")
+
+    print("--- Test passed ---")
+
+
+if __name__ == "__main__":
+    test_factory_creates_mongo_connection()