If decimal_places is 0, cast to int after rounding

This commit is contained in:
Philip Lundrigan 2015-12-09 15:16:42 -07:00
parent d84bea3621
commit 3938b9f3f6
3 changed files with 6 additions and 0 deletions

View File

@ -132,6 +132,8 @@ class ArestSensor(Entity):
value = float(value) * float(self._corr_factor) value = float(value) * float(self._corr_factor)
if self._decimal_places is not None: if self._decimal_places is not None:
value = round(value, self._decimal_places) value = round(value, self._decimal_places)
if self._decimal_places == 0:
value = int(value)
return value return value
else: else:
return values.get(self._variable, 'n/a') return values.get(self._variable, 'n/a')

View File

@ -79,6 +79,8 @@ class CommandSensor(Entity):
value = float(value) * float(self._corr_factor) value = float(value) * float(self._corr_factor)
if self._decimal_places is not None: if self._decimal_places is not None:
value = round(value, self._decimal_places) value = round(value, self._decimal_places)
if self._decimal_places == 0:
value = int(value)
self._state = value self._state = value
except ValueError: except ValueError:
self._state = value self._state = value

View File

@ -140,6 +140,8 @@ class RestSensor(Entity):
value = float(value) * float(self._corr_factor) value = float(value) * float(self._corr_factor)
if self._decimal_places is not None: if self._decimal_places is not None:
value = round(value, self._decimal_places) value = round(value, self._decimal_places)
if self._decimal_places == 0:
value = int(value)
self._state = value self._state = value
except ValueError: except ValueError:
self._state = RestSensor.extract_value(value, self._variable) self._state = RestSensor.extract_value(value, self._variable)