Skip to content
Snippets Groups Projects
Unverified Commit 746aa7f5 authored by Ravi Tripathi's avatar Ravi Tripathi Committed by GitHub
Browse files

Merge pull request #28 from eesaanatluri/feat-resolve-uid-gid

Feat resolve uid gid
parents 137f4d23 8298392c
No related branches found
No related tags found
5 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
#!/usr/bin/env python
import sys
import json
import ldap
import logging
import argparse
import rc_util
from os import popen
from rc_rmq import RCRMQ
task = 'create_account'
# Instantiate rabbitmq object
rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'})
args = rc_util.get_args()
# Logger
logger = rc_util.get_logger()
#Account creation
def create_account(msg):
logger.info(f'Account creation request received: {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}')
except Exception:
logger.exception("Fatal cmsh error:")
# Define your callback function
def resolve_uid_gid(ch, method, properties, body):
# Retrieve message
msg = json.loads(body)
logger.info("Received {}".format(msg))
username = msg['username']
success = False
# Determine next available UID
try:
user_exists_cmd = "/usr/bin/getent passwd {username}"
user_exists = popen(user_exists_cmd).read().rstrip()
if user_exists:
logger.info("The user, {} already exists".format(username))
msg['uid'] = user_exists.split(':')[2]
msg['gid'] = user_exists.split(':')[3]
else:
cmd_uid = "/usr/bin/getent passwd | \
awk -F: '($3>10000) && ($3<20000) && ($3>maxuid) { maxuid=$3; } END { print maxuid+1; }'"
msg['uid'] = popen(cmd_uid).read().rstrip()
logger.info(f"UID query: {cmd_uid}")
cmd_gid = "/usr/bin/getent group | \
awk -F: '($3>10000) && ($3<20000) && ($3>maxgid) { maxgid=$3; } END { print maxgid+1; }'"
msg['gid'] = popen(cmd_gid).read().rstrip()
logger.info(f"GID query: {cmd_gid}")
create_account(msg)
msg['task'] = task
msg['success'] = True
except Exception:
logger.exception("Fatal UID resolution 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,
'routing_key': "request.*",
'cb': resolve_uid_gid
})
logger.info("Disconnected")
rc_rmq.disconnect()
pika==1.1.0
pyldap==3.0.0.post1
dataset==1.3.1
Jinja2==2.11.2
sh==1.12.14
\ No newline at end of file
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