|
| 1 | +""" |
| 2 | + Control Trigger Interface |
| 3 | + Monitors control state changes and |
| 4 | + checks new state against any |
| 5 | + thresholds if provided. |
| 6 | +""" |
| 7 | +from mudpi.utils import decode_event_data |
| 8 | +from mudpi.exceptions import ConfigError |
| 9 | +from mudpi.extensions import BaseInterface |
| 10 | +from mudpi.extensions.trigger import Trigger |
| 11 | +from mudpi.logger.Logger import Logger, LOG_LEVEL |
| 12 | + |
| 13 | + |
| 14 | +class Interface(BaseInterface): |
| 15 | + |
| 16 | + def load(self, config): |
| 17 | + """ Load control Trigger component from configs """ |
| 18 | + trigger = ControlTrigger(self.mudpi, config) |
| 19 | + if trigger: |
| 20 | + self.add_component(trigger) |
| 21 | + return True |
| 22 | + |
| 23 | + def validate(self, config): |
| 24 | + """ Validate the trigger config """ |
| 25 | + if not isinstance(config, list): |
| 26 | + config = [config] |
| 27 | + |
| 28 | + for conf in config: |
| 29 | + if not conf.get('source'): |
| 30 | + raise ConfigError('Missing `source` key in Sensor Trigger config.') |
| 31 | + |
| 32 | + return config |
| 33 | + |
| 34 | + |
| 35 | +class ControlTrigger(Trigger): |
| 36 | + """ A trigger that listens to states |
| 37 | + and checks for new state that |
| 38 | + matches any thresholds. |
| 39 | + """ |
| 40 | + |
| 41 | + # Used for onetime subscribe |
| 42 | + _listening = False |
| 43 | + |
| 44 | + |
| 45 | + def init(self): |
| 46 | + """ Listen to the state for changes """ |
| 47 | + super().init() |
| 48 | + if self.mudpi.is_prepared: |
| 49 | + if not self._listening: |
| 50 | + # TODO: Eventually get a handler returned to unsub just this listener |
| 51 | + self.mudpi.events.subscribe('control', self.handle_event) |
| 52 | + self._listening = True |
| 53 | + return True |
| 54 | + |
| 55 | + """ Methods """ |
| 56 | + def handle_event(self, event): |
| 57 | + """ Handle the event data from the event system """ |
| 58 | + _event_data = decode_event_data(event) |
| 59 | + if _event_data.get('event'): |
| 60 | + try: |
| 61 | + if _event_data['event'] == 'ControlUpdated': |
| 62 | + if _event_data['component_id'] == self.source: |
| 63 | + _value = self._parse_data(_event_data["state"]) |
| 64 | + if self.evaluate_thresholds(_value): |
| 65 | + self.active = True |
| 66 | + if self._previous_state != self.active: |
| 67 | + # Trigger is reset, Fire |
| 68 | + self.trigger(_event_data) |
| 69 | + else: |
| 70 | + # Trigger not reset check if its multi fire |
| 71 | + if self.frequency == 'many': |
| 72 | + self.trigger(_event_data) |
| 73 | + else: |
| 74 | + self.active = False |
| 75 | + except Exception as error: |
| 76 | + Logger.log(LOG_LEVEL["error"], |
| 77 | + f'Error evaluating thresholds for trigger {self.id}') |
| 78 | + Logger.log(LOG_LEVEL["debug"], error) |
| 79 | + self._previous_state = self.active |
| 80 | + |
| 81 | + def unload(self): |
| 82 | + # Unsubscribe once bus supports single handler unsubscribes |
| 83 | + return |
| 84 | + |
| 85 | + def _parse_data(self, data): |
| 86 | + """ Get nested data if set otherwise return the data """ |
| 87 | + if isinstance(data, dict): |
| 88 | + print('dict con') |
| 89 | + return data if not self.nested_source else data.get(self.nested_source, None) |
| 90 | + return data |
| 91 | + |
0 commit comments