Make BetterJSONEncoder use OrderedDict for namedtuples.

Author: chebee7i

raven/utils/basic.py

@@ -1,5 +1,11 @@
 from __future__ import absolute_import
 
+try:
+    from collections.abc import Mapping
+except ImportError:
+    # Python < 3.3
+    from collections import Mapping
+
 from functools import update_wrapper
 import threading
 
@@ -30,11 +36,11 @@ def varmap(func, var, context=None, name=None):
         return func(name, '<...>')
     context[objid] = 1
 
-    if isinstance(var, (list, tuple)):
+    if isinstance(var, (list, tuple)) and not is_namedtuple(var):
         ret = [varmap(func, f, context, name) for f in var]
     else:
         ret = func(name, var)
-        if isinstance(ret, dict):
+        if isinstance(ret, Mapping):
             ret = dict((k, varmap(func, v, context, k))
                        for k, v in iteritems(var))
     del context[objid]

raven/utils/json.py

@@ -9,10 +9,14 @@
 from __future__ import absolute_import
 
 import codecs
+import collections
 import datetime
 import uuid
 import json
 
+from .basic import is_namedtuple
+
+
 try:
     JSONDecodeError = json.JSONDecodeError
 except AttributeError:
@@ -25,12 +29,17 @@ class BetterJSONEncoder(json.JSONEncoder):
         datetime.datetime: lambda o: o.strftime('%Y-%m-%dT%H:%M:%SZ'),
         set: list,
         frozenset: list,
-        bytes: lambda o: o.decode('utf-8', errors='replace')
+        bytes: lambda o: o.decode('utf-8', errors='replace'),
+        collections.namedtuple: lambda o: o._asdict(),
     }
 
     def default(self, obj):
+        obj_type = type(obj)
+        if obj_type not in self.ENCODER_BY_TYPE and is_namedtuple(obj):
+            obj_type = collections.namedtuple
+
         try:
-            encoder = self.ENCODER_BY_TYPE[type(obj)]
+            encoder = self.ENCODER_BY_TYPE[obj_type]
         except KeyError:
             try:
                 return super(BetterJSONEncoder, self).default(obj)