mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-08-23 22:26:23 +00:00
danielsagi
e6a3c12098
* removed arp and dns hunters usage due to it's violations of the scapy GPL2 license * added installation of arp and dns hunters to Dockerfile * added explicit new version to plugins in dockerfile installation * ignore B020 flake8
26 lines
1003 B
Python
26 lines
1003 B
Python
import logging
|
|
|
|
DEFAULT_LEVEL = logging.INFO
|
|
DEFAULT_LEVEL_NAME = logging.getLevelName(DEFAULT_LEVEL)
|
|
LOG_FORMAT = "%(asctime)s %(levelname)s %(name)s %(message)s"
|
|
|
|
|
|
def setup_logger(level_name, logfile):
|
|
# Remove any existing handlers
|
|
# Unnecessary in Python 3.8 since `logging.basicConfig` has `force` parameter
|
|
for h in logging.getLogger().handlers[:]:
|
|
h.close()
|
|
logging.getLogger().removeHandler(h)
|
|
|
|
if level_name.upper() == "NONE":
|
|
logging.disable(logging.CRITICAL)
|
|
else:
|
|
log_level = getattr(logging, level_name.upper(), None)
|
|
log_level = log_level if isinstance(log_level, int) else None
|
|
if logfile is None:
|
|
logging.basicConfig(level=log_level or DEFAULT_LEVEL, format=LOG_FORMAT)
|
|
else:
|
|
logging.basicConfig(filename=logfile, level=log_level or DEFAULT_LEVEL, format=LOG_FORMAT)
|
|
if not log_level:
|
|
logging.warning(f"Unknown log level '{level_name}', using {DEFAULT_LEVEL_NAME}")
|