auto_trade_sys/check_db.py
薇薇安 449ad01ede a
2026-02-03 09:49:25 +08:00

36 lines
776 B
Python

import os
import sys
import pymysql
from pathlib import Path
from dotenv import load_dotenv
# Load env
backend_dir = Path(__file__).parent / 'backend'
load_dotenv(backend_dir / '.env', override=True)
host = os.getenv('DB_HOST', 'localhost')
port = int(os.getenv('DB_PORT', 3306))
user = os.getenv('DB_USER', 'root')
password = os.getenv('DB_PASSWORD', '')
print(f"Connecting to {host}:{port} as {user}")
try:
conn = pymysql.connect(
host=host,
port=port,
user=user,
password=password,
charset='utf8mb4'
)
cursor = conn.cursor()
cursor.execute("SHOW DATABASES")
dbs = cursor.fetchall()
print("Databases:")
for db in dbs:
print(db)
conn.close()
except Exception as e:
print(f"Error: {e}")