From 2582e2323abaa8c1a764f00ace7b1011cb2704a4 Mon Sep 17 00:00:00 2001
From: Paul Logston <code@logston.me>
Date: Thu, 18 Jun 2015 22:38:33 -0400
Subject: [PATCH] Add exception traceback handling to Formatter

---
 fluent/handler.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/fluent/handler.py b/fluent/handler.py
index 308843d..3c9160c 100644
--- a/fluent/handler.py
+++ b/fluent/handler.py
@@ -3,6 +3,7 @@
 import logging
 import socket
 import sys
+import traceback
 
 try:
     import simplejson as json
@@ -51,6 +52,9 @@ def format(self, record):
         data = dict([(key, value % record.__dict__)
                      for key, value in self._fmt_dict.items()])
 
+        if record.exc_info:
+            self._add_exception_to_record(record)
+
         self._structuring(data, record.msg)
         return data
 
@@ -83,6 +87,24 @@ def _add_dic(data, dic):
             if isinstance(key, basestring):
                 data[str(key)] = value
 
+    def _add_exception_to_record(self, record):
+        """
+        Adds a traceback to record.msg
+        """
+        tb = traceback.format_exception(*record.exc_info)
+        tb = ''.join(tb)
+
+        if isinstance(record.msg, dict):
+            record.msg.update({"traceback": tb})
+        elif isinstance(record.msg, basestring):
+            record.msg = {"message": record.msg,
+                          "traceback": tb}
+        else:
+            record.msg = {
+                "message": record.getMessage(),
+                "traceback": tb
+            }
+
 
 class FluentHandler(logging.Handler):
     '''