def replace_log_level(query):
log_level_regex = 'level\:\s*[a-zA-Z]+\s*'
match = re.search(log_level_regex, query)
if match:
log_level_name = match.group(0).split(':')[1].strip().upper()
log_level_code = LogLevel.find_by_level_name(log_level_name)
if not log_level_code:
message = "The given log level({}) is invalid. Use one of the following: {}"
cli_error(
message.format(log_level_name, LogLevel.list_levels()))
exit()
return re.sub(log_level_regex, 'level:%s ' % log_level_code, query)
else:
return query
def test_list_levels(self):
self.assertEquals(
['CRITICAL', 'ERROR', 'WARNING', 'NOTICE', 'INFO', 'DEBUG'],
LogLevel.list_levels())