Spaces:
Sleeping
Sleeping
File size: 1,148 Bytes
377ed3a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import pymysql
import json
import pandas as pd
import re
def initialize_database():
# Database Connection
db_params = {"host": "localhost",
"user": "cms-readonly-user",
"password": "%Reed!!",
"port": 3307,
"database": "veda_prod_v1"
}
db = pymysql.connect(**db_params)
return db
def execute_query(query):
db = initialize_database()
cursor = db.cursor()
try:
cursor.execute(query)
description = cursor.description
result = cursor.fetchall() # Fetch all rows from the result set
db.commit()
return description, result
except Exception as e:
print("Error executing query:", e)
db.rollback()
return None # Return None if an error occurs
finally:
db.close()
def _get_details_mantra_json(self, query):
description, data = execute_query(query)
df = pd.DataFrame(data)
df.columns = [x[0] for x in description]
mantra_json = df['mantra_json'].values[0]
cleaned_data = re.sub('<[^<]+?>', '', mantra_json)
return json.loads(cleaned_data)
|