# test_pensav_db.py
import os
import pymysql

def fetch_data():
    conn = pymysql.connect(
        host=os.getenv("DB_HOST"),
        port=int(os.getenv("DB_PORT", 3306)),
        user=os.getenv("DB_USER"),
        password=os.getenv("DB_PASS"),
        database=os.getenv("DB_NAME"),
        cursorclass=pymysql.cursors.DictCursor
    )
    # rest of your code...

    try:
        with conn.cursor() as cursor:
            # Fetch the most recent row by Date
            cursor.execute("""
                SELECT IPV, MPV
                FROM pensavT
                ORDER BY Date DESC
                LIMIT 1
            """)
            row = cursor.fetchone()
            if not row or "IPV" not in row or "MPV" not in row:
                raise ValueError("Missing IPV or MPV in DB row")

            total_value = float(row["IPV"]) + float(row["MPV"])

            return {
                "start_value": total_value
            }

    finally:
        conn.close()