use msgspec to parse message

kontor-scripts/read_queue.py

@@ -1,39 +1,65 @@
 import logging
 
 import stomp
-import json
 import time
+import yaml
 import msgspec
+from pathlib import Path
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
+from platformdirs import PlatformDirs
+
 
 parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
+parser.add_argument("--config", "-c", default="kontor-docker")
 parser.add_argument('--verbose', '-v', action='count', default=0)
 parser.add_argument("--server", "-s", default="127.0.0.1")
 args = parser.parse_args()
 logger = logging.getLogger(__name__)
 
+def get_logger(level: int, config: str):
+    """
+    create Logger with configuration from config file
+    """
+    dirs = PlatformDirs(config)
+    logging_config = Path(dirs.user_config_dir, "logging-config.yaml")
+    with open(logging_config, "rt", encoding="UTF-8") as f:
+        config_dict = yaml.safe_load(f.read())
+    logging.config.dictConfig(config_dict)
+    log = logging.getLogger("development")
+    if level is not None:
+        match level:
+            case 0:
+                log.setLevel(logging.INFO)
+            case 1:
+                log.setLevel(logging.DEBUG)
+            case _:
+                log.setLevel(logging.CRITICAL)
+    return log
+
+
 class Link(msgspec.Struct):
     url: str
 
 class MyListener(stomp.ConnectionListener):
-    def __init__(self):
+    def __init__(self, log):
+        self.log = log
         pass
 
     def on_error(self, frame):
-        logger.info("received an error %s", frame.body)
+        self.log.info("received an error %s", frame.body)
 
     def on_message(self, frame):
-        logger.info("received a message %s", frame.body)
+        self.log.info("received a message %s", frame.body)
         link = msgspec.json.decode(frame.body, type=Link)
-        logger.info("found link: %s", link.url)
+        self.log.info("found link: %s", link.url)
 
 
 if __name__ == '__main__':
-    logger.setLevel(logging.INFO)
+    logger = get_logger(args.verbose, args.config)
     logger.info("kontor.read_queue started")
     host = [(args.server, 61616)]
     conn = stomp.Connection(host_and_ports=host)
-    conn.set_listener('', MyListener())
+    conn.set_listener('', MyListener(logger))
     conn.connect(username='artemis', passcode='artemis', wait=True)
     conn.subscribe(destination='add_link', id=1, ack='auto', headers={})
     time.sleep(5)