From 3d437a03db54c13dc36e59e5602ead75d61724b4 Mon Sep 17 00:00:00 2001
From: dv-sharma <65087388+dv-sharma@users.noreply.github.com>
Date: Sat, 24 Feb 2024 00:23:04 -0500
Subject: [PATCH 1/2] Create awsrdsmysqlp.py

Created a python script to execute query on mysql RDS instance on AWS using boto3,mysql.connector and tabulate module
---
 awsrdsmysqlp.py | 52 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 awsrdsmysqlp.py

diff --git a/awsrdsmysqlp.py b/awsrdsmysqlp.py
new file mode 100644
index 0000000..b342bb2
--- /dev/null
+++ b/awsrdsmysqlp.py
@@ -0,0 +1,52 @@
+import boto3
+import json
+import mysql.connector
+import tabulate
+
+# Initialize AWS Secrets Manager client
+secrets_manager = boto3.client('secretsmanager')
+db_host = "DB_HOST"
+
+def get_secretvalue():
+    secret_name = secrets_manager.get_secret_value(SecretId='YOUR_SECRET_ARN')
+    secret_dict = json.loads(secret_name['SecretString'])
+    db_username = secret_dict['username']
+    db_password = secret_dict['password']
+    return db_username, db_password
+    
+def execute_query(db_name, query):
+    db_username, db_password = get_secretvalue()
+
+    for db in db_name:
+        try:
+            connection = mysql.connector.connect(
+                host=db_host,
+                user=db_username,
+                password=db_password,
+                database=db
+            )
+            cursor = connection.cursor()
+            cursor.execute(query)
+            result = cursor.fetchall()
+            row_count = cursor.rowcount
+
+            if query.upper().startswith(("SELECT", "SHOW", "DESCRIBE")):
+                if result:
+                    header = [column_name[0] for column_name in cursor.description]
+                    table_output = [header] + list(result)
+                    print(f"Running query on: {db}")
+                    print(tabulate.tabulate(table_output, tablefmt="pipe"))
+                else:
+                    print("Empty result")
+            else:
+                print(f"{row_count} rows affected in {db}")          
+        
+        except Exception as e:
+            print("Error:", e)
+        finally:
+            if 'connection' in locals():
+                connection.close()
+
+db_names = input("Enter the DB names followed by space:").split()
+query = input("Enter the query:")
+execute_query(db_names, query)

From d97c2f25136cbc0c78a833ce1fefba5ad5e3c026 Mon Sep 17 00:00:00 2001
From: dv-sharma <65087388+dv-sharma@users.noreply.github.com>
Date: Sat, 24 Feb 2024 02:11:05 -0500
Subject: [PATCH 2/2] Update awsrdsmysqlp.py

connection.commit() added to retain INSERT,UPDATE changes
---
 awsrdsmysqlp.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/awsrdsmysqlp.py b/awsrdsmysqlp.py
index b342bb2..d6e76fd 100644
--- a/awsrdsmysqlp.py
+++ b/awsrdsmysqlp.py
@@ -39,6 +39,7 @@ def execute_query(db_name, query):
                 else:
                     print("Empty result")
             else:
+                connection.commit()
                 print(f"{row_count} rows affected in {db}")          
         
         except Exception as e: