commons-metrics 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

commons_metrics/__init__.py

@@ -1,4 +1,5 @@
 from .util import Util
+from .database import DatabaseConnection
 
-__all__ = ['Util']
-__version__ = '0.1.0'
+__all__ = ['Util', 'DatabaseConnection']
+__version__ = '0.0.5'

commons_metrics/data_base.py

@@ -0,0 +1,45 @@
+import psycopg2
+
+class DataBase:
+    def __init__(self, db_name):
+        self.connection = None
+
+    def connect(self, host, port, dbname, user, password):
+        try:
+            self.connection = psycopg2.connect(
+                host=host,
+                port=port,
+                database=dbname,
+                user=user,
+                password=password,
+                connect_timeout=30
+            )
+
+        except Exception as e:
+            if self.connection:
+                self.connection.rollback()
+            msg = f"Database connection error: {str(e)}"
+            raise Exception(msg)
+
+    def disconnect(self):
+        if self.connectWWWion:
+            self.connection.close()
+
+    def save_component(technical_name, id_type):
+        try:
+            cursor = connection.cursor()
+            params_insert = (
+                technical_name,
+                id_type
+            )
+
+            #cursor.execute("select id_component into v_id_component from schmesys.component where technical_name = v_component_name;", params)
+            cursor.execute("INSERT INTO schmesys.component(technical_name,id_type) values (%s, %d);", params_insert)
+        except Exception as e:
+            if self.connection:
+                self.connection.rollback()
+            msg = f"Database connection error: {str(e)}"
+            raise Exception(msg)
+        finally:
+            if cursor:
+                cursor.close()

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/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,7 +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")
+
+

{commons_metrics-0.0.3.dist-info → commons_metrics-0.0.5.dist-info}/METADATA

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

commons_metrics-0.0.5.dist-info/RECORD

@@ -0,0 +1,9 @@
+commons_metrics/__init__.py,sha256=vbEyX-oQv35rlSa0zSi-4CacNpyoCqxsedHAzDKQtwU,127
+commons_metrics/data_base.py,sha256=7I9ywLqQMtTlK1uuubM8WrffwDsL9uyP3sygqanUQvQ,1439
+commons_metrics/database.py,sha256=570TtLZ9psNzvIp75UFLYph34cKVEz6eGJgxXyRyjW4,1285
+commons_metrics/util.py,sha256=98zuynalXumQRh-BB0Bcjyoh6vS2BTOUM8tVgr7iS9Q,1225
+commons_metrics-0.0.5.dist-info/licenses/LICENSE,sha256=jsHZ2Sh1wCL74HC25pDDGXCyQ0xgsTAy62FvEnehKIg,1067
+commons_metrics-0.0.5.dist-info/METADATA,sha256=6xfET92d3JXR9fRdIE5gGJZ2D2tRKxzYVo4CHcC-XvU,401
+commons_metrics-0.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+commons_metrics-0.0.5.dist-info/top_level.txt,sha256=lheUN-3OKdU3A8Tg8Y-1IEB_9i_vVRA0g_FOiUsTQz8,16
+commons_metrics-0.0.5.dist-info/RECORD,,

commons_metrics-0.0.3.dist-info/RECORD

@@ -1,7 +0,0 @@
-commons_metrics/__init__.py,sha256=EqBumKI85i-qgaT_d7Vw_IWiKg7ARdY642aw3gLnMBI,64
-commons_metrics/util.py,sha256=qxpXQOPBSh2jD8try1nD8fRRdyZwMAZWQQKFT6XajSU,1249
-commons_metrics-0.0.3.dist-info/licenses/LICENSE,sha256=jsHZ2Sh1wCL74HC25pDDGXCyQ0xgsTAy62FvEnehKIg,1067
-commons_metrics-0.0.3.dist-info/METADATA,sha256=L7lny4AYLCNf_gIa4_F9DUy-WqlfXzfMvFiqZCoHxgs,401
-commons_metrics-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-commons_metrics-0.0.3.dist-info/top_level.txt,sha256=lheUN-3OKdU3A8Tg8Y-1IEB_9i_vVRA0g_FOiUsTQz8,16
-commons_metrics-0.0.3.dist-info/RECORD,,