seperate rich logging system

console.py

@@ -8,9 +8,15 @@ from rich.highlighter import NullHighlighter
 from rich.logging import RichHandler
 from rich.theme import Theme
 
+
 console = Console(
     width=150,
-    theme=Theme({'appname': 'italic bright_cyan'}),
+    theme=Theme({
+        # 'appname': 'italic bright_cyan',
+        'room': 'italic bright_cyan',
+        'component': 'violet'
+    }),
+    log_time_format='%Y-%m-%d %I:%M:%S %p',
 )
 
 
@@ -22,8 +28,34 @@ class UnMarkupFormatter(AppNameFormatter):
         return self.md_regex.sub(r'\g<text>', result)
 
 
-def create_handler() -> RichHandler:
-    handler = RichHandler(
+class RoomControllerFormatter(logging.Formatter):
+    def __init__(self, room: str, component: str = None):
+        self.log_fields = {'room': room}
+
+        fmt = '[room]{room:>12}[/]'
+        if component is not None:
+            fmt += ' [component]{component:<9}[/]'
+            self.log_fields['component'] = component
+        fmt += ' {message}'
+
+        datefmt = '%Y-%m-%d %I:%M:%S %p'
+        style = '{'
+        validate=True
+
+        super().__init__(fmt, datefmt, style, validate)
+        # console.print(f'Format: [bold yellow]{fmt}[/]')
+        
+    def format(self, record: logging.LogRecord):
+        parts = record.name.split('.')
+        record.room = parts[1]
+        if len(parts) == 3:
+            record.component = parts[2]
+
+        return super().format(record)
+
+
+def new_handler() -> RichHandler:
+    return RichHandler(
         console=console,
         highlighter=NullHighlighter(),
         markup=True,
@@ -31,10 +63,24 @@ def create_handler() -> RichHandler:
         omit_repeated_times=False,
         log_time_format='%Y-%m-%d %I:%M:%S %p',
     )
-    handler.setFormatter(AppNameFormatter(fmt='[appname]{appname}[/] {message}', style='{'))
+
+
+def setup_handler(**kwargs) -> RichHandler:
+    handler = new_handler()
+    handler.setFormatter(RoomControllerFormatter(**kwargs))
     return handler
 
 
+def setup_component_logging(self):
+    typ = type(self).__name__
+    logger = logging.getLogger(f'room_control.{self.args["app"]}')
+    self.logger = logger.getChild(typ)
+    if len(self.logger.handlers) == 0:
+        self.logger.setLevel(logging.INFO)
+        self.logger.addHandler(setup_handler(room=self.args["app"], component=typ))
+        self.logger.propagate = False
+
+
 def init_logging(self: ADAPI, level):
     for h in logging.getLogger('AppDaemon').handlers:
         og_formatter = h.formatter
@@ -45,7 +91,7 @@ def init_logging(self: ADAPI, level):
     if not any(isinstance(h, RichHandler) for h in self.logger.handlers):
         self.logger.propagate = False
         self.logger.setLevel(level)
-        self.logger.addHandler(create_handler())
+        self.logger.addHandler(setup_handler())
         self.log(f'Added rich handler for [bold green]{self.logger.name}[/]')
         # self.log(f'Formatter for RichHandler: {handler.formatter}')