import sys import logging def get_string_arg(args, required): """ Take an array of element, take first element and return it stripped. If array is empty or None, print an error message with required and exit in error. :param args: an array of arguments. Can be empty or None :param required: the message to print in no args (or None) :return: The arg stripped """ if not args: print(f'Missing {required}') sys.exit(1) logging.debug(args) return args[0].strip() # TODO Move this function def arg_string(arg, required = ''): # TODO To meld with get_string_args if required and arg.startswith('-'): print('required ', required) sys.exit('1') else: return arg.strip() def arg_number(arg, required = ''): if not arg.isdigit(): print('Invalid value') # TODO sys.exit(1) return int(arg)