Skip to content
Snippets Groups Projects
Commit 1bb3b2b5 authored by Eesaan Atluri's avatar Eesaan Atluri
Browse files

Use rc_util class methods for logging and arg parsing in subscribe mail lists agent

parent e375efb7
No related branches found
No related tags found
6 merge requests!147Merge previous default branch feat-cod-rmq into main,!85kill nginx process running under user from login node,!51Fix acct create wait,!39WIP:Feat cod rmq,!38WIP: Feat cod rmq,!30Feat subscribe mail lists
...@@ -4,6 +4,7 @@ import json ...@@ -4,6 +4,7 @@ import json
import smtplib import smtplib
import logging import logging
import argparse import argparse
import rc_util
from email.message import EmailMessage from email.message import EmailMessage
from rc_rmq import RCRMQ from rc_rmq import RCRMQ
...@@ -13,24 +14,11 @@ task = 'subscribe_mail_list' ...@@ -13,24 +14,11 @@ task = 'subscribe_mail_list'
rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'}) rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'})
# Parse arguments # Parse arguments
parser = argparse.ArgumentParser() args = rc_util.get_args()
parser.add_argument('-v', '--verbose', action='store_true', help='verbose output')
parser.add_argument('-n', '--dry-run', action='store_true', help='enable dry run mode')
args = parser.parse_args()
#Default Log level
log_lvl = logging.WARNING
if args.verbose:
log_lvl = logging.DEBUG
if args.dry_run:
log_lvl = logging.INFO
# Logger # Logger
logging.basicConfig(format='%(asctime)s %(levelname)s [%(module)s] - %(message)s', level=log_lvl) logger = rc_util.get_logger()# Define your callback function
logger = logging.getLogger(__name__)
# Define your callback function
def mail_list_subscription(ch, method, properties, body): def mail_list_subscription(ch, method, properties, body):
# Retrieve message # Retrieve message
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment