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-01-25')
        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/api/stroomprijs_morgen.php?token=5a7d9b371fe147cfc8100bcf6d9ebd55"
        response = requests.get(url)
        if response.status_code != 200:
            raise Exception('Fout bij het ophalen van de data.')

        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):
                    prices = [float(entry[f]) for f in price_fields if f in entry and entry[f] != '']
                    
                    if prices:
                        avg_price = sum(prices) / len(prices)
                        rows_to_insert.append((entry['datum'], avg_price))

            # 5. Database bijwerken
            if rows_to_insert:
                # Wis de tabel
                cursor.execute("DELETE FROM dynamic_price_data_tommorow")
                
                # Voeg alleen de gefilterde rijen toe
                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.")

    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()

if __name__ == "__main__":
    insert_dynamic_price_data_tomorrow()