Thomas Peetz
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""
|
|
read file with URLs and store in DB
|
|
"""
|
|
|
|
from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
|
|
|
|
import msgspec
|
|
import stomp
|
|
|
|
from log import get_logger
|
|
|
|
parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
|
parser.add_argument("-u", "--url", help="link")
|
|
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")
|
|
parser.add_argument("--destination", "-d", default="media.link")
|
|
args = parser.parse_args()
|
|
|
|
|
|
class Link(msgspec.Struct):
|
|
url: str
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger = get_logger(args.verbose, args.config)
|
|
logger.info("kontor.add_link_stomp started")
|
|
|
|
server_url = [(args.server, args.port)]
|
|
conn = stomp.Connection(host_and_ports=server_url)
|
|
conn.connect(username="artemis", passcode="artemis", wait=True)
|
|
|
|
link: Link = Link(url=args.url)
|
|
json_bytes = msgspec.json.encode(link)
|
|
conn.send(body=json_bytes.decode(), destination=args.destination)
|
|
|
|
logger.info("kontor.add_link finished")
|