import os
from sqlalchemy import create_engine, text
from dotenv import load_dotenv

load_dotenv()

host = os.getenv('MYSQL_HOST')
user = os.getenv('MYSQL_USER')
pw = os.getenv('MYSQL_PASSWORD')
db_name = os.getenv('MYSQL_DATABASE')

url = f"mysql+pymysql://{user}:{pw}@{host}/{db_name}"
engine = create_engine(url)

def audit():
    with engine.connect() as conn:
        print("--- All Unique Strategy Names in portfolio_state ---")
        names = conn.execute(text("SELECT DISTINCT strategy_name FROM portfolio_state")).fetchall()
        for n in names:
            print(f"'{n[0]}'")
            
        print("\n--- Recent Records for Strategy 'GridBot' ---")
        # Checking if there's any weird duplicate timestamps
        grid_records = conn.execute(text("SELECT ts, equity FROM portfolio_state WHERE strategy_name='GridBot' ORDER BY ts DESC LIMIT 5")).fetchall()
        for r in grid_records:
            print(f"{r[0]} -> {r[1]}")

if __name__ == "__main__":
    audit()