def create_set_request(self, setting, value=None):
address = self._key_to_address(setting)
if value is not None:
entry = Setting.Entry(key=setting, value=value)
data = Setting(entries=[entry]).SerializeToString()
else:
data = None
return self._factory.create_set_request({address: data})
def _get_setting_entry(context, address):
setting = Setting()
try:
entries_list = context.get_state([address], timeout=STATE_TIMEOUT_SEC)
except FutureTimeoutError:
LOGGER.warning('Timeout occured on context.get_state([%s])', address)
raise InternalError('Unable to get {}'.format(address))
if entries_list:
setting.ParseFromString(entries_list[0].data)
return setting