File size: 1,469 Bytes
c101c53 |
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 |
import psycopg2
import os
from utils import TEMP_DIR
def connect_sql_db(url, sql_user, sql_port, sql_pass, sql_db_name, session_hash):
try:
conn = psycopg2.connect(
database=sql_db_name,
user=sql_user,
password=sql_pass,
host=url, # e.g., "localhost" or an IP address
port=sql_port # default is 5432
)
print("Connected to PostgreSQL")
# Create a cursor object to execute SQL queries
cur = conn.cursor()
# Example: Execute a query
cur.execute("""SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'""")
table_tuples = cur.fetchall()
table_names = []
for table in table_tuples:
table_names.append(table[0])
print(table_names)
# Close the cursor and connection
cur.close()
conn.close()
print("Connection closed.")
session_path = 'sql'
dir_path = TEMP_DIR / str(session_hash) / str(session_path)
os.makedirs(dir_path, exist_ok=True)
return ["success","<p style='color:green;text-align:center;font-size:18px;'>SQL database connected successful</p>", table_names]
except Exception as e:
print("UPLOAD ERROR")
print(e)
return ["error",f"<p style='color:red;text-align:center;font-size:18px;font-weight:bold;'>ERROR: {e}</p>"]
|