Fix in code logic

Serios · Aug 5, 2019 · dd7738e · dd7738e
1 parent 9421f63
commit dd7738e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/custom_components/varna_public_transport/sensor.py b/custom_components/varna_public_transport/sensor.py
@@ -161,6 +161,9 @@ def device_state_attributes(self):
 
                           attr['lines'] = lines_count
                       i += 1
+                else:
+                    attr['stop_lines'] = {}
+                    attr['lines'] = 0
 
             if self._mode == 'schedule' or self._mode == 'all':
 
@@ -201,6 +204,9 @@ def device_state_attributes(self):
 
                               attr['lines_schedules'] = schedules_lines_count
                           i += 1
+                    else:
+                        attr['stop_lines_schedules'] = {}
+                        attr['lines_schedules'] = 0
 
             _LOGGER.debug(attr)
             return attr