commons-metrics 0.0.25__tar.gz → 0.0.26__tar.gz

{commons_metrics-0.0.25 → commons_metrics-0.0.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commons_metrics
-Version: 0.0.25
+Version: 0.0.26
 Summary: A simple library for basic statistical calculations
 Author: Bancolombia
 Author-email: omar.david.pino@email.com

{commons_metrics-0.0.25 → commons_metrics-0.0.26}/commons_metrics/__init__.py

@@ -12,4 +12,4 @@ from .text_simplifier import TextSimplifier
 from .variable_finder import VariableFinder
 
 __all__ = ['Util', 'DatabaseConnection', 'ComponentRepository', 'UpdateDesignSystemComponents', 'GitHubAPIClient', 'AzureDevOpsClient', 'S3FileManager', 'CacheManager', 'CommonsReposClient', 'DateUtils', 'TextSimplifier', 'VariableFinder']
-__version__ = '0.0.25'
+__version__ = '0.0.26'

{commons_metrics-0.0.25 → commons_metrics-0.0.26}/commons_metrics/database.py

@@ -10,6 +10,7 @@ class DatabaseConnection:
     
     def __init__(self):
         self.connection = None
+        self._credentials = None
 
     def connect(self, credentials):
         """Establishes database connection using credentials"""
@@ -20,8 +21,14 @@ class DatabaseConnection:
                 database=credentials['dbname'],
                 user=credentials['username'],
                 password=credentials['password'],
-                connect_timeout=30
+                connect_timeout=30,
+                sslmode='require',
+                keepalives=1,
+                keepalives_idle=30,
+                keepalives_interval=10,
+                keepalives_count=5
             )
+            self._credentials = credentials
         except Exception as e:
             raise Exception(f"Database connection error: {str(e)}")
 
@@ -40,6 +47,22 @@ class DatabaseConnection:
         if self.connection and not self.connection.closed:
             self.connection.rollback()
 
+    def ensure_connection(self):
+        """Verifies connection is alive and reconnects if necessary"""
+        try:
+            if self.connection is None or self.connection.closed:
+                if self._credentials:
+                    print("⚠️ Reconnecting to database...")
+                    self.connect(self._credentials)
+                else:
+                    raise Exception("Connection is closed and no credentials available for reconnection")
+        except Exception as e:
+            if self._credentials:
+                print(f"⚠️ Connection lost, attempting reconnection: {str(e)}")
+                self.connect(self._credentials)
+            else:
+                raise Exception(f"Cannot ensure connection: {str(e)}")
+
     def get_connection_database_from_secret(secret_name: str, logger: str, aws_region: str) -> ComponentRepository:
         """
         Retrieve connection database from AWS secrets manager

{commons_metrics-0.0.25 → commons_metrics-0.0.26}/commons_metrics/repositories.py

@@ -145,34 +145,30 @@ class ComponentRepository:
 
     def get_or_create_current_month_date(self) -> int:
         """
-        Obtiene el ID de la fecha del mes actual en measurement_systems_history.
-        Si no existe, crea un registro con la fecha de hoy.
+        Obtiene el ID de la fecha de HOY en measurement_systems_history.
+        Si no existe, crea un registro con CURRENT_DATE y devuelve su ID.
         
         Returns:
-            int: ID del registro de fecha
+            int: ID del registro de fecha de hoy
         """
         cursor = None
         try:
+            self.db.ensure_connection()
             cursor = self.db.connection.cursor()
-            
-            # Buscar si existe un registro del mes actual
+
             query_select = """
             SELECT id
             FROM schmetrc.measurement_systems_history
-            WHERE EXTRACT(YEAR FROM date) = EXTRACT(YEAR FROM CURRENT_DATE)
-              AND EXTRACT(MONTH FROM date) = EXTRACT(MONTH FROM CURRENT_DATE)
+            WHERE date = CURRENT_DATE
             LIMIT 1
             """
             cursor.execute(query_select)
             result = cursor.fetchone()
-            
+
             if result:
-                # Ya existe un registro del mes actual
                 date_id = result[0]
-                print(f"   ℹ️ Usando fecha existente del mes actual (ID: {date_id})")
                 return date_id
             else:
-                # No existe, crear uno nuevo con la fecha de hoy
                 query_insert = """
                 INSERT INTO schmetrc.measurement_systems_history (date)
                 VALUES (CURRENT_DATE)
@@ -181,9 +177,8 @@ class ComponentRepository:
                 cursor.execute(query_insert)
                 date_id = cursor.fetchone()[0]
                 self.db.commit_transaction()
-                print(f"   ✅ Creado nuevo registro de fecha (ID: {date_id})")
                 return date_id
-            
+
         except Exception as e:
             self.db.rollback_transaction()
             raise RuntimeError(f"Error getting or creating current month date: {str(e)}")

{commons_metrics-0.0.25 → commons_metrics-0.0.26}/commons_metrics.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commons_metrics
-Version: 0.0.25
+Version: 0.0.26
 Summary: A simple library for basic statistical calculations
 Author: Bancolombia
 Author-email: omar.david.pino@email.com

{commons_metrics-0.0.25 → commons_metrics-0.0.26}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='commons_metrics',
-    version='0.0.25',
+    version='0.0.26',
     description='A simple library for basic statistical calculations',
     #long_description=open('USAGE.md').read(),
     #long_description_content_type='text/markdown',