Added RoundupJSONEncoder

.. to handle classes from roundup

committer: Ralf Schlatterbeck <[email protected]>

Author: kinggreedy

roundup/rest.py

@@ -50,7 +50,7 @@ def get_collection(self, class_name, input):
             raise Unauthorised('Permission to view %s denied' % class_name)
         class_obj = self.db.getclass(class_name)
         prop_name = class_obj.labelprop()
-        result = [{'id': item_id, 'name': class_obj.get(item_id, prop_name)}
+        result = [{'id': item_id, prop_name: class_obj.get(item_id, prop_name)}
                   for item_id in class_obj.list()
                   if self.db.security.hasPermission('View', self.db.getuid(),
                                                     class_name,
@@ -71,6 +71,7 @@ def get_element(self, class_name, item_id, input):
                                                     class_name, prop_name,
                                                     item_id)]
         result = dict(result)
+        result['id'] = item_id
 
         return result
 
@@ -187,7 +188,15 @@ def dispatch(self, method, uri, input):
         except AttributeError:
             raise NotImplementedError('Method is invalid')
         finally:
-            output = json.JSONEncoder().encode(output)
+            output = RoundupJSONEncoder().encode(output)
 
         print "Length: %s - Content(50 char): %s" % (len(output), output[:50])
         return output
+
+class RoundupJSONEncoder(json.JSONEncoder):
+    def default(self, obj):
+        try:
+            result = json.JSONEncoder.default(self, obj)
+        except TypeError:
+            result = str(obj)
+        return result