From a74d182c473e0e3a8b665df35a81d7c283a59f12 Mon Sep 17 00:00:00 2001 From: Bo-Chun Louis Chen <louistw@uab.edu> Date: Fri, 10 Apr 2020 15:40:54 -0500 Subject: [PATCH] Add get_args function in rc_util --- rc_util.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/rc_util.py b/rc_util.py index 2b39ae5..946c560 100644 --- a/rc_util.py +++ b/rc_util.py @@ -1,3 +1,4 @@ +import argparse from rc_rmq import RCRMQ import json @@ -44,3 +45,10 @@ def consume(username, callback=worker, debug=False): rc_rmq.disconnect() return { 'success' : True } + +def get_args(): + # Parse arguments + parser = argparse.ArgumentParser() + 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') + return parser.parse_args() -- GitLab