Skip to content
Snippets Groups Projects

Feat rc_util add function

Closed Bo-Chun Chen requested to merge github/fork/diedpigs/feat-util-add-function into develop
1 file
+ 18
0
Compare changes
  • Side-by-side
  • Inline
+ 18
0
import logging
import argparse
from rc_rmq import RCRMQ
import json
@@ -5,6 +6,7 @@ import json
rc_rmq = RCRMQ({'exchange': 'Register'})
confirm_rmq = RCRMQ({'exchange': 'Confirm'})
tasks = {'ohpc_account': False, 'ohpc_homedir': False, 'ood_account': False, 'slurm_account': False}
logger_fmt = '%(asctime)s [%(module)s] - %(message)s'
def add_account(username, full='', reason=''):
rc_rmq.publish_msg({
@@ -49,3 +51,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__)
Loading