def __init__(self): configuration = ConfigurationStore().get() self.maker_id = configuration.get_maker_id() self.device_id = configuration.get_device_id() self.device_status = configuration.get_device_status() self.bot_service = BoTService()
class ActionService: def __init__(self): self.configuration = ConfigurationStore().get() self.bot_service = BoTService() self.key_generator = KeyGenerator() self.store = Store() def get_actions(self): Logger.info(LOCATION, 'Retrieving actions...') try: actions = self.bot_service.get(ACTIONS_ENDPOINT) Logger.success( LOCATION, 'Successfully retrieved ' + str(len(actions)) + ' action(s) from server') Store.set_actions(actions) return actions except falcon.HTTPServiceUnavailable: Logger.warning( LOCATION, 'Unable to retrieve actions from server. Loading locally stored action(s)...' ) actions = self.store.get_actions() Logger.success( LOCATION, 'Successfully loaded ' + str(len(actions)) + ' cached action(s)') return actions def trigger(self, action_id, value=None, alternative_id=None): Logger.info(LOCATION, 'Triggering action: ' + action_id) action = self._get_action(action_id) self._validate_frequency(action) Logger.success(LOCATION, 'Action valid') data = self._create_trigger_body(action_id, value, alternative_id) try: self.bot_service.post(ACTIONS_ENDPOINT, data) Logger.success(LOCATION, 'Successfully triggered action: ' + action_id) self.store.set_last_triggered(action_id, time.time()) return True # TODO : Make exception more specific except: Logger.error(LOCATION, 'Unable to trigger action: ' + action_id) return False def _validate_frequency(self, action): last_triggered = self.store.get_last_triggered(action[ACTION_ID]) if last_triggered is None: return # It was never triggered, so it is valid, unless we ever introduce Frequency: 'never' frequency = action[FREQUENCY] if frequency not in FrequenciesInSeconds.keys(): self._handle_unsupported_frequency(frequency) if FrequenciesInSeconds[frequency] > time.time() - last_triggered: self._handle_maximum_frequency(frequency) def _get_action(self, action_id): actions = self.get_actions() for action in actions: if action[ACTION_ID] == action_id: Logger.success(LOCATION, 'Action found') return action Logger.error(LOCATION, 'Action not found') raise falcon.HTTPNotFound(description='Action not found') def _create_trigger_body(self, action_id, value, alternative_id): data = { ACTION_ID: action_id, DEVICE_ID: self.configuration.get_device_id(), QUEUE_ID: self.key_generator.generate_uuid() } if alternative_id is not None: data[ALTERNATIVE_ID] = str(alternative_id) if value is not None: data[VALUE] = str(value) return data @staticmethod def _handle_unsupported_frequency(frequency): error = 'Frequency not supported: ' + frequency Logger.error(LOCATION, error) raise falcon.HTTPBadRequest(description=error) @staticmethod def _handle_maximum_frequency(frequency): error = 'Maximum ' + frequency + ' triggers reached' Logger.error(LOCATION, error) raise falcon.HTTPBadRequest(description=error)