Thomas Peetz
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
import logging
|
|
|
|
import stomp
|
|
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, log):
|
|
self.log = log
|
|
pass
|
|
|
|
def on_error(self, frame):
|
|
self.log.info("received an error %s", frame.body)
|
|
|
|
def on_message(self, frame):
|
|
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)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
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(logger))
|
|
conn.connect(username='artemis', passcode='artemis', wait=True)
|
|
conn.subscribe(destination='add_link', id=1, ack='auto', headers={})
|
|
time.sleep(5)
|
|
conn.disconnect()
|
|
logger.info("kontor.read_queue finished")
|