Update sensor.py

miumida · Apr 20, 2020 · 11e2daf · 11e2daf
1 parent de91599
commit 11e2daf
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/custom_components/seoul_bus/sensor.py b/custom_components/seoul_bus/sensor.py
@@ -584,12 +584,12 @@ def device_state_attributes(self):
         attr = {}
 
         for key in self._data:
-           attr[key] = self._data[key]
+           attr[_BUS_PROPERTIES[key]] = self._data[key]
 
-        attr['syncDate'] = self._sync_date
+        attr[_BUS_PROPERTIES['syncDate']] = self._sync_date
 
-        attr['isUpdate']   = self._isUpdate
-        attr['start time'] = self._stt_time
-        attr['end time']   = self._end_time
+        attr[_BUS_PROPERTIES['isUpdate']]   = self._isUpdate
+        attr[_BUS_PROPERTIES['start_time']] = self._stt_time
+        attr[_BUS_PROPERTIES['end_time']]   = self._end_time
 
         return attr