Add get json serializable function

rapid7 · Oct 13, 2017 · b447fb9 · b447fb9
1 parent aee6cc6
commit b447fb9
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/whois/_3_adjust.py b/whois/_3_adjust.py
@@ -47,6 +47,17 @@ def __init__(self, data):
         except KeyError:
             pass
 
+    def get_json_serializable(self):
+        properties = self.__dict__
+
+        for key in properties:
+            if type(properties[key]) is datetime.datetime:
+                properties[key] = properties[key].isoformat()
+            if type(properties[key]) is set:
+                properties[key] = list(properties[key])
+
+        return properties
+
 
 # http://docs.python.org/library/datetime.html#strftime-strptime-behavior
 DATE_FORMATS = [