From 4c5e4ba753face81fd73aa3bae1fe3c012e48859 Mon Sep 17 00:00:00 2001 From: ahothan Date: Tue, 15 Aug 2017 15:52:12 -0700 Subject: NFVBENCH-4 Add support for log file Change-Id: Id630944944d472322a763fe41ccb0e1ed9fb3066 Signed-off-by: ahothan --- nfvbench/log.py | 24 +++++++++++++++--------- 1 file changed, 15 insertions(+), 9 deletions(-) (limited to 'nfvbench/log.py') diff --git a/nfvbench/log.py b/nfvbench/log.py index 22afefe..f308171 100644 --- a/nfvbench/log.py +++ b/nfvbench/log.py @@ -14,27 +14,33 @@ import logging +_product_name = 'nfvbench' -def setup(product_name): +def setup(): # logging.basicConfig() formatter_str = '%(asctime)s %(levelname)s %(message)s' handler = logging.StreamHandler() handler.setFormatter(logging.Formatter(formatter_str)) # Add handler to logger - logger = logging.getLogger(product_name) + logger = logging.getLogger(_product_name) logger.addHandler(handler) +def add_file_logger(logfile): + if logfile: + file_formatter_str = '%(asctime)s %(levelname)s %(message)s' + file_handler = logging.FileHandler(logfile, mode='w') + file_handler.setFormatter(logging.Formatter(file_formatter_str)) + logger = logging.getLogger(_product_name) + logger.addHandler(file_handler) -def set_level(product, debug=False): +def set_level(debug=False): log_level = logging.DEBUG if debug else logging.INFO - logger = logging.getLogger(product) + logger = logging.getLogger(_product_name) logger.setLevel(log_level) - -def getLogger(product): - logger = logging.getLogger(product) - +def getLogger(): + logger = logging.getLogger(_product_name) return logger -LOG = getLogger('nfvbench') +LOG = getLogger() -- cgit 1.2.3-korg