small bugfix

button.py

@@ -5,12 +5,10 @@ from appdaemon.plugins.mqtt.mqttapi import Mqtt
 from room_control import RoomController
 
 
-class ButtonController(Mqtt):
-    def initialize(self):
-        task = self.get_app(self.args['app'])
-        self.app: RoomController = asyncio.get_event_loop().run_until_complete(task)
+class Button(Mqtt):
+    async def initialize(self):
+        self.app: RoomController = await self.get_app(self.args['app'])
         self.setup_buttons(self.args['button'])
-        # self.log(f'Done')
 
     def setup_buttons(self, buttons):
         if isinstance(buttons, list):
@@ -26,29 +24,27 @@ class ButtonController(Mqtt):
         self.log(f'"{topic}" controls app {self.app.name}')
 
     async def handle_button(self, event_name, data, kwargs):
-        topic = data['topic']
-        self.log(f'Button event for: {topic}')
         try:
             payload = json.loads(data['payload'])
             action = payload['action']
-            button = kwargs['button']
         except json.JSONDecodeError:
             self.log(f'Error decoding JSON from {data["payload"]}', level='ERROR')
         except KeyError as e:
             return
         else:
-            self.log(f'{button}: {action}')
-            await self.handle_action(action)
+            if action != '':
+                await self.handle_action(action)
 
     async def handle_action(self, action: str):
-        if action == '':
-            return
-        elif action == 'single':
-            cause = 'button single click'
-            state = await self.get_state(entity_id=self.args['ref_entity'])
+        if action == 'single':
+            self.log(f' {action.upper()} '.center(50, '='))
+            state = await self.get_state(self.args['ref_entity'])
+            kwargs = {
+                'kwargs': {'cause': f'button single click: toggle while {state}'}
+            }
             if state == 'on':
-                self.app.deactivate(cause=cause)
+                self.app.deactivate(**kwargs)
             else:
-                await self.app.activate(cause=cause)
+                await self.app.activate(**kwargs)
         else:
             pass

door.py

@@ -4,8 +4,5 @@ from room_control import RoomController
 
 class Door(Hass):
     async def initialize(self):
-        await self.listen_state(self.door_open, entity_id=self.args['door'], new='on')
-    
-    async def door_open(self, entity, attribute, old, new, kwargs):
         app: RoomController = await self.get_app(self.args['app'])
-        await app.activate_all_off()
+        await self.listen_state(app.activate_all_off, entity_id=self.args['door'], new='on', cause='door open')

motion.py

@@ -1,28 +1,12 @@
-import asyncio
-from datetime import timedelta
 import re
+from datetime import timedelta
 
 from appdaemon.entity import Entity
 from appdaemon.plugins.hass.hassapi import Hass
 from room_control import RoomController
 
+from appdaemon import utils
 
-class CustomEventLoopPolicy(asyncio.DefaultEventLoopPolicy):
-    def get_event_loop(self):
-        try:
-            # Try to get the current event loop
-            loop = super().get_event_loop()
-        except RuntimeError as ex:
-            if "There is no current event loop" in str(ex):
-                # If there's no current loop, create a new one and set it
-                loop = self.new_event_loop()
-                self.set_event_loop(loop)
-            else:
-                raise
-        return loop
-
-# Set the custom event loop policy
-asyncio.set_event_loop_policy(CustomEventLoopPolicy())
 
 class Motion(Hass):
     @property
@@ -38,50 +22,39 @@ class Motion(Hass):
         return self.get_entity(self.args['ref_entity'])
 
     @property
-    def ref_entity_state(self) -> bool:
-        return self.ref_entity.get_state() == 'on'
+    async def ref_entity_state(self) -> bool:
+        return (await self.ref_entity.get_state()) == 'on'
 
-    def initialize(self):
-        try:
-            loop = asyncio.get_event_loop()
-        except RuntimeError:
-            loop = asyncio.new_event_loop()
-
-        self.app: RoomController = loop.run_until_complete(self.get_app(self.args['app']))
+    async def initialize(self):
+        self.app: RoomController = await self.get_app(self.args['app'])
         self.log(f'Connected to app {self.app.name}')
 
-        self.listen_state(self.callback_light_on, self.ref_entity.entity_id, new='on')
-        self.listen_state(self.callback_light_off, self.ref_entity.entity_id, new='off')
-
-        loop.run_until_complete(self.sync_state())
-
-    async def sync_state(self):
-        """Synchronizes the callbacks with the state of the light.
-
-        Essentially mimics the `state_change` callback based on the current state of the light.
-        """
-        if self.ref_entity_state:
-            await self.callback_light_on()
-        else:
-            await self.callback_light_off()
+        base_kwargs = dict(
+            entity_id=self.ref_entity.entity_id,
+            immediate=True, # avoids needing to sync the state
+        )
+        # don't need to await these because they'll already get turned into a task by the utils.sync_wrapper decorator
+        self.listen_state(**base_kwargs, attribute='brightness', callback=self.callback_light_on)
+        self.listen_state(**base_kwargs, new='off', callback=self.callback_light_off)
 
     async def listen_motion_on(self):
         """Sets up the motion on callback to activate the room
         """
-        self.log(f'Waiting for motion on {self.sensor.friendly_name}')
-        self.motion_on_handle = await self.listen_state(
+        await self.cancel_motion_callback()
+        await self.listen_state(
             callback=self.app.activate_all_off,
             entity_id=self.sensor.entity_id,
             new='on',
             oneshot=True,
             cause='motion on'
         )
+        self.log(f'Waiting for motion on {self.sensor.friendly_name}')
 
     async def listen_motion_off(self, duration: timedelta):
         """Sets up the motion off callback to deactivate the room
         """
-        self.log(f'Waiting for motion to stop on {self.sensor.friendly_name}')
-        self.motion_off_handle = await self.listen_state(
+        await self.cancel_motion_callback()
+        await self.listen_state(
             callback=self.app.deactivate,
             entity_id=self.sensor.entity_id,
             new='off',
@@ -89,19 +62,22 @@ class Motion(Hass):
             oneshot=True,
             cause='motion off'
         )
+        self.log(f'Waiting for motion to stop on {self.sensor.friendly_name} for {duration}')
 
+    @utils.sync_wrapper
     async def callback_light_on(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
         """Called when the light turns on
         """
-        self.log('Light on callback')
-        await self.cancel_motion_callback(new='on')
-        await self.listen_motion_off(await self.app.off_duration())
+        if new is not None:
+            self.log(f'{entity} turned on')
+            duration = await self.app.off_duration()
+            await self.listen_motion_off(duration)
 
+    @utils.sync_wrapper
     async def callback_light_off(self, entity=None, attribute=None, old=None, new=None, kwargs=None):
         """Called when the light turns off
         """
-        self.log('Light off callback')
-        await self.cancel_motion_callback(new='off')
+        self.log(f'{entity} turned off')
         await self.listen_motion_on()
 
     async def get_app_callbacks(self, name: str = None):
@@ -123,13 +99,13 @@ class Motion(Hass):
             if info['entity'] == self.sensor.entity_id
         }
 
-    async def cancel_motion_callback(self, new: str):
+    async def cancel_motion_callback(self):
         callbacks = await self.get_sensor_callbacks()
-        # self.log(f'Found {len(callbacks)}')
+        # self.log(f'Found {len(callbacks)} callbacks for {self.sensor.entity_id}')
         for handle, info in callbacks.items():
             entity = info["entity"]
-            new_match = re.match('new=(?P<new>.*?)\s', info['kwargs'])
-            # self.log(f'{handle}: {info["entity"]}: {info["kwargs"]}')
-            if new_match is not None and new_match.group("new") == new:
+            kwargs = info['kwargs']
+            if (m := re.match('new=(?P<new>.*?)\s', kwargs)) is not None:
+                new = m.group('new')
                 await self.cancel_listen_state(handle)
-                self.log(f'cancelled: {await self.friendly_name(entity)}: {new}')
+                self.log(f'cancelled callback for sensor {entity} turning {new}')

room_control.py

@@ -1,7 +1,7 @@
 import asyncio
 from copy import deepcopy
-from datetime import time, timedelta
-from typing import List
+from datetime import datetime, time, timedelta
+from typing import Dict, List
 
 import appdaemon.utils as utils
 import astral
@@ -22,7 +22,7 @@ class RoomController(Hass, Mqtt):
 
     async def initialize(self):
         self.app_entities = await self.gather_app_entities()
-        self.log(f'entities: {self.app_entities}')
+        # self.log(f'entities: {self.app_entities}')
         await self.refresh_state_times()
         await self.run_daily(callback=self.refresh_state_times, start='00:00:00')
 
@@ -62,15 +62,16 @@ class RoomController(Hass, Mqtt):
 
         # schedule the transitions
         for state in self.states:
-            dt = str(state['time'])[:8]
-            self.log(f'Scheduling transition at: {dt}')
+            t: time = state['time']
             try:
-                await self.run_at(callback=self.activate_any_on, start=dt)
+                await self.run_at(callback=self.activate_any_on, start=t.strftime('%H:%M:%S'), cause='scheduled transition')
             except ValueError:
                 # happens when the callback time is in the past
                 pass
             except Exception as e:
                 self.log(f'Failed with {type(e)}: {e}')
+            else:
+                self.log(f'Scheduled transition at: {t.strftime("%I:%M:%S %p")}')
 
     async def parse_states(self):
         async def gen():
@@ -98,47 +99,61 @@ class RoomController(Hass, Mqtt):
                 yield state
 
         states = [s async for s in gen()]
-        states = sorted(states, key=lambda s: s['time'])
+        states = sorted(states, key=lambda s: s['time'], reverse=True)
         return states
 
     async def current_state(self, time: time = None):
         if (await self.sleep_bool()):
+            self.log(f'sleep: active')
             if (state := self.args.get('sleep_state')):
                 return state
             else:
                 return {}
         else:
-            now = await self.get_now()
-            self.log(f'Getting state for datetime: {now}')
+            # now: datetime = await self.get_now()
+            # self.log(f'Getting state for datetime: {now.strftime("%I:%M:%S %p")}')
             time = time or (await self.get_now()).time()
-            for state in self.states[::-1]:
+            time_fmt = "%I:%M %p"
+            self.log(f'Getting state before: {time.strftime(time_fmt)}')
+            for state in self.states:
+                time_str = state["time"].strftime(time_fmt)
                 if state['time'] <= time:
-                    self.log(f'Selected state from {state["time"]}')
+                    self.log(f'Selected state from {time_str}')
                     return state
+                else:
+                    self.log(f'Not {time_str}')
             else:
-                return self.states[-1]
+                self.log(f'Defaulting to first state')
+                return self.states[0]
 
     async def current_scene(self, time: time = None):
         if (state := (await self.current_state(time=time))) is not None:
             return state['scene']
 
-    @property
-    def all_off(self) -> bool:
+    async def app_entity_states(self) -> Dict[str, str]:
+        states = {
+            entity: (await self.get_state(entity))
+            for entity in self.app_entities
+        }
+        return states
+
+    async def all_off(self) -> bool:
         """"All off" is the logic opposite of "any on"
 
         Returns:
             bool: Whether all the lights associated with the app are off
         """
-        return all(self.get_state(entity) != 'on' for entity in self.app_entities)
+        states = await self.app_entity_states()
+        return all(state != 'on' for entity, state in states.items())
 
-    @property
-    def any_on(self) -> bool:
+    async def any_on(self) -> bool:
         """"Any on" is the logic opposite of "all off"
 
         Returns:
             bool: Whether any of the lights associated with the app are on
         """
-        return any(self.get_state(entity) == 'on' for entity in self.app_entities)
+        states = await self.app_entity_states()
+        return any(state == 'on' for entity, state in states.items())
 
     async def sleep_bool(self) -> bool:
         if (sleep_var := self.args.get('sleep')):
@@ -178,7 +193,13 @@ class RoomController(Hass, Mqtt):
         except Exception:
             return timedelta()
 
-    async def activate(self, *args, cause: str = 'unknown', **kwargs):
+    @utils.sync_wrapper
+    async def activate(self, entity = None, attribute = None, old = None, new = None, kwargs = None):
+        if kwargs is not None:
+            cause = kwargs.get('cause', 'unknown')
+        else:
+            cause = 'unknown'
+            
         self.log(f'Activating: {cause}')
         scene = await self.current_scene()
 
@@ -202,26 +223,27 @@ class RoomController(Hass, Mqtt):
         else:
             self.log(f'ERROR: unknown scene: {scene}')
 
+    @utils.sync_wrapper
     async def activate_all_off(self, *args, **kwargs):
-        """Activate if all of the entities are off
+        """Activate if all of the entities are off. Args and kwargs are passed directly to self.activate()
         """
-        if self.all_off:
-            self.log(f'Activate all off kwargs: {kwargs}')
-            await self.activate(*args, **kwargs)
+        if (await self.all_off()):
+            self.activate(*args, **kwargs)
         else:
             self.log(f'Skipped activating - everything is not off')
 
+    @utils.sync_wrapper
     async def activate_any_on(self, *args, **kwargs):
-        """Activate if any of the entities are on
+        """Activate if any of the entities are on. Args and kwargs are passed directly to self.activate()
         """
-        if self.any_on:
-            await self.activate(*args, **kwargs)
+        if (await self.any_on()):
+            self.activate(*args, **kwargs)
         else:
             self.log(f'Skipped activating - everything is off')
 
-    def deactivate(self, *args, cause: str = 'unknown', **kwargs):
+    def deactivate(self, entity = None, attribute = None, old = None, new = None, kwargs = None):
+        cause = kwargs.get('cause', 'unknown')
         self.log(f'Deactivating: {cause}')
-        for entity in self.app_entities:
-            self.turn_off(entity)
-            self.log(f'Turned off {entity}')
-
+        for e in self.app_entities:
+            self.turn_off(e)
+            self.log(f'Turned off {e}')