use msgspec to parse message

kontor-scripts/read_queue.py

@@ -8,42 +8,23 @@ import msgspec
 from pathlib import Path
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
 from platformdirs import PlatformDirs
+from log import get_logger
 
 
 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")
+parser.add_argument("--port", "-p", default="61616")
 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, log):
+    def __init__(self, log, conn):
         self.log = log
+        self.conn = conn
         pass
 
     def on_error(self, frame):
@@ -53,14 +34,17 @@ class MyListener(stomp.ConnectionListener):
         self.log.info("received a message %s", frame.body)
         link = msgspec.json.decode(frame.body, type=Link)
         self.log.info("found link: %s", link.url)
+        json_bytes = msgspec.json.encode(link)
+        self.conn.send(body=json_bytes, destination="add_link_accepted")
+        self.conn.send(body=json_bytes, destination="update_title")
 
 
 if __name__ == '__main__':
-    logger = get_logger(args.verbose, args.config)
+    logger = get_logger(args.verbose, __file__)
     logger.info("kontor.read_queue started")
-    host = [(args.server, 61616)]
+    host = [(args.server, args.port)]
     conn = stomp.Connection(host_and_ports=host)
-    conn.set_listener('', MyListener(logger))
+    conn.set_listener('', MyListener(logger, conn))
     conn.connect(username='artemis', passcode='artemis', wait=True)
     conn.subscribe(destination='add_link', id=1, ack='auto', headers={})
     time.sleep(5)