1 files changed, 170 insertions, 0 deletions
diff --git a/osdf/logging/oof_mdc_context.py b/osdf/logging/oof_mdc_context.py
new file mode 100644
index 0000000..9c9b52c
--- /dev/null
+++ b/osdf/logging/oof_mdc_context.py
@@ -0,0 +1,170 @@
+# -------------------------------------------------------------------------
+#   Copyright (c) 2020 AT&T Intellectual Property
+#
+#   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 re
+import sys
+
+from onaplogging.marker import Marker
+from onaplogging.marker import MARKER_TAG
+from onaplogging.mdcContext import _replace_func_name
+from onaplogging.mdcContext import fetchkeys
+from onaplogging.mdcContext import findCaller as fc
+from onaplogging.mdcContext import MDC
+
+from osdf.utils.mdc_utils import set_error_details
+
+
+def findCaller(self, stack_info=False, stacklevel=1):
+    """replacing onaplogging.mdcContext with this method to work with py3.8
+
+    """
+    return fc(stack_info)
+
+
+def mdc_mapper():
+    """Convert the MDC dict into comma separated, name=value string
+
+    :return: string format
+    """
+    return ','.join(f'{k}={v}' for (k, v) in MDC.result().items() if k not in ['customField2'])
+
+
+@fetchkeys
+def info(self, msg, *args, **kwargs):
+    """Wrapper method for log.info is called
+
+    """
+    if self.isEnabledFor(logging.INFO):
+        MDC.put('customField2', mdc_mapper())
+        self._log(logging.INFO, no_sep(msg), args, **kwargs)
+
+
+@fetchkeys
+def debug(self, msg, *args, **kwargs):
+    """Wrapper method for log.debug is called
+
+    msg: log message
+    args: logging args
+    kwargs: all the optional args
+    """
+    if self.isEnabledFor(logging.DEBUG):
+        self._log(logging.DEBUG, no_sep(msg), args, **kwargs)
+
+
+@fetchkeys
+def warning(self, msg, *args, **kwargs):
+    """Wrapper method for log.warning is called
+
+    msg: log message
+    args: logging args
+    kwargs: all the optional args
+    """
+    if self.isEnabledFor(logging.WARNING):
+        self._log(logging.WARNING, no_sep(msg), args, **kwargs)
+
+
+@fetchkeys
+def exception(self, msg, *args, **kwargs):
+    """Wrapper method for log.exception is called
+
+    msg: log message
+    args: logging args
+    kwargs: all the optional args
+    """
+    kwargs['exc_info'] = 1
+    self.error(no_sep(msg), *args, **kwargs)
+
+
+@fetchkeys
+def critical(self, msg, *args, **kwargs):
+    """Wrapper method for log.critical
+
+    msg: log message
+    args: logging args
+    kwargs: all the optional args
+    """
+    if self.isEnabledFor(logging.CRITICAL):
+        self._log(logging.CRITICAL, no_sep(msg), args, **kwargs)
+
+
+@fetchkeys
+def error(self, msg, *args, **kwargs):
+    """Wrapper method for log.error is called
+
+    msg: log message
+    args: logging args
+    kwargs: all the optional args
+    """
+    if self.isEnabledFor(logging.ERROR):
+        if not MDC.get('errorCode'):
+            set_error_details(400, 'Internal Error')
+        MDC.put('customField2', mdc_mapper())
+        self._log(logging.ERROR, no_sep(msg), args, **kwargs)
+
+
+@fetchkeys
+def log(self, level, msg, *args, **kwargs):
+    """Wrapper method for log.log is called
+
+    msg: log message
+    args: logging args
+    kwargs: all the optional args
+    """
+    if not isinstance(level, int):
+        if logging.raiseExceptions:
+            raise TypeError("level must be an integer")
+        else:
+            return
+
+    if self.isEnabledFor(level):
+        self._log(level, no_sep(msg), args, **kwargs)
+
+
+def handle(self, record):
+    """Wrapper method for log.handle is called
+
+    """
+    c_marker = getattr(self, MARKER_TAG, None)
+
+    if isinstance(c_marker, Marker):
+        setattr(record, MARKER_TAG, c_marker)
+
+    if (not self.disabled) and self.filter(record):
+        self.callHandlers(record)
+
+
+def no_sep(message):
+    """This method will remove newline, | from the message
+
+    """
+    if message is None:
+        return ''
+    return re.sub(r'[\|\n]', ' ', str(message))
+
+
+def patch_logging_mdc():
+    """The patch to add MDC ability in logging Record instance at runtime
+
+    """
+    local_module = sys.modules[__name__]
+    for attr in dir(logging.Logger):
+        if attr in _replace_func_name:
+            new_func = getattr(local_module, attr, None)
+            if new_func:
+                setattr(logging.Logger, attr, new_func)