def hydrate(force=False, modules: list = None):
# go get everything
if JanisShed._has_been_hydrated and not force:
return
if not modules:
modules = []
modules.extend(JanisShed._get_datatype_entrypoints())
modules.extend(JanisShed._get_tool_entrypoints())
level = None
cl = Logger.CONSOLE_LEVEL
if JanisShed.should_trace:
level = cl if cl >= LogLevel.DEBUG else LogLevel.DEBUG
Logger.log(
f"Setting CONSOLE_LEVEL to {LogLevel.get_str(level) or 'None'} while traversing modules"
)
Logger.set_console_level(level)
seen_modules = set()
seen_classes = set()
for m in modules:
JanisShed.traverse_module(m,
seen_modules=seen_modules,
seen_classes=seen_classes)
Logger.set_console_level(cl)
Logger.log(
f"Restoring CONSOLE_LEVEL to {LogLevel.get_str(cl)} now that Janis shed has been hydrated"
)
JanisShed._has_been_hydrated = True
def test_mute_unmute(self):
Logger.set_console_level(LogLevel.DEBUG)
self.assertIsNotNone(Logger.CONSOLE_LEVEL)
Logger.mute()
self.assertEqual(Logger.CONSOLE_LEVEL, LogLevel.CRITICAL)
Logger.unmute()
self.assertEqual(Logger.CONSOLE_LEVEL, LogLevel.DEBUG)
def check_logger_args(args):
level = LogLevel.INFO
if args.verbose:
level = LogLevel.VERBOSE
if args.debug:
level = LogLevel.DEBUG
if args.logInfo:
level = LogLevel.INFO
if args.logWarn:
level = LogLevel.WARNING
if args.logCritical:
level = LogLevel.CRITICAL
if args.logNone:
level = None
if args.logLevel:
level = LogLevel.from_str(args.logLevel)
Logger.set_console_level(level)
def hydrate_from(modules: list):
level = None
cl = Logger.CONSOLE_LEVEL
if JanisShed.should_trace:
level = cl if cl >= LogLevel.DEBUG else LogLevel.DEBUG
Logger.log(
f"Setting CONSOLE_LEVEL to {LogLevel.get_str(level) or 'None'} while traversing modules"
)
Logger.set_console_level(level)
seen_modules = set()
seen_classes = set()
for m in modules:
JanisShed.traverse_module(m,
seen_modules=seen_modules,
seen_classes=seen_classes)
Logger.set_console_level(cl)
Logger.log(
f"Restoring CONSOLE_LEVEL to {LogLevel.get_str(cl)} now that Janis shed has been hydrated"
)
def test_set_console_level(self):
Logger.set_console_level(LogLevel.DEBUG)
