import datetime
from givenergy_modbus.client import GivEnergyClient
from givenergy_modbus.model.battery import Battery
from givenergy_modbus.model.inverter import Inverter, Model
from givenergy_modbus.model.register_cache import RegisterCache

client = GivEnergyClient(host="192.168.99.99")

# change configuration on the device:
client.enable_charge_target(80)
# set a charging slot from 00:30 to 04:30
client.set_charge_slot_1((datetime.time(hour=0, minute=30), datetime.time(hour=4, minute=30)))
# set the inverter to charge when there's excess, and discharge otherwise. it will also respect charging slots.
client.set_mode_dynamic()

rc = RegisterCache()
client.update_inverter_registers(rc)
inverter = Inverter.from_orm(rc)
assert inverter.inverter_serial_number == 'SA1234G567'
assert inverter.inverter_model == Model.Hybrid
assert inverter.v_pv1 == 1.4  # V
assert inverter.e_battery_discharge_day == 8.1  # kWh
assert inverter.enable_charge_target
assert inverter.dict() == {
    'inverter_serial_number': 'SA1234G567',
    'device_type_code': '3001',
    'charge_slot_1': (datetime.time(0, 30), datetime.time(7, 30)),
    'f_ac1': 49.98,
    ...
}
assert inverter.json() == '{"inverter_serial_number": "SA1234G567", "device_type_code": "3001", ...'

# it is _probably_ a good idea to not share register caches across devices
rc = RegisterCache()
client.update_battery_registers(rc, battery_number=0)
battery = Battery.from_orm(rc)

assert battery.serial_number == 'BG1234G567'
assert battery.v_battery_cell_01 == 3.117
assert battery.dict() == {
    'bms_firmware_version': 3005,
    'design_capacity': 160.0,
    ...
}
assert battery.json() == '{"battery_serial_number": "BG1234G567", "v_battery_cell_01": 3.117, ...'