Vorbereitung Release 0.2.0

kontor-scripts/read_list.py

@@ -1,12 +1,12 @@
 """
 read file with URLs and store in DB
 """
-import uuid
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
-import datetime
-
-import mariadb
-from setup import get_database_cursors, get_logger, get_scripts, get_meta_data
+import logging
+import json
+from proton import Message, Event
+from proton.handlers import MessagingHandler
+from proton.reactor import Container
 
 parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
 parser.add_argument('-f', '--links', help='file with links')
@@ -20,38 +20,61 @@ def read_links_file(links_file):
     return lines
 
 
-def add_link_to_db(statement, connection, video_url, log):
-    entry_id = str(uuid.uuid4())
-    current_date_time = datetime.datetime.now()
-    try:
-        cur = connection.cursor()
-        cur.execute(statement, (entry_id, current_date_time, current_date_time, 0, video_url, True, True, None, None, None, None))
-        connection.commit()
-        log.info(f'link {video_url} added to db')
-    except mariadb.Error as insert_error:
-        log.debug("insert failed with %s", insert_error)
-        entry_id = None
-    return entry_id
+class AddLinkMessage(MessagingHandler):
+    def __init__(self, server, url, log):
+        super(AddLinkMessage, self).__init__()
+        log.info("create AddLinkMessage")
+        self.server = server
+        self.address = "KontorMediaFile::add_link_file"
+        self.url = url
+        self.log = log
 
+    def on_start(self, event: Event):
+        self.log.info("Connection...")
+        conn = event.container.connect(self.server, user="artemis", password="artemis")
+        event.container.create_sender(conn, self.address)
+
+    def on_connection_opened(self, event: Event) -> None:
+        self.log.debug("connection open")
+
+    def on_connection_error(self, event: Event) -> None:
+        self.log.info(f"error: {event}")
+
+    def on_disconnected(self, event: Event) -> None:
+        self.log.debug(f"disconnected: {repr(event)}")
+
+
+    def on_sendable(self, event: Event):
+        self.log.info("send message")
+        message = Message(body=self.url, address=self.address, content_type="application/json", durable=True)
+        delivery = event.sender.send(message)
+        self.log.info(f"Delivery {delivery} sent")
+        event.connection.close()
+
+    def on_accepted(self, event: Event) -> None:
+        self.log.info(f"accepted Delivery: {event.delivery.remote_state}")
+
+
+    def on_rejected(self, event: Event) -> None:
+        self.log.info(f"rejected Delivery: {event.delivery}")
 
 if __name__ == '__main__':
-    logger = get_logger(args.verbose)
-    logger.info('kontor.read_list started')
-    s_conn, m_conn = get_database_cursors(logger)
-    meta_data_tables = get_meta_data(m_conn)
-    scripts = get_scripts(meta_data_tables, logger)
-    tables = {}
-    for table_id in scripts:
-        tables[scripts[table_id]['name']] = table_id
-    media_file_id = tables['media_file']
-    insert_statement = scripts[tables['media_file']]['insert_mariadb']
+    logging.basicConfig(level=logging.INFO, format='[%(asctime)s] {%(filename)s:%(lineno)d} %(levelname)s - %(message)s')
+    logging.info('kontor.read_list started')
+    #conn = stomp.Connection([('127.0.0.1', '61616')])
+    #conn.connect('artemis', 'artemis', wait=True)
     if args.links:
-        logger.info("read links from file")
+        logging.info("read links from file")
         links = read_links_file(args.links)
         for link in links:
-            logger.info("add link to db")
-            add_link_to_db(insert_statement, m_conn, link.strip(), logger)
-    else:
-        logger.info('script used: {}'.format(insert_statement))
-    logger.info('kontor.read_list finished')
+            data_dict = {'url': link.strip()}
+            data = json.dumps(data_dict)
+            logging.info("send link message")
+            handler = AddLinkMessage("amqp://127.0.0.1:5672", data, logging)
+            container = Container(handler)
+            container.container_id = "process_add_links"
+            container.run()
+            # conn.send(body=data, destination='KontorMediaFile::add_link_file', headers={'content-type': 'application/json'})
+    #conn.disconnect()
+    logging.info('kontor.read_list finished')