import time
from datetime import datetime
from data_base_influx import DataBaseInflux
from forecaster.weather_forecaster import WeatherForecaster
from heat_pump import HeatPump
from pv_inverter import PvInverter
from simulators.pv_plant_simulator import PvWattsSubarrayConfig, PvWattsPlant
from solaredge_meter import SolaredgeMeter
from shelly_pro_3m import ShellyPro3m
from energysystem import EnergySystem
from sg_ready_controller import SgReadyController
from pvlib.location import Location
import datetime as dt

# For dev-System run in terminal: ssh -N -L 127.0.0.1:8111:10.0.0.10:502 pi@192.168.1.146
# For productive-System change IP-adress in heatpump to '10.0.0.10' and port to 502

interval_seconds = 10

es = EnergySystem()

db = DataBaseInflux(
    url="http://192.168.1.146:8086",
    token="Cw_naEZyvJ3isiAh1P4Eq3TsjcHmzzDFS7SlbKDsS6ZWL04fMEYixWqtNxGThDdG27S9aW5g7FP9eiq5z1rsGA==",
    org="allmende",
    bucket="allmende_db"
)

# hp_master = HeatPump(device_name='hp_master', ip_address='10.0.0.10', port=502)
# hp_slave = HeatPump(device_name='hp_slave', ip_address='10.0.0.11', port=502)
# shelly = ShellyPro3m(device_name='wohnung_2_6', ip_address='192.168.1.121')
wr_master = PvInverter(device_name='solaredge_master', ip_address='192.168.1.112', unit=1)
wr_slave = PvInverter(device_name='solaredge_slave', ip_address='192.168.1.112', unit=3)
meter = SolaredgeMeter(device_name='solaredge_meter', ip_address='192.168.1.112')

es.add_components(wr_master, wr_slave)#hp_master, hp_slave, shelly, wr_master, wr_slave, meter)
# controller = SgReadyController(es)
#
# # FORECASTING
# latitude = 48.041
# longitude = 7.862
# TZ = "Europe/Berlin"
# HORIZON_DAYS = 2
# weather_forecaster = WeatherForecaster(latitude=latitude, longitude=longitude)
# site = Location(latitude=latitude, longitude=longitude, altitude=35, tz=TZ, name="Gundelfingen")
#
# p_module = 435
# upper_roof_north = PvWattsSubarrayConfig(name="north", pdc0_w=(29+29+21)*p_module, tilt_deg=10, azimuth_deg=20, dc_loss=0.02, ac_loss=0.01)
# upper_roof_south = PvWattsSubarrayConfig(name="south", pdc0_w=(29+21+20)*p_module, tilt_deg=10, azimuth_deg=200, dc_loss=0.02, ac_loss=0.01)
# upper_roof_east = PvWattsSubarrayConfig(name="east", pdc0_w=7*p_module, tilt_deg=10, azimuth_deg=110, dc_loss=0.02, ac_loss=0.01)
# upper_roof_west = PvWattsSubarrayConfig(name="west", pdc0_w=7*p_module, tilt_deg=10, azimuth_deg=290, dc_loss=0.02, ac_loss=0.01)
# cfgs = [upper_roof_north, upper_roof_south, upper_roof_east, upper_roof_west]
# pv_plant = PvWattsPlant(site, cfgs)
#
# now = datetime.now()
# next_forecast_at = (now + dt.timedelta(hours=1)).replace(minute=0, second=0, microsecond=0)
while True:
    now = datetime.now()
    if now.second % interval_seconds == 0 and now.microsecond < 100_000:
        state = es.get_state_and_store_to_database(db)
        # mode = controller.perform_action(heat_pump_name='hp_master', meter_name='solaredge_meter', state=state)
        #
        # if mode == 'mode1':
        #     mode_as_binary = 0
        # else:
        #     mode_as_binary = 1
        # db.store_data('sg_ready', {'mode': mode_as_binary})

    # if now >= next_forecast_at:
    #     # Start der Prognose: ab der kommenden vollen Stunde
    #     start_hour_local = (now + dt.timedelta(hours=1)).replace(minute=0, second=0, microsecond=0)
    #     weather = weather_forecaster.get_hourly_forecast(start_hour_local, HORIZON_DAYS)
    #     total = pv_plant.get_power(weather)
    #     db.store_forecasts('pv_forecast', total)
    #
    #     # Nächste geplante Ausführung definieren (immer volle Stunde)
    #     # Falls wir durch Delay mehrere Stunden verpasst haben, hole auf:
    #     while next_forecast_at <= now:
    #         next_forecast_at = (next_forecast_at + dt.timedelta(hours=1)).replace(minute=0, second=0, microsecond=0)


    time.sleep(0.1)