ohpc_account_create.py

#!/usr/bin/env python
import sys
import json
import subprocess
from pwd import getpwnam
from rc_rmq import RCRMQ

task = "ohpc_account"

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

def ohpc_account_create(ch, method, properties, body):
    msg = json.loads(body)
    print("Message received {}".format(msg))
    username = msg['username']
    success = False
    try:
        subprocess.call(["sudo", "useradd", username])
        print("[{}]: User {} has been added".format(task, username))
        success = True
    except:
        e = sys.exc_info()[0]
        print("[{}]: Error: {}".format(task, e))

    ch.basic_ack(delivery_tag=method.delivery_tag)
    msg['uid'] = getpwnam(username).pw_uid
    msg['gid'] = getpwnam(username).pw_gid

    # send confirm message
    rc_rmq.publish_msg({
        'routing_key': 'confirm.' + username,
        'msg': {
            'task': task,
            'success': success
        }
    })

    if success:
        # send create message to other agent
        rc_rmq.publish_msg({
            'routing_key': 'create.' + username,
            'msg': msg
        })

print("Start Listening to queue: {}".format(task))
rc_rmq.start_consume({
    'queue': task,
    'routing_key': 'request.*',
    'cb': ohpc_account_create
})