import requests
import mysql.connector
from mysql.connector import Error
from datetime import datetime, timedelta

def insert_dynamic_price_data_tomorrow():
    connection = None 

    try:
        # 1. Bepaal de datum van morgen (bijv. '2026-03-31')
        tomorrow_date = (datetime.now() + timedelta(days=1)).strftime('%Y-%m-%d')
        print(f"Filteren op data voor: {tomorrow_date}")

        # 2. Data ophalen
        url = "https://enever.nl/apiv3/stroomprijs_morgen.php?token=5a7d9b371fe147cfc8100bcf6d9ebd55"
        # Let op: De API voor morgen geeft vaak pas na 15:00 uur resultaat
        response = requests.get(url)
        if response.status_code != 200:
            raise Exception(f'Fout bij het ophalen van de data. Status: {response.status_code}')

        data = response.json()
        if not data or 'data' not in data:
            raise Exception('Geen data gevonden in API respons.')

        # 3. Verbinden met MySQL
        connection = mysql.connector.connect(
            host='127.0.0.1',
            database='alfen',
            user='alfen_user',
            password='5uVgr%f%s2P5GR@3q!',
            port=3307
        )

        if connection.is_connected():
            cursor = connection.cursor()

            price_fields = [
                'prijsAA', 'prijsAIP', 'prijsANWB', 'prijsBE', 'prijsEE',
                'prijsEN', 'prijsEVO', 'prijsEZ', 'prijsFR', 'prijsGSL',
                'prijsMDE', 'prijsNE', 'prijsTI', 'prijsVDB', 'prijsVON',
                'prijsWE', 'prijsZG', 'prijsZP'
            ]

            rows_to_insert = []

            # 4. Loop door de data en filter op morgen
            for entry in data['data']:
                # Controleer of de datum in de API begint met de datum van morgen
                if entry['datum'].startswith(tomorrow_date):
                    # FIX: Maak de datum string schoon voor MySQL (T en tijdzone verwijderen)
                    clean_dt = entry['datum'].replace('T', ' ').split('+')[0]
                    
                    # Verzamel prijzen en zet ze om naar floats
                    prices = []
                    for f in price_fields:
                        if f in entry and entry[f] is not None and entry[f] != '':
                            try:
                                prices.append(float(entry[f]))
                            except ValueError:
                                continue
                    
                    if prices:
                        avg_price = sum(prices) / len(prices)
                        rows_to_insert.append((clean_dt, avg_price))

            # 5. Database bijwerken
            if rows_to_insert:
                # Wis de tabel voor morgen (zodat je altijd de meest recente data hebt)
                cursor.execute("DELETE FROM dynamic_price_data_tommorow")
                
                # Gebruik executemany voor snelheid
                query = "INSERT INTO dynamic_price_data_tommorow (datetime, price) VALUES (%s, %s)"
                cursor.executemany(query, rows_to_insert)
                
                connection.commit()
                print(f"Succes! {len(rows_to_insert)} rijen toegevoegd voor {tomorrow_date}.")
            else:
                print(f"Geen data gevonden voor {tomorrow_date} in de API-respons. (Dit is normaal voor 15:00u)")

    except Error as e:
        print(f"Database fout: {e}")
    except Exception as e:
        print(f"Fout: {e}")
    finally:
        if connection and connection.is_connected():
            cursor.close()
            connection.close()
            print("Verbinding gesloten.")

if __name__ == "__main__":
    insert_dynamic_price_data_tomorrow()