Merge pull request #30 from eesaanatluri/feat-subscribe-mail-lists

Feat subscribe mail lists

subscribe_mail_lists.py

+#!/usr/bin/env python
+import sys
+import json
+import smtplib
+import logging
+import argparse
+import rc_util
+from email.message import EmailMessage
+from rc_rmq import RCRMQ
+
+task = 'subscribe_mail_list'
+
+# Instantiate rabbitmq object
+rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'})
+
+# Parse arguments
+args = rc_util.get_args()
+
+# Logger
+logger = rc_util.get_logger()# Define your callback function
+
+def mail_list_subscription(ch, method, properties, body):
+
+    # Retrieve message
+    msg = json.loads(body)
+    logger.info("Received msg {}".format(msg))
+    username = msg['username']
+    fullname = msg['fullname']
+    email = msg['email']
+
+    mail_list_admin = 'root@localhost' #change this during deploy
+    mail_list = 'LISTSERV@LISTSERV.UAB.EDU'
+
+    listserv_cmd = f'QUIET ADD hpc-announce {email} {fullname} \
+                   \nQUIET ADD hpc-users {email} {fullname}'
+
+    logger.info("Adding user{} to mail list".format(username))
+    success = False
+    try:
+        # Create a text/plain message
+        email_msg = EmailMessage()
+
+        email_msg['From'] = mail_list_admin
+        email_msg['To'] = mail_list
+        email_msg['Subject'] = ''
+
+        # Create an smtp object and send email
+        s = smtplib.SMTP('localhost')
+
+        email_msg.set_content(listserv_cmd)
+        if not args.dry_run:
+            s.send_message(email_msg)
+        logging.info(f'This email will add user {username} to listserv \n{email_msg}')
+
+        s.quit()
+        msg['task'] = task
+        msg['success'] = True
+    except Exception:
+        logger.exception("Fatal error:")
+
+    # Acknowledge message
+    ch.basic_ack(delivery_tag=method.delivery_tag)
+
+    # send confirm message
+    logger.debug('rc_rmq.publish_msg()')
+    rc_rmq.publish_msg({
+        'routing_key': 'confirm.' + username,
+        'msg': msg
+    })
+    logger.info('confirmation sent')
+
+logger.info("Start listening to queue: {}".format(task))
+rc_rmq.start_consume({
+    'queue': task,      # Define your Queue name
+    'routing_key': "verify.*", # Define your routing key
+    'cb': mail_list_subscription # Pass in callback function you just define
+})
+
+logger.info("Disconnected")
+rc_rmq.disconnect()