More minor fixes
This commit is contained in:
parent
acabc57019
commit
c0d9f01e29
@ -104,10 +104,12 @@ class Entity:
|
||||
class LightEntity(Entity):
|
||||
|
||||
def set_state(self, old, new):
|
||||
if new == "on":
|
||||
if new == "on" or new == True:
|
||||
self._hass.call_service("light/turn_on", entity_id = self._entity)
|
||||
elif new == "off":
|
||||
self._state = "on"
|
||||
elif new == "off" or new == False:
|
||||
self._hass.call_service("light/turn_off", entity_id = self._entity)
|
||||
self._state = "off"
|
||||
|
||||
|
||||
class SwitchEntity(Entity):
|
||||
|
@ -7,7 +7,6 @@ class OutsideLights(Entities):
|
||||
self.register_entity(l, l)
|
||||
|
||||
self.listen_event(self.update, 'TOD_TOD')
|
||||
self.listen_event(self.update, 'TOD_DARK')
|
||||
self.run_in(lambda *_: self.update(None, None, None), 2)
|
||||
|
||||
def update(self, event, data, kwarg):
|
||||
|
@ -11,6 +11,8 @@ class TimeOfDay(Timers, Entities):
|
||||
|
||||
self.run_in(self._setup_inputs, 1)
|
||||
|
||||
self.updating = False
|
||||
|
||||
def _setup_inputs(self, kwargs):
|
||||
inputs = [
|
||||
'morning',
|
||||
@ -29,24 +31,20 @@ class TimeOfDay(Timers, Entities):
|
||||
del e['name']
|
||||
del e['default']
|
||||
self.register_entity(i, name, True, default, e)
|
||||
for i in inputs:
|
||||
self.e[i].listen(self._update, {'trigger': 'setting', 'entity': i})
|
||||
|
||||
self._update(None, None, {'trigger': 'init'})
|
||||
|
||||
def _update(self, old=None, new=None, kwarg=None):
|
||||
|
||||
if self.updating:
|
||||
return
|
||||
|
||||
if kwarg is None:
|
||||
kwarg = old
|
||||
trigger = kwarg.get('trigger', None)
|
||||
|
||||
# Tell listeners if Time Of Day or Dark has changed
|
||||
if kwarg.get('entity', None) == 'tod':
|
||||
self.fire_event('TOD_TOD', old = old, new = new)
|
||||
return
|
||||
if kwarg.get('entity', None) == 'dark':
|
||||
self.fire_event('TOD_DARK', old = old, new = new)
|
||||
return
|
||||
|
||||
self.log(f"TOD - updated by {trigger}")
|
||||
|
||||
# Set up triggers for each TOD
|
||||
@ -60,20 +58,20 @@ class TimeOfDay(Timers, Entities):
|
||||
)
|
||||
|
||||
# Set up triggers for sunrise and sunset
|
||||
if trigger != 'sunrise':
|
||||
sunrise = float(self.e['sunrise'].state)
|
||||
sunrise = timedelta(minutes=sunrise)
|
||||
sunrise = (self.sunrise() + sunrise).time()
|
||||
if trigger != 'sunrise':
|
||||
self.run_once(
|
||||
'sunrise',
|
||||
self._update,
|
||||
sunrise,
|
||||
trigger='sunrise'
|
||||
)
|
||||
if trigger != 'sunset':
|
||||
sunset = float(self.e['sunset'].state)
|
||||
sunset = timedelta(minutes=sunset)
|
||||
sunset = (self.sunset() + sunset).time()
|
||||
if trigger != 'sunset':
|
||||
self.run_once(
|
||||
'sunset',
|
||||
self._update,
|
||||
@ -104,14 +102,23 @@ class TimeOfDay(Timers, Entities):
|
||||
self.log(f"TOD - Time of day is {tod}, sun {'has set' if dark else 'is up'}")
|
||||
|
||||
# Update outputs
|
||||
self.updating = True
|
||||
self.e['dark'].state = dark
|
||||
self.e['dark'].push()
|
||||
self.e['tod'].state = tod
|
||||
self.e['tod'].push()
|
||||
self.e['sunrise'].attr['time'] = sunrise.strftime("%H:%M:%S")
|
||||
self.e['sunrise'].push()
|
||||
self.e['sunset'].attr['time'] = sunset.strftime("%H:%M:%S")
|
||||
self.e['sunset'].push()
|
||||
self.updating = False
|
||||
|
||||
# Tell listeners if Time Of Day or Dark has changed
|
||||
self.fire_event('TOD_TOD', old = old, new = new)
|
||||
|
||||
@property
|
||||
def tod(self):
|
||||
return self.e['tod'].state
|
||||
@property
|
||||
def dark(self):
|
||||
return self.e['dark'].state
|
||||
return self.e['dark'].state == "on"
|
||||
|
Loading…
x
Reference in New Issue
Block a user