Upload the contribution of vstf as bottleneck network framework.

[bottlenecks.git] / vstf / vstf / common / log.py

diff --git a/vstf/vstf/common/log.py b/vstf/vstf/common/log.py
new file mode 100755 (executable)
index 0000000..b34a8a9
--- /dev/null
+++ b/vstf/vstf/common/log.py
@@ -0,0 +1,41 @@
+import logging
+
+
+def _init_log_to_file(log_file, level, _format):
+    file_handler = logging.FileHandler(log_file)
+    file_handler.setLevel(level)
+    file_handler.setFormatter(logging.Formatter(_format))
+    return file_handler
+
+
+def _init_log_to_console(level, _format):
+    console = logging.StreamHandler()
+    console.setLevel(level)
+    console.setFormatter(logging.Formatter(_format))
+    return console
+
+
+def _init_log(log_file, level=logging.INFO, clevel=logging.INFO):
+    _format = '%(asctime)s <%(levelname)s> [%(funcName)s.%(lineno)d]: %(message)s'
+    # _format = '%(asctime)s [%(levelname)s] %(message)s'
+    _verbose = '%(levelname)s %(asctime)s [%(filename)s:%(lineno)d] %(funcName)s ### %(message)s'
+    _simple = '<%(levelname)s> [%(filename)s:%(lineno)d] ### %(message)s'
+    file_handler = _init_log_to_file(log_file, level, _verbose)
+    console = _init_log_to_console(clevel, _simple)
+    return file_handler, console
+
+
+def setup_logging(level=logging.INFO, log_file="/var/log/esp_test.log", clevel=logging.WARNING):
+    log = logging.getLogger()
+    log.setLevel(level)
+    file_handler, console = _init_log(log_file, level, clevel)
+    log.addHandler(file_handler)
+    log.addHandler(console)
+
+
+if __name__ == "__main__":
+    setup_logging()
+    logger = logging.getLogger("common")
+    logger.info('this is a test.')
+    logger.warning('this is a test.')
+    logger.error('this is a test.')