python-databases 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

python_databases/elasticsearch_infrastructure/elasticsearch.py

@@ -6,6 +6,7 @@ from datetime import datetime, timezone
 from enum import Enum
 from typing import Optional, Any
 import pandas as pd
+from custom_python_logger import get_logger
 from elasticsearch import Elasticsearch, helpers
 from retrying import retry
 
@@ -25,7 +26,7 @@ class ElasticSearch(ABC):
             username: Optional[str],
             password: Optional[str]
     ):
-        self.logger = logging.getLogger(self.__class__.__name__)
+        self.logger = get_logger(self.__class__.__name__)
         self._change_elasticsearch_logger()
 
         self.protocol = protocol
@@ -43,7 +44,7 @@ class ElasticSearch(ABC):
         self.elk_client = None
 
     def _change_elasticsearch_logger(self):
-        tracer = logging.getLogger('elasticsearch')
+        tracer = get_logger('elasticsearch')
         tracer.setLevel(logging.CRITICAL)  # or desired level
         # tracer.addHandler(logging.FileHandler('indexer.log'))
 
@@ -260,7 +261,7 @@ class ElasticSearchOnPrem(ElasticSearch):
             password=password
         )
 
-        self.logger = logging.getLogger(self.__class__.__name__)
+        self.logger = get_logger(self.__class__.__name__)
 
         self.connect_to_elasticsearch()
 
@@ -303,7 +304,7 @@ class ElasticSearchCloud(ElasticSearch):
             password=password
         )
 
-        self.logger = logging.getLogger(self.__class__.__name__)
+        self.logger = get_logger(self.__class__.__name__)
 
         self.connect_to_elasticsearch()
 

python_databases/postgresql_infrastructure/__init__.py

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

python_databases/postgresql_infrastructure/postgresql.py

@@ -2,6 +2,7 @@ import logging
 import os
 from typing import Optional, Union, Any
 import psycopg2
+from custom_python_logger import get_logger
 from psycopg2 import extras
 from retrying import retry
 
@@ -15,7 +16,7 @@ class PostgreSQL:
         username: Optional[str] = None,
         password: Optional[str] = None
     ):
-        self.logger = logging.getLogger(self.__class__.__name__)
+        self.logger = get_logger(self.__class__.__name__)
 
         self.name = name or os.getenv("POSTGRESQL_NAME")
         self.host = host or os.getenv("POSTGRESQL_HOST")
@@ -42,7 +43,7 @@ class PostgreSQL:
             self.create_connection()
         return self._cursor
 
-    @retry(stop_max_attempt_number=3, wait_fixed=180000)
+    # @retry(stop_max_attempt_number=3, wait_fixed=180000)
     def create_connection(self):
         self.logger.info(f"Create connection to {self.host}:{self.port}")
 
@@ -53,7 +54,7 @@ class PostgreSQL:
             user=self.username,
             password=self.password,
         )
-        self._cursor = self.client.cursor(cursor_factory=extras.DictCursor)
+        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}")

{python_databases-0.1.2.dist-info → python_databases-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-databases
-Version: 0.1.2
+Version: 0.1.4
 Summary: A Python library for managing and interacting with multiple databases seamlessly.
 Home-page: https://github.com/aviz92/python-databases
 Author: Avi Zaguri

python_databases-0.1.4.dist-info/RECORD

@@ -0,0 +1,10 @@
+python_databases/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_databases/elasticsearch_infrastructure/__init__.py,sha256=zLL3GJwgu5a55TgRhyl_mDbjc-qQBWOskRc0y9diqyw,296
+python_databases/elasticsearch_infrastructure/elasticsearch.py,sha256=WWlYi7icu3mJLEFDL98XbrG2bhAyBx1u_Xv8EPEgQt8,11176
+python_databases/postgresql_infrastructure/__init__.py,sha256=h1eJx0ogbtcF6UnN2c7uW5iCuadDab3So1VI15y3kWA,150
+python_databases/postgresql_infrastructure/postgresql.py,sha256=zH8y5w2k2djhHLefDJhlGwa6MtoZKONhJ5vMxIxDdlc,2856
+python_databases-0.1.4.dist-info/licenses/LICENSE,sha256=cSikHY6SZFsPZSBizCDAJ0-Bjjzxt-JtX6TVbKxwimo,1067
+python_databases-0.1.4.dist-info/METADATA,sha256=i0_YUTvulCbHXObsil2bPLbHsXv3XDh_IxpHsx6QWrg,1792
+python_databases-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+python_databases-0.1.4.dist-info/top_level.txt,sha256=fXcMUntaYJfKeFS7FjQ6DY4WVXYBsgdrDmhOudYs-vg,17
+python_databases-0.1.4.dist-info/RECORD,,

python_databases-0.1.2.dist-info/RECORD

@@ -1,10 +0,0 @@
-python_databases/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_databases/elasticsearch_infrastructure/__init__.py,sha256=zLL3GJwgu5a55TgRhyl_mDbjc-qQBWOskRc0y9diqyw,296
-python_databases/elasticsearch_infrastructure/elasticsearch.py,sha256=U8PVhjImk_S35KJHKEFK7Fl8cL-0A36yzLJ_KgtNTIY,11160
-python_databases/postgresql_infrastructure/__init__.py,sha256=LG-zblOfMP6hVPKsBg0_Vu7Np90aDwZoQtZkamDOsus,46
-python_databases/postgresql_infrastructure/postgresql.py,sha256=Hnn58f6TT2NPOyAou5EvBuH4xE6qGQfYep8wufrC_r8,2816
-python_databases-0.1.2.dist-info/licenses/LICENSE,sha256=cSikHY6SZFsPZSBizCDAJ0-Bjjzxt-JtX6TVbKxwimo,1067
-python_databases-0.1.2.dist-info/METADATA,sha256=wM_Kf2fxKvMk48tlW3Kx-DNnXE06-rwuQTT5hpuHSgc,1792
-python_databases-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-python_databases-0.1.2.dist-info/top_level.txt,sha256=fXcMUntaYJfKeFS7FjQ6DY4WVXYBsgdrDmhOudYs-vg,17
-python_databases-0.1.2.dist-info/RECORD,,