import pymysql
import os
from dotenv import load_dotenv
from datetime import datetime

load_dotenv()

try:
    conn = pymysql.connect(
        host=os.getenv('MYSQL_HOST'),
        user=os.getenv('MYSQL_USER'),
        password=os.getenv('MYSQL_PASSWORD'),
        database=os.getenv('MYSQL_DATABASE'),
        port=int(os.getenv('MYSQL_PORT', 3306)),
        cursorclass=pymysql.cursors.DictCursor
    )
    with conn.cursor() as cursor:
        cursor.execute("SELECT strategy_name, ts FROM portfolio_state")
        results = cursor.fetchall()
        print(f"--- Portfolio State (Current Time: {datetime.now()}) ---")
        for row in results:
            print(f"Strategy: {row['strategy_name']}, Updated At: {row['ts']}")
            
        cursor.execute("SELECT MAX(ts_event) as last_snapshot FROM market_snapshots")
        res = cursor.fetchone()
        print(f"Last Market Snapshot: {res['last_snapshot']}")
        
    conn.close()
except Exception as e:
    print(f"FAILURE: {e}")