shakecities/sites/db.py

35 lines
788 B
Python
Raw Normal View History

2023-11-08 19:11:55 +11:00
import mysql.connector
import os
import dotenv
2023-11-08 22:14:06 +11:00
import json
2023-11-08 19:11:55 +11:00
dotenv.load_dotenv()
# Database connection
dbargs = {
'host':os.getenv('DB_HOST'),
'user':os.getenv('DB_USER'),
'password':os.getenv('DB_PASSWORD'),
'database':os.getenv('DB_NAME')
}
2023-11-08 20:43:33 +11:00
def get_website_data(domain):
2023-11-08 19:11:55 +11:00
connection = mysql.connector.connect(**dbargs)
cursor = connection.cursor()
cursor.execute("""
SELECT * FROM site WHERE domain = %s
""", (domain,))
2023-11-08 20:43:33 +11:00
data = cursor.fetchall()
2023-11-08 19:11:55 +11:00
cursor.close()
connection.close()
2023-11-08 20:43:33 +11:00
if data == []:
2023-11-08 22:14:06 +11:00
return "No data found for this domain"
parsed = data[0][2]
parsed = json.loads(parsed)
parsed = parsed['data']
# Decoding
parsed = parsed.encode('utf-8').decode('unicode-escape')
return parsed