Skip to content
Snippets Groups Projects
ood_account_create.py 1.17 KiB
Newer Older
#!/usr/bin/env python
import sys
import json
Bo-Chun Chen's avatar
Bo-Chun Chen committed
import subprocess
Bo-Chun Chen's avatar
Bo-Chun Chen committed
from rc_rmq import RCRMQ
Bo-Chun Chen's avatar
Bo-Chun Chen committed
task = 'ood_account'
Bo-Chun Chen's avatar
Bo-Chun Chen committed
# Instantiate rabbitmq object
rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'})
def ood_account_create(ch, method, properties, body):
    msg = json.loads(body)
    print("Message received {}".format(msg))
    username = msg['username']
    user_uid = str(msg['uid'])
    user_gid = str(msg['gid'])
Bo-Chun Chen's avatar
Bo-Chun Chen committed
    success = False
    try:
        subprocess.call(["sudo", "groupadd", "-r", "-g", user_gid, username])
        subprocess.call(["sudo", "useradd", "-u", user_uid, "-g", user_gid, username])
Bo-Chun Chen's avatar
Bo-Chun Chen committed
        print("[{}]: User {} has been added".format(task, username))
        success = True
        e = sys.exc_info()[0]
        print("[{}]: Error: {}".format(task, e))
Bo-Chun Chen's avatar
Bo-Chun Chen committed
    ch.basic_ack(delivery_tag=method.delivery_tag)

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

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