From 60ff70de04e66770e318df0dccd7a5ff4c3801b5 Mon Sep 17 00:00:00 2001
From: Gergely Polonkai <gergely@polonkai.eu>
Date: Sun, 15 Jan 2017 17:36:01 +0100
Subject: [PATCH] Initial version

It is working as a Python module, but it lacks everything else.
---
 flask_logging_extras/__init__.py | 63 ++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 flask_logging_extras/__init__.py

diff --git a/flask_logging_extras/__init__.py b/flask_logging_extras/__init__.py
new file mode 100644
index 0000000..51d7099
--- /dev/null
+++ b/flask_logging_extras/__init__.py
@@ -0,0 +1,63 @@
+"""
+Extra functionality for Flask logging
+
+Flask-Logging-Extras is a Flask extension that plugs into the logging
+mechanism of Flask applications.
+"""
+
+import logging
+
+__version_info__ = ('0', '0', '1')
+__version__ = '.'.join(__version_info__)
+__author__ = 'Gergely Polonkai'
+__license__ = 'BSD'
+__copyright__ = '(c) 2015 GT2'
+
+class FlaskExtraLogger(logging.getLoggerClass()):
+    """
+    A logger class that is capable of adding extra keywords to log formatters
+    """
+
+    def __init__(self, *args, **kwargs):
+        if 'app' in kwargs and kwargs['app'] is not None:
+            raise TypeError(
+                "Cannot initialise {classname} with an app.  "
+                "See the documentation of Flask-Logging-Extras for more info."
+                .format(classname=self.__class__.__name__))
+
+        super(FlaskExtraLogger, self).__init__(*args, **kwargs)
+
+    def _log(self, *args, **kwargs):
+        if 'extra' not in kwargs:
+            kwargs['extra'] = {}
+
+        for kw in self._valid_keywords:
+            if kw in kwargs:
+                kwargs['extra'][kw] = kwargs[kw]
+
+        super(FlaskExtraLogger, self)._log(*args, **kwargs)
+
+    def init_app(self, app):
+        self.app = app
+
+        self.app.config.setdefault('FLASK_LOGGING_EXTRAS_KEYWORDS', [])
+
+        for kw in self.app.config['FLASK_LOGGING_EXTRAS_KEYWORDS']:
+            if kw in ['exc_info', 'extra', 'stack_info']:
+                raise ValueError(
+                    '"{keyword}" member of FLASK_LOGGING_EXTRAS_KEYWORDS is '
+                    'reserved for internal use.')
+
+
+def register_logger_class(cls=FlaskExtraLogger):
+    """
+    Register a new logger class
+
+    It is effectively a wrapper around logging.setLoggerClass().
+    """
+
+    if not issubclass(cls, logging.Logger):
+        raise TypeError(
+            "The logger class must be a subclass of logging.Logger!")
+
+    logging.setLoggerClass(cls)