From 6210465bd43c4ef4fc8f91eb1702f6c9435466fe Mon Sep 17 00:00:00 2001
From: Bo-Chun Louis Chen <louistw@uab.edu>
Date: Fri, 10 Apr 2020 16:30:38 -0500
Subject: [PATCH] Add get_logger function in rc_util
---
rc_util.py | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
diff --git a/rc_util.py b/rc_util.py
index 946c560..0e7c4c1 100644
--- a/rc_util.py
+++ b/rc_util.py
@@ -1,9 +1,11 @@
+import logging
import argparse
from rc_rmq import RCRMQ
import json
rc_rmq = RCRMQ({'exchange': 'RegUsr', 'exchange_type': 'topic'})
tasks = {'ohpc_account': None, 'ood_account': None, 'slurm_account': None}
+logger_fmt = '%(asctime)s [%(module)s] - %(message)s'
def add_account(username, email, full='', reason=''):
rc_rmq.publish_msg({
@@ -52,3 +54,19 @@ def get_args():
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()
+
+def get_logger(args=None):
+ if args is None:
+ args = get_args()
+
+ logger_lvl = logging.WARNING
+
+ if args.verbose:
+ logger_lvl = logging.DEBUG
+
+ if args.dry_run:
+ logger_lvl = logging.INFO
+
+ logging.basicConfig(format=logger_fmt, level=logger_lvl)
+ return logging.getLogger(__name__)
+
--
GitLab