Merge pull request #375 from stripe/ob-repr-datetime

Use custom JSON encoder to handle datetime objects

stripe/stripe_object.py

@@ -1,5 +1,6 @@
 from __future__ import absolute_import, division, print_function
 
+import datetime
 import warnings
 from copy import deepcopy
 
@@ -33,6 +34,12 @@ def _serialize_list(array, previous):
 
 
 class StripeObject(dict):
+    class ReprJSONEncoder(util.json.JSONEncoder):
+        def default(self, obj):
+            if isinstance(obj, datetime.datetime):
+                return api_requestor._encode_datetime(obj)
+            return super(StripeObject.ReprJSONEncoder, self).default(obj)
+
     def __init__(self, id=None, api_key=None, stripe_version=None,
                  stripe_account=None, **params):
         super(StripeObject, self).__init__()
@@ -212,7 +219,8 @@ def __repr__(self):
             return unicode_repr
 
     def __str__(self):
-        return util.json.dumps(self, sort_keys=True, indent=2)
+        return util.json.dumps(self, sort_keys=True, indent=2,
+                               cls=self.ReprJSONEncoder)
 
     def to_dict(self):
         warnings.warn(

tests/test_stripe_object.py

@@ -1,5 +1,6 @@
 from __future__ import absolute_import, division, print_function
 
+import datetime
 import pickle
 from copy import copy, deepcopy
 
@@ -178,6 +179,7 @@ def test_repr(self):
             'foo', 'bar', myparam=5)
 
         obj['object'] = u'\u4e00boo\u1f00'
+        obj.date = datetime.datetime.fromtimestamp(1511136000)
 
         res = repr(obj)
 
@@ -186,6 +188,7 @@ def test_repr(self):
 
         self.assertTrue(u'<StripeObject \u4e00boo\u1f00' in res)
         self.assertTrue(u'id=foo' in res)
+        self.assertTrue(u'"date": 1511136000' in res)
 
     def test_pickling(self):
         obj = stripe.stripe_object.StripeObject(