Merge pull request #61 from diedpigs/feat-error-handle-cli

Feat error handle cli

create_account.py

@@ -3,6 +3,7 @@ import json
 import sys
 import rc_util
 import argparse
+import signal
 
 parser = argparse.ArgumentParser()
 parser.add_argument('username', help='username that will be created')
@@ -14,25 +15,40 @@ 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()
 
-logger = rc_util.get_logger(args)
+timeout = 60
 
 if args.email == '':
     args.email = args.username
     if '@' not in args.email:
         args.email = args.username + '@' + args.domain
 
+def timeout_handler(signum, frame):
+    print("Process timeout, there's might some issue with agents")
+    rc_util.rc_rmq.stop_consume()
+
+
 def callback(channel, method, properties, body):
     msg = json.loads(body)
     username = msg['username']
 
-    logger.info(f'Account for {username} has been created.')
+    if msg['success']:
+        print(f'Account for {username} has been created.')
+    else:
+        print(f"There's some issue while creating account for {username}")
+        errmsg = msg.get('errmsg', [])
+        for err in errmsg:
+            print(err)
 
     rc_util.rc_rmq.stop_consume()
     rc_util.rc_rmq.delete_queue()
 
 
 rc_util.add_account(args.username, email=args.email, full=args.full_name, reason=args.reason)
-logger.info(f'Account for {args.username} requested.')
+print(f'Account for {args.username} requested.')
+
+# Set initial timeout timer
+signal.signal(signal.SIGALRM, timeout_handler)
+signal.setitimer(signal.ITIMER_REAL, timeout)
 
-logger.info('Waiting for completion...')
+print('Waiting for completion...')
 rc_util.consume(args.username, routing_key=f'complete.{args.username}', callback=callback)