#!/usr/bin/env python3

from pyfroniusreg import gen24_registers as froniusreg

from pymodbus.client.tcp import ModbusTcpClient

fronius1 = ModbusTcpClient("172.19.107.211", port=502, timeout=10)
fronius1.connect()

soc = froniusreg.scaledStateOfCharge.getValue(fronius1)
print(" SOC: %s%%" % soc)

discharge = froniusreg.scaledOutWRte.getValue(fronius1)
print("Pre DRate: %d%%" % discharge)

charge = froniusreg.scaledInWRte.getValue(fronius1)
print("Pre CRate: %d%%" % charge)

mode = froniusreg.StorCtl_Mode.getValue(fronius1)
print("Pre Mode: %d" % mode)

# This should be 'limit discharge' mode
print("Setting control mode to limit discharge")
err = froniusreg.StorCtl_Mode.setValue(fronius1, 2)

# Charge battery at a rate from -2% discharge to 100% charge
# as a percentage of the MaxChaRte, which in our case is 25600W
print("Setting discharge rate to -10%")
err = froniusreg.scaledOutWRte.setValue(fronius1, int(-10))
print("Setting charge rate to 100%")
err = froniusreg.scaledInWRte.setValue(fronius1, int(100))

discharge = froniusreg.scaledOutWRte.getValue(fronius1)
print("Post DRate: %d%%" % discharge)

charge = froniusreg.scaledInWRte.getValue(fronius1)
print("Post CRate: %d%%" % charge)

mode = froniusreg.StorCtl_Mode.getValue(fronius1)
print("Post Mode: %d" % mode)