powerdial_round1_homepage:
  module: powerdial
  class: PowerDial
  lwt_topic: hasp/round1/LWT
  device_topic: round1
  page: 1
  arcs:
    - ids: [100]
      sensors:
      - [sensor.total_power_consumption, 1, "#ffff00"]
    - ids: [101,102]
      sensors:
      - [sensor.solaredge_ac_power, 1, "#00ff00"]
      - [sensor.electricity_meter_power_consumption, 1000, "#ff0000"]
    - ids: [110, 111]
      complement: true
      sensors:
       -  [sensor.electricity_meter_power_production, 1000, "#ff00ff"]
       -  [sensor.car_solar_power_2, 1, "#1affff"]
  scale: 10000

 1 @dataclass
 2 class Dial:
 3     entity_name: str
 4     multiplier: float
 5     color: str
 6     complement: bool
 7     scale: int
 8     value: float
 9     @property
10     def watts(self)->float:
11         return self.value * self.multiplier
12     @property
13     def widget_value(self)->str:
14         val = self.watts
15         if self.complement:
16             val = self.scale - val
17         return str(val)
18
19 @dataclass
20 class ArcWidget:
21     ids: List[int]
22     sensors: List[Dial]
23
24 class PowerDial(HassHaspWidget):
25     """
26     This layers multi color arcs on top of each other.
27     Suggested use:
28     The first is from left to right, and will show grid use and solar
29     The second is from right to left, and will show grid injection and battery charge power
30     of these pairs, the one with the smallest value will be in the foreground so at a glance
31     the power level of each entity can be estimated.
32     """
33     arcs: List[ArcWidget] = None
34
35     @property
36     def use_random_values(self):
37         return self.args.get('use_random_values',False)
38
39     def arc_color_topic(self, id_):
40         return f"hasp/{self.device_topic}/command/p{self.page}b{id_}.line_color10"
41
42     def arc_value_topic(self, id_):
43         return f"hasp/{self.device_topic}/command/p{self.page}b{id_}.val"
44
45     def recalculate(self):
46         """ Order the arcs so the shortest one rendered using the one in the foreground """
47         for arc in self.arcs:
48             arc.sensors.sort(key= lambda x: x.watts if x.complement else -x.watts)
49
50     def redraw(self):
51         for arc in self.arcs:
52             for widget_id, sensor in zip(arc.ids,arc.sensors):
53                 self.mqtt_publish(self.arc_value_topic(widget_id), sensor.widget_value)
54                 self.mqtt_publish(self.arc_color_topic(widget_id), sensor.color)
55
56     def mqtt_publish(self, topic, payload):
57         self.call_service("mqtt/publish", topic=topic, payload=payload)
58
59     def update(self, entity_name, attribute, old, new, cb_args):
60         for arc in self.arcs:
61             for sensor in arc.sensors:
62                 if sensor.entity_name == entity_name:
63                     sensor.value = float(new)
64                 if self.use_random_values:
65                     sensor.value = ((sensor.scale/10)+(sensor.scale/10)*random.random()) / sensor.multiplier
66         self.recalculate()
67         self.redraw()
68
69     def hass_subscribe(self):
70         for arc in self.arcs:
71             for sensor in arc.sensors:
72                 self.listen_state(self.update, sensor.entity_name)
73
74     def initialize(self):
75         HassHaspWidget.initialize(self)
76         scale = int(self.args['scale'])
77         self.arcs = []
78         for arc in self.args['arcs']:
79             complement = bool(arc.get('complement', False))
80             sensors = [
81                 Dial(
82                     entity_name = sensor[0],
83                     multiplier= float(sensor[1]),
84                     color= sensor[2],
85                     complement= complement,
86                     scale = scale,
87                     value = 0) \
88                 for sensor in arc['sensors']]
89             self.arcs += [ ArcWidget(arc['ids'], sensors)]
90
91         self.mqtt_subscribe()
92         self.hass_subscribe()
93         self.redraw()