Add support for exc_traceback

Author: EvaSDK

fluent/handler.py

@@ -2,6 +2,7 @@
 
 import logging
 import socket
+import traceback
 
 try:
     import simplejson as json
@@ -16,6 +17,10 @@ class FluentRecordFormatter(logging.Formatter, object):
 
     Best used with server storing data in an ElasticSearch cluster for example.
 
+    Supports an extra `exc_traceback` format argument that represents a
+    traceback when logging an exception as return by
+    :meth:`traceback.extract_tb`.
+
     :param fmt: a dict with format string as values to map to provided keys.
     """
     def __init__(self, fmt=None, datefmt=None):
@@ -37,9 +42,17 @@ def format(self, record):
         super(FluentRecordFormatter, self).format(record)
         # Add ours
         record.hostname = self.hostname
+
         # Apply format
-        data = dict([(key, value % record.__dict__)
-                     for key, value in self._fmt_dict.items()])
+        data = {}
+        for key, value in self._fmt_dict.items():
+            if value.find('%(exc_traceback)') >= 0:
+                if record.exc_info:
+                    data[key] = traceback.extract_tb(record.exc_info[2])
+                else:
+                    data[key] = None
+            else:
+                data[key] = value % record.__dict__
 
         self._structuring(data, record.msg)
         return data