Skip to content
Snippets Groups Projects
Commit 1b5c98e9 authored by Bo-Chun Chen's avatar Bo-Chun Chen
Browse files

Utilize argparse library

parent 5e5cba3a
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,!33Update create account script,!38WIP: Feat cod rmq
#!/usr/bin/env python3 #!/usr/bin/env python3
import sys
import rc_util import rc_util
import argparse
if len(sys.argv) < 2: parser = argparse.ArgumentParser()
print("Usage: {} USERNAME [EMAIL] [FULL_NAME] [REASON]".format(sys.argv[0]), file=sys.stderr) parser.add_argument('username', help='username that will be created')
exit(1) parser.add_argument('email', nargs='?', default='', help="User's email")
parser.add_argument('full_name', nargs='?', default='', help="User's full name")
parser.add_argument('reason', nargs='?', default='', help='Reason of requesting')
parser.add_argument('--domain', default='localhost', help='domain of email')
args = parser.parse_args()
domain = 'uab.edu' if args.email == '':
user_name = sys.argv[1] args.email = args.username
email = sys.argv[2] if len(sys.argv) >= 3 else '' if '@' not in args.email:
full_name = sys.argv[3] if len(sys.argv) >= 4 else '' args.email = args.username + '@' + args.domain
reason = sys.argv[4] if len(sys.argv) >= 5 else ''
if email == '': rc_util.add_account(args.username, email=args.email, full=args.full_name, reason=args.reason)
if '@' in user_name: print(f'Account for {args.username} requested.')
email = user_name
else:
email = user_name + '@' + domain
rc_util.add_account(user_name, email=email, full=full_name, reason=reason)
print("Account requested for user: {}".format(user_name))
print("Waiting for confirmation...") print("Waiting for confirmation...")
rc_util.consume(user_name) rc_util.consume(args.username)
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