From ec557f8d44609f72d07500157741858b44555ca2 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <paulus@paulusschoutsen.nl>
Date: Sat, 21 Mar 2015 22:26:41 -0700
Subject: [PATCH] Fix broken tellstick and systemmonitor sensors
---
homeassistant/components/sensor/systemmonitor.py | 6 +++++-
homeassistant/components/sensor/tellstick.py | 6 +++++-
2 files changed, 10 insertions(+), 2 deletions(-)
diff --git a/homeassistant/components/sensor/systemmonitor.py b/homeassistant/components/sensor/systemmonitor.py
index 99e2453c487..e03e987802b 100644
--- a/homeassistant/components/sensor/systemmonitor.py
+++ b/homeassistant/components/sensor/systemmonitor.py
@@ -50,7 +50,7 @@ class SystemMonitorSensor(Entity):
self.argument = argument
self.type = sensor_type
self._state = None
- self.unit_of_measurement = SENSOR_TYPES[sensor_type][1]
+ self._unit_of_measurement = SENSOR_TYPES[sensor_type][1]
self.update()
@property
@@ -62,6 +62,10 @@ class SystemMonitorSensor(Entity):
""" Returns the state of the device. """
return self._state
+ @property
+ def unit_of_measurement(self):
+ return self._unit_of_measurement
+
def update(self):
if self.type == 'disk_use_percent':
self._state = psutil.disk_usage(self.argument).percent
diff --git a/homeassistant/components/sensor/tellstick.py b/homeassistant/components/sensor/tellstick.py
index 699821fa6b6..5720b65a669 100644
--- a/homeassistant/components/sensor/tellstick.py
+++ b/homeassistant/components/sensor/tellstick.py
@@ -99,7 +99,7 @@ class TellstickSensor(Entity):
def __init__(self, name, sensor, datatype, sensor_info):
self.datatype = datatype
self.sensor = sensor
- self.unit_of_measurement = sensor_info.unit or None
+ self._unit_of_measurement = sensor_info.unit or None
self._name = "{} {}".format(name, sensor_info.name)
@@ -112,3 +112,7 @@ class TellstickSensor(Entity):
def state(self):
""" Returns the state of the device. """
return self.sensor.value(self.datatype).value
+
+ @property
+ def unit_of_measurement(self):
+ return self._unit_of_measurement
--
GitLab