brightcm_account_create.py

#!/usr/bin/env python
import sys
import json
import logging
import argparse
from os import popen
from rc_rmq import RCRMQ

task = 'bright_account'

# Instantiate rabbitmq object
rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'})

# Parse arguments
parser = argparse.ArgumentParser()
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
logging.basicConfig(format='%(asctime)s %(levelname)s [%(module)s] - %(message)s', level=log_lvl)
logger = logging.getLogger(__name__)

# Define your callback function
def bright_account_create(ch, method, properties, body):
    # Retrieve message
    msg = json.loads(body)
    logger.info("Received {}".format(msg))
    username = msg['username']
    uid = msg['uid']
    email = msg['email']
    fullname = msg['fullname']
    success = False
    try:
        # Bright command to create user
        cmd = '/cm/local/apps/cmd/bin/cmsh -c '
        cmd += f'"user; add {username}; set userid {uid}; set email {email}; set commonname \\"{fullname}\\"; '
        cmd += 'commit;"'

        if not args.dry_run:
            popen(cmd)
        logger.info(f'Bright command to create user:{cmd}')
        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': {
            'task': task,
            'success': success
        }
    })
    logger.info('confirmation sent')

    if success:
        # send create message to verify dir permissions agent
        logger.debug(f'The task {task} finished successfully')
        rc_rmq.publish_msg({
            'routing_key': 'verify.' + username,
            'msg': msg
        })
        logger.info('verify msg sent to next agent')

logger.info("Start listening to queue: {}".format(task))
rc_rmq.start_consume({
    'queue': task,
    'routing_key': "create.*",
    'cb': bright_account_create
})

logger.info("Disconnected")
rc_rmq.disconnect()