python-databases 0.1.1__tar.gz → 0.1.3__tar.gz

python_databases-0.1.3/PKG-INFO

@@ -0,0 +1,75 @@
+Metadata-Version: 2.4
+Name: python-databases
+Version: 0.1.3
+Summary: A Python library for managing and interacting with multiple databases seamlessly.
+Home-page: https://github.com/aviz92/python-databases
+Author: Avi Zaguri
+Author-email: 
+Project-URL: Repository, https://github.com/aviz92/python-databases
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: setuptools
+Requires-Dist: wheel
+Requires-Dist: dotenv
+Requires-Dist: custom-python-logger
+Requires-Dist: retrying
+Requires-Dist: elasticsearch<9
+Requires-Dist: pandas
+Requires-Dist: psycopg2-binary
+Requires-Dist: pytest
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Python Databases
+A flexible and extensible Python library that provides a unified interface for interacting with multiple types of databases.
+
+---
+
+## Features
+- ✅ Unified interface for multiple database connectors
+- ✅ Support for:
+  - Elasticsearch
+    - On-Premises deployments
+    - Cloud deployments using `cloud_id`
+  - PostgreSQL
+    - Basic support with plans for SQLAlchemy integration
+
+---
+
+## Installation
+```bash
+pip install python-databases
+```
+
+---
+
+## 📌 Roadmap
+
+* [x] Elasticsearch support
+  - [x] On-Premises deployments
+  - [x] Cloud deployments using `cloud_id`
+* [x] PostgreSQL
+  - [ ] SQLAlchemy support
+
+---
+
+## 🤝 Contributing
+If you have a helpful tool, pattern, or improvement to suggest:
+Fork the repo <br>
+Create a new branch <br>
+Submit a pull request <br>
+I welcome additions that promote clean, productive, and maintainable development. <br>
+
+---
+
+## 🙏 Thanks
+Thanks for exploring this repository! <br>
+Happy coding! <br>

python_databases-0.1.3/README.md

@@ -0,0 +1,45 @@
+# Python Databases
+A flexible and extensible Python library that provides a unified interface for interacting with multiple types of databases.
+
+---
+
+## Features
+- ✅ Unified interface for multiple database connectors
+- ✅ Support for:
+  - Elasticsearch
+    - On-Premises deployments
+    - Cloud deployments using `cloud_id`
+  - PostgreSQL
+    - Basic support with plans for SQLAlchemy integration
+
+---
+
+## Installation
+```bash
+pip install python-databases
+```
+
+---
+
+## 📌 Roadmap
+
+* [x] Elasticsearch support
+  - [x] On-Premises deployments
+  - [x] Cloud deployments using `cloud_id`
+* [x] PostgreSQL
+  - [ ] SQLAlchemy support
+
+---
+
+## 🤝 Contributing
+If you have a helpful tool, pattern, or improvement to suggest:
+Fork the repo <br>
+Create a new branch <br>
+Submit a pull request <br>
+I welcome additions that promote clean, productive, and maintainable development. <br>
+
+---
+
+## 🙏 Thanks
+Thanks for exploring this repository! <br>
+Happy coding! <br>

python_databases-0.1.3/python_databases/postgresql_infrastructure/__init__.py

@@ -0,0 +1,7 @@
+from dotenv import load_dotenv
+
+from python_databases.postgresql_infrastructure.postgresql import PostgreSQL
+
+load_dotenv()
+
+__all__ = ['PostgreSQL']

python_databases-0.1.3/python_databases/postgresql_infrastructure/postgresql.py

@@ -0,0 +1,83 @@
+import logging
+import os
+from typing import Optional, Union, Any
+import psycopg2
+from psycopg2 import extras
+from retrying import retry
+
+
+class PostgreSQL:
+    def __init__(
+        self,
+        name: Optional[str] = None,
+        host: Optional[str] = None,
+        port: Optional[str] = None,
+        username: Optional[str] = None,
+        password: Optional[str] = None
+    ):
+        self.logger = logging.getLogger(self.__class__.__name__)
+
+        self.name = name or os.getenv("POSTGRESQL_NAME")
+        self.host = host or os.getenv("POSTGRESQL_HOST")
+        self.port = port or os.getenv("POSTGRESQL_PORT")
+        self.username = username or os.getenv("POSTGRESQL_USERNAME")
+        self.password = password or os.getenv("POSTGRESQL_PASSWORD")
+
+        self._client = None
+        self._cursor = None
+
+        self.create_connection()
+
+    @property
+    def client(self):
+        if not self._client or self._client.closed:
+            self.logger.warning("Re-establishing PostgreSQL connection")
+            self.create_connection()
+        return self._client
+
+    @property
+    def cursor(self):
+        if not self._cursor or self._cursor.closed:
+            self.logger.warning("Re-creating PostgreSQL cursor")
+            self.create_connection()
+        return self._cursor
+
+    # @retry(stop_max_attempt_number=3, wait_fixed=180000)
+    def create_connection(self):
+        self.logger.info(f"Create connection to {self.host}:{self.port}")
+
+        self._client = psycopg2.connect(
+            database=self.name,
+            host=self.host,
+            port=self.port,
+            user=self.username,
+            password=self.password,
+        )
+        self._cursor = self._client.cursor(cursor_factory=extras.DictCursor)
+
+    def execute_query(self, query: str, params: Optional[Union[tuple, dict[str, Any]]] = None) -> None:
+        self.logger.debug(f"Fetch rows from DB by query {query} with params {params}")
+        try:
+            self.cursor.execute(query, params)
+        except psycopg2.ProgrammingError as e:
+            raise psycopg2.ProgrammingError(f"Error executing query: {query}") from e
+        self.logger.debug(f"Finished to fetch rows by query")
+
+    def fetch_all(self, query: str, params: Optional[Union[tuple, dict[str, Any]]] = None) -> list[dict[str, Any]]:
+        self.execute_query(query, params)
+        try:
+            results = self.cursor.fetchall()
+        except psycopg2.ProgrammingError as e:
+            raise psycopg2.ProgrammingError(f"Error fetching rows: {query}") from e
+        self.logger.debug(f"Fetched {len(results)} rows")
+        return results
+
+    def close(self):
+        self.cursor.close()
+        self.client.close()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()

python_databases-0.1.3/python_databases.egg-info/PKG-INFO

@@ -0,0 +1,75 @@
+Metadata-Version: 2.4
+Name: python-databases
+Version: 0.1.3
+Summary: A Python library for managing and interacting with multiple databases seamlessly.
+Home-page: https://github.com/aviz92/python-databases
+Author: Avi Zaguri
+Author-email: 
+Project-URL: Repository, https://github.com/aviz92/python-databases
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: setuptools
+Requires-Dist: wheel
+Requires-Dist: dotenv
+Requires-Dist: custom-python-logger
+Requires-Dist: retrying
+Requires-Dist: elasticsearch<9
+Requires-Dist: pandas
+Requires-Dist: psycopg2-binary
+Requires-Dist: pytest
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Python Databases
+A flexible and extensible Python library that provides a unified interface for interacting with multiple types of databases.
+
+---
+
+## Features
+- ✅ Unified interface for multiple database connectors
+- ✅ Support for:
+  - Elasticsearch
+    - On-Premises deployments
+    - Cloud deployments using `cloud_id`
+  - PostgreSQL
+    - Basic support with plans for SQLAlchemy integration
+
+---
+
+## Installation
+```bash
+pip install python-databases
+```
+
+---
+
+## 📌 Roadmap
+
+* [x] Elasticsearch support
+  - [x] On-Premises deployments
+  - [x] Cloud deployments using `cloud_id`
+* [x] PostgreSQL
+  - [ ] SQLAlchemy support
+
+---
+
+## 🤝 Contributing
+If you have a helpful tool, pattern, or improvement to suggest:
+Fork the repo <br>
+Create a new branch <br>
+Submit a pull request <br>
+I welcome additions that promote clean, productive, and maintainable development. <br>
+
+---
+
+## 🙏 Thanks
+Thanks for exploring this repository! <br>
+Happy coding! <br>

{python_databases-0.1.1 → python_databases-0.1.3}/python_databases.egg-info/SOURCES.txt

@@ -11,4 +11,7 @@ python_databases.egg-info/dependency_links.txt
 python_databases.egg-info/requires.txt
 python_databases.egg-info/top_level.txt
 python_databases/elasticsearch_infrastructure/__init__.py
-python_databases/elasticsearch_infrastructure/elasticsearch.py
+python_databases/elasticsearch_infrastructure/elasticsearch.py
+python_databases/postgresql_infrastructure/__init__.py
+python_databases/postgresql_infrastructure/postgresql.py
+tests/test_postgres.py

{python_databases-0.1.1 → python_databases-0.1.3}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-package_version = '0.1.1'
+package_version = '0.1.3'
 
 package_name = 'python-databases'
 package_description = 'A Python library for managing and interacting with multiple databases seamlessly.'

python_databases-0.1.3/tests/test_postgres.py

@@ -0,0 +1,46 @@
+# test_postgresql.py
+
+import pytest
+from unittest.mock import patch, MagicMock
+from python_databases.postgresql_infrastructure import PostgreSQL
+
+
+@pytest.fixture
+def mock_connection():
+    with patch("psycopg2.connect") as mock_connect:
+        mock_conn = MagicMock()
+        mock_cursor = MagicMock()
+        mock_conn.cursor.return_value = mock_cursor
+        mock_connect.return_value = mock_conn
+        yield mock_conn, mock_cursor
+
+
+def test_connection_created(mock_connection):
+    conn, _ = mock_connection
+    db = PostgreSQL(name="test_db", host="localhost", port="5432", username="user", password="pass")
+    assert db.client == conn
+    assert conn.cursor.call_count == 2
+
+
+def test_execute_query_success(mock_connection):
+    _, cursor = mock_connection
+    db = PostgreSQL(name="test_db", host="localhost", port="5432", username="user", password="pass")
+    db.execute_query("SELECT 1", None)
+    cursor.execute.assert_called_with("SELECT 1", None)
+
+
+def test_fetch_all_success(mock_connection):
+    _, cursor = mock_connection
+    cursor.fetchall.return_value = [{"id": 1}]
+    db = PostgreSQL(name="test_db", host="localhost", port="5432", username="user", password="pass")
+    results = db.fetch_all("SELECT id FROM test")
+    cursor.execute.assert_called_with("SELECT id FROM test", None)
+    assert results == [{"id": 1}]
+
+
+def test_context_manager_closes_connection(mock_connection):
+    conn, cursor = mock_connection
+    with PostgreSQL(name="test_db", host="localhost", port="5432", username="user", password="pass") as db:
+        pass
+    cursor.close.assert_called_once()
+    conn.close.assert_called_once()

python_databases-0.1.1/PKG-INFO

@@ -1,31 +0,0 @@
-Metadata-Version: 2.4
-Name: python-databases
-Version: 0.1.1
-Summary: A Python library for managing and interacting with multiple databases seamlessly.
-Home-page: https://github.com/aviz92/python-databases
-Author: Avi Zaguri
-Author-email: 
-Project-URL: Repository, https://github.com/aviz92/python-databases
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: setuptools
-Requires-Dist: wheel
-Requires-Dist: dotenv
-Requires-Dist: custom-python-logger
-Requires-Dist: retrying
-Requires-Dist: elasticsearch<9
-Requires-Dist: pandas
-Requires-Dist: psycopg2-binary
-Requires-Dist: pytest
-Dynamic: author
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: project-url
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# Python Databases

python_databases-0.1.1/README.md

@@ -1 +0,0 @@
-# Python Databases

python_databases-0.1.1/python_databases.egg-info/PKG-INFO

@@ -1,31 +0,0 @@
-Metadata-Version: 2.4
-Name: python-databases
-Version: 0.1.1
-Summary: A Python library for managing and interacting with multiple databases seamlessly.
-Home-page: https://github.com/aviz92/python-databases
-Author: Avi Zaguri
-Author-email: 
-Project-URL: Repository, https://github.com/aviz92/python-databases
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: setuptools
-Requires-Dist: wheel
-Requires-Dist: dotenv
-Requires-Dist: custom-python-logger
-Requires-Dist: retrying
-Requires-Dist: elasticsearch<9
-Requires-Dist: pandas
-Requires-Dist: psycopg2-binary
-Requires-Dist: pytest
-Dynamic: author
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: project-url
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# Python Databases