File size: 1,331 Bytes
86be40b | 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 mysql.connector
import os
from dotenv import load_dotenv
# Load environment variables from backend/.env
load_dotenv('backend/.env')
def check_db():
try:
conn = mysql.connector.connect(
host=os.getenv('DB_HOST'),
port=os.getenv('DB_PORT'),
user=os.getenv('DB_USER'),
password=os.getenv('DB_PASSWORD'),
database=os.getenv('DB_NAME')
)
cursor = conn.cursor()
print("\n--- Aiven Cloud DB Status ---")
# Check Sources
cursor.execute("SELECT COUNT(*) FROM sources")
source_count = cursor.fetchone()[0]
print(f"Total Sources: {source_count}")
# Check Chunks
cursor.execute("SELECT COUNT(*) FROM chunks")
chunk_count = cursor.fetchone()[0]
print(f"Total Chunks stored: {chunk_count}")
if source_count > 0:
print("\nRecent Sources:")
cursor.execute("SELECT title, type, chunk_count FROM sources ORDER BY created_at DESC LIMIT 5")
for title, stype, count in cursor.fetchall():
print(f" - [{stype.upper()}] {title} ({count} chunks)")
cursor.close()
conn.close()
except Exception as e:
print(f"Error connecting to cloud DB: {e}")
if __name__ == "__main__":
check_db()
|