cube updates

apps/cubes/cube.py

@@ -1,35 +1,52 @@
 import json
+import sys
 from copy import deepcopy
+from pathlib import Path
 
 from appdaemon.plugins.hass.hassapi import Hass
 from appdaemon.plugins.mqtt.mqttapi import Mqtt
 
+rc_path = (Path(__file__).resolve().parents[1] / 'room_control').as_posix()
+sys.path.insert(0, rc_path)
+
+from console import console, init_logging, deinit_logging
+
 
 class AqaraCube(Hass, Mqtt):
     def initialize(self):
+        if (level := self.args.get('rich', False)):
+            init_logging(self, level)
+
         topic = f'zigbee2mqtt/{self.args["cube"]}'
         self.mqtt_subscribe(topic, namespace='mqtt')
-        self.listen_event(self.handle_event, "MQTT_MESSAGE", topic=topic, namespace='mqtt')
+        self.listen_event(self.handle_event, 'MQTT_MESSAGE', topic=topic, namespace='mqtt')
         self.log(f'Listening for cube events on: {topic}')
 
         self.app = self.get_app(self.args['app'])
         self.log(f'Connected to other app: {self.app.name}')
 
-    def handle_event(self, event_name, data, kwargs):
+    def terminate(self):
-        topic = data['topic']
+        self.log('[bold red]Terminating[/]', level='DEBUG')
+        deinit_logging(self)
+        self.log('Success', level='DEBUG')
+
+    def handle_event(self, event_name, data, cb_args):
+        topic_prefix, cube_name = data['topic'].split('/')
+        self.log(f'{event_name} on [bright_black]{topic_prefix}/[/][blue]{cube_name}[/]', data, level='DEBUG')
+
         payload = json.loads(data['payload'])
         action = payload['action']
-        
-        if action == '':
-            return
-        if (arg := self.args.get(action, None)) is not None:
-            self.action_handler(action=action, description=arg)
-        elif (handler := getattr(self, f'handle_{action}', None)):
-            handler(payload)
-        else:
-            self.log(f'Unhandled cube action: {action}')
 
-    def action_handler(self, action: str , description: str):
+        if action == '' or action == 'wakeup':
+            return
+        else:
+            self.log(json.dumps(payload, indent=2), level='DEBUG')
+            if (arg := self.args.get(action, False)):
+                self.action_handler(action=action, description=arg)
+            elif handler := getattr(self, f'handle_{action}', None):
+                handler(payload)
+
+    def action_handler(self, action: str, description: str):
         self.log(f'{self.args["cube"]}: {action}: {description}')
 
         if description == 'activate':
@@ -38,25 +55,28 @@ class AqaraCube(Hass, Mqtt):
         elif description.startswith('scene.'):
             self.call_service('scene/turn_on', entity_id=description, namespace='default')
             self.log(f'Turned on {description}')
-        
+
         elif description.startswith('toggle'):
             cause = f'{self.args["cube"]} {action}'
             self.app.toggle_activate(kwargs={'cause': cause})
 
+        else:
+            self.log(f'Unhandled action: {action}', level='WARNING')
+
     # def handle_rotate_right(self, payload):
     #     self.log(f'{self.args["cube"]}: Rotate right')
-    
+
     # def handle_rotate_left(self, payload):
     #     self.log(f'{self.args["cube"]}: Rotate left')
-    
+
     # def handle_slide(self, payload):
     #     self.log(f'{self.args["cube"]}: Slide')
 
     # def handle_flip180(self, payload):
     #     self.log(f'{self.args["cube"]}: Flipped 180')
-    
+
     # def handle_flip90(self, payload):
     #     self.log(f'{self.args["cube"]}: Flipped 90')
-    
+
     # def handle_shake(self, payload):
     #     self.log(f'{self.args["cube"]}: Shake')

apps/cubes/cube1.yaml

@@ -1,8 +1,9 @@
 cube1:
   module: cube
   class: AqaraCube
+  rich: INFO
   cube: Cube 1
   app: bedroom
   flip90: toggle
   flip180: scene.bedsport
-  shake: activate
+  # shake: activate