commons-metrics 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

commons_metrics/__init__.py

@@ -1,4 +1,6 @@
 from .util import Util
+from .database import DatabaseConnection
+from .repositories import ComponentRepository
 
-__all__ = ['Util']
-__version__ = '0.0.4'
+__all__ = ['Util', 'DatabaseConnection', 'ComponentRepository']
+__version__ = '0.0.6'

commons_metrics/database.py

@@ -0,0 +1,39 @@
+import psycopg2
+
+class DatabaseConnection:
+    """
+    Class to handle PostgreSQL database connections
+    using AWS Secrets Manager credentials
+    """
+    
+    def __init__(self):
+        self.connection = None
+
+    def connect(self, credentials):
+        """Establishes database connection using credentials"""
+        try:
+            self.connection = psycopg2.connect(
+                host=credentials['host'],
+                port=credentials['port'],
+                database=credentials['dbname'],
+                user=credentials['username'],
+                password=credentials['password'],
+                connect_timeout=30
+            )
+        except Exception as e:
+            raise Exception(f"Database connection error: {str(e)}")
+
+    def close(self):
+        """Closes the database connection"""
+        if self.connection and not self.connection.closed:
+            self.connection.close()
+
+    def commit_transaction(self):
+        """Commits pending transactions"""
+        if self.connection and not self.connection.closed:
+            self.connection.commit()
+
+    def rollback_transaction(self):
+        """Rolls back pending transactions in case of error"""
+        if self.connection and not self.connection.closed:
+            self.connection.rollback()

commons_metrics/repositories.py

@@ -0,0 +1,43 @@
+class ComponentRepository:
+    """
+    Repository class for component-specific database operations
+    """
+    
+    def __init__(self, db_connection):
+        """
+        Initialize repository with database connection
+        """
+        self.db = db_connection
+    
+    def save_component(self, technical_name, id_type, status=1):
+        """
+        Inserts a new component into schmesys.component table
+        
+        Args:
+            technical_name (str): Technical name of the component
+            id_type (int): Component type ID
+            status (int): Component status (default: 1)
+        """
+        cursor = None
+        try:
+            cursor = self.db.connection.cursor()
+            
+            insert_query = """
+                INSERT INTO schmesys.component(technical_name, id_type, status) 
+                VALUES (%s, %s, %s)
+                RETURNING id;
+            """
+            
+            cursor.execute(insert_query, (technical_name, id_type, status))
+            component_id = cursor.fetchone()[0]
+            self.db.commit_transaction()
+            
+            return component_id
+            
+        except Exception as e:
+            self.db.rollback_transaction()
+            raise Exception(f"Error saving component '{technical_name}': {str(e)}")
+        finally:
+            if cursor:
+                cursor.close()
+

commons_metrics/util.py

@@ -1,11 +1,14 @@
 import boto3
 import json
+
 class Util:
+    """Utility class for common operations"""
+    
     @staticmethod
-    def get_secret_aws(env: str, logger):
-        secret_name = f"nu0051005-measurement-systems-{env}-mesysqcd-secretrds-cnx1"
-        region_name = "us-east-1"
-
+    def get_secret_aws(secret_name: str, logger, region_name: str):
+        """
+        Retrieves AWS Secrets Manager secret and returns database credentials
+        """
         try:
             session = boto3.session.Session()
             client = session.client(
@@ -31,9 +34,5 @@ class Util:
             msg = f"Error getting the secret '{secret_name}': {str(e)}"
             logger.error(msg)
             raise Exception(msg)
-        
-    @staticmethod
-    def show(env: str, logger):
-        print("Hola mundo")
-        print(f"Environment: {env}")
+
 

{commons_metrics-0.0.4.dist-info → commons_metrics-0.0.6.dist-info}/METADATA

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

commons_metrics-0.0.6.dist-info/RECORD

@@ -0,0 +1,10 @@
+commons_metrics/__init__.py,sha256=W2Qz4cA0bOvTGIlf3R0eNMg6DJ8XhgK4eChf8gOyw8k,196
+commons_metrics/data_base.py,sha256=7I9ywLqQMtTlK1uuubM8WrffwDsL9uyP3sygqanUQvQ,1439
+commons_metrics/database.py,sha256=570TtLZ9psNzvIp75UFLYph34cKVEz6eGJgxXyRyjW4,1285
+commons_metrics/repositories.py,sha256=JruFtcT3IOTOKn3euM67c0SZtJKmYVc07gN5dMAFSmw,1365
+commons_metrics/util.py,sha256=98zuynalXumQRh-BB0Bcjyoh6vS2BTOUM8tVgr7iS9Q,1225
+commons_metrics-0.0.6.dist-info/licenses/LICENSE,sha256=jsHZ2Sh1wCL74HC25pDDGXCyQ0xgsTAy62FvEnehKIg,1067
+commons_metrics-0.0.6.dist-info/METADATA,sha256=-FNZi97I757Id2ukBPtK_PvJvq1SGtTvFdihDL75GSA,401
+commons_metrics-0.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+commons_metrics-0.0.6.dist-info/top_level.txt,sha256=lheUN-3OKdU3A8Tg8Y-1IEB_9i_vVRA0g_FOiUsTQz8,16
+commons_metrics-0.0.6.dist-info/RECORD,,

commons_metrics-0.0.4.dist-info/RECORD

@@ -1,8 +0,0 @@
-commons_metrics/__init__.py,sha256=RRtS7dpbGr6V0p2t_wGcqMOnyW1IxADMwo25i3BjKyU,64
-commons_metrics/data_base.py,sha256=7I9ywLqQMtTlK1uuubM8WrffwDsL9uyP3sygqanUQvQ,1439
-commons_metrics/util.py,sha256=H8lyVqfy73AvohCedBRoadpFnqRhGRiEnjI7b6tvg2w,1288
-commons_metrics-0.0.4.dist-info/licenses/LICENSE,sha256=jsHZ2Sh1wCL74HC25pDDGXCyQ0xgsTAy62FvEnehKIg,1067
-commons_metrics-0.0.4.dist-info/METADATA,sha256=bWcn6USQP6aqgq3Gferi-sv74yf5qlM805m2nVk2fpM,401
-commons_metrics-0.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-commons_metrics-0.0.4.dist-info/top_level.txt,sha256=lheUN-3OKdU3A8Tg8Y-1IEB_9i_vVRA0g_FOiUsTQz8,16
-commons_metrics-0.0.4.dist-info/RECORD,,