import numpy as np
class PVAkku:
    def __init__(self, kapazitaet_wh):
        # Kapazität des Akkus in Wh
        self.kapazitaet_wh = kapazitaet_wh
        # Initialer Ladezustand des Akkus in Wh
        self.soc_wh = 0
        self.discharge_array = np.full(24, 1)

    def reset(self):
        self.soc_wh = 0
        self.discharge_array = np.full(24, 1)
        
    def set_discharge_per_hour(self, discharge_array):
        assert(len(discharge_array) == 24)
        self.discharge_array = discharge_array

    def ladezustand_in_prozent(self):
        return (self.soc_wh / self.kapazitaet_wh) * 100

    def energie_abgeben(self, wh, hour):
        if self.discharge_array[hour] == 0:
            return 0.0
        if self.soc_wh >= wh:
            self.soc_wh -= wh
            return wh
        else:
            abgegebene_energie = self.soc_wh
            self.soc_wh = 0
            return abgegebene_energie

    def energie_laden(self, wh):
        if self.soc_wh + wh <= self.kapazitaet_wh:
            self.soc_wh += wh
        else:
            self.soc_wh = self.kapazitaet_wh



if __name__ == '__main__':
    # Beispiel zur Nutzung der Klasse
    akku = PVAkku(10000) # Ein Akku mit 10.000 Wh Kapazität
    print(f"Initialer Ladezustand: {akku.ladezustand_in_prozent()}%")

    akku.energie_laden(5000)
    print(f"Ladezustand nach Laden: {akku.ladezustand_in_prozent()}%, Aktueller Energieinhalt: {akku.aktueller_energieinhalt()} Wh")

    abgegebene_energie_wh = akku.energie_abgeben(3000)
    print(f"Abgegebene Energie: {abgegebene_energie_wh} Wh, Ladezustand danach: {akku.ladezustand_in_prozent()}%, Aktueller Energieinhalt: {akku.aktueller_energieinhalt()} Wh")

    akku.energie_laden(6000)
    print(f"Ladezustand nach weiterem Laden: {akku.ladezustand_in_prozent()}%, Aktueller Energieinhalt: {akku.aktueller_energieinhalt()} Wh")