1 files changed, 120 insertions, 0 deletions
diff --git a/netmodel/util/log.py b/netmodel/util/log.py
new file mode 100644
index 00000000..68eb9a7f
--- /dev/null
+++ b/netmodel/util/log.py
@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2017 Cisco and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+import logging
+import logging.config
+import os
+import sys
+
+colors = {
+    'white':        "\033[1;37m",
+    'yellow':       "\033[1;33m",
+    'green':        "\033[1;32m",
+    'blue':         "\033[1;34m",
+    'cyan':         "\033[1;36m",
+    'red':          "\033[1;31m",
+    'magenta':      "\033[1;35m",
+    'black':        "\033[1;30m",
+    'darkwhite':    "\033[0;37m",
+    'darkyellow':   "\033[0;33m",
+    'darkgreen':    "\033[0;32m",
+    'darkblue':     "\033[0;34m",
+    'darkcyan':     "\033[0;36m",
+    'darkred':      "\033[0;31m",
+    'darkmagenta':  "\033[0;35m",
+    'darkblack':    "\033[0;30m",
+    'off':          "\033[0;0m"
+}
+
+def textcolor(color, string):
+    """
+    This function is useful to output information to the stdout by exploiting
+    different colors, depending on the result of the last command executed.
+
+    It is possible to chose one of the following colors: white, yellow, green,
+    blue, cyan, red, magenta, black, darkwhite, darkyellow, darkgreen,
+    darkblue, darkcyan, darkred, darkmagenta, darkblack, off
+
+    :param color: The color of the output string, chosen from the previous
+            list.
+    :param string: The string to color
+    :return: The colored string if the color is valid, the original string
+            otherwise.
+    """
+    try:
+        return colors[color] + string + colors['off']
+    except:
+        return string
+
+FMT = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+
+MAP_LEVELNAME_COLOR = {
+    'DEBUG'     : 'off',
+    'INFO'      : 'cyan',
+    'WARNING'   : 'yellow',
+    'ERROR'     : 'red',
+    'CRITICAL'  : 'red'
+}
+
+DEFAULT_COLOR = 'blue'
+
+DEPRECATED_REPEAT = False
+DEPRECATED_DONE = list()
+
+DEBUG_PATHS = 'vicn'
+
+class DebugPathsFilter:
+    def __init__(self, debug_paths):
+        self._debug_paths = debug_paths
+
+    def filter(self, record):
+        return record.levelname != 'DEBUG' or record.name in self._debug_paths
+
+class ColorFormatter(logging.Formatter):
+    def __init__(self, *args, debug_paths = None, **kwargs):
+        self._debug_paths = debug_paths
+        super().__init__(*args, **kwargs)
+
+    def format(self, record):
+        formatted = super().format(record)
+
+        try:
+            color = record.category
+        except AttributeError:
+            color = MAP_LEVELNAME_COLOR.get(record.levelname, DEFAULT_COLOR)
+
+        return textcolor(color, formatted)
+
+def initialize_logging():
+    # Use logger config
+    config_path = os.path.join(os.path.dirname(__file__),
+            os.path.pardir, os.path.pardir, 'config', 'logging.conf')
+    if os.path.exists(config_path):
+        logging.config.fileConfig(config_path, disable_existing_loggers=False)
+
+    root = logging.getLogger()
+    root.setLevel(logging.DEBUG)
+
+    # Stdout handler
+    ch = logging.StreamHandler(sys.stdout)
+    ch.setLevel(logging.INFO)
+    #formatter = logging.Formatter(FMT)
+    formatter = ColorFormatter(FMT, debug_paths = DEBUG_PATHS)
+    ch.setFormatter(formatter)
+    ch.addFilter(DebugPathsFilter(DEBUG_PATHS))
+    root.addHandler(ch)