Wrap the Database.info result to a TrombiDict

inoi · May 4, 2011 · 6f1501e · 6f1501e
1 parent a736187
commit 6f1501e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/test/test_client.py b/test/test_client.py
@@ -260,6 +260,7 @@ def get_info(db):
         db.info(check_info)
 
     def check_info(info):
+        eq(info.error, False)
         eq(info['db_name'], 'testdb')
         eq(info['doc_count'], 0)
         assert 'update_seq' in info

diff --git a/trombi/client.py b/trombi/client.py
@@ -99,6 +99,13 @@ def __init__(self, data):
         super(TrombiResult, self).__init__()
 
 
+class TrombiDict(dict):
+    error = False
+
+    def to_basetype(self):
+        return dict(self)
+
+
 def _jsonize_params(params):
     result = dict()
     for key, value in params.iteritems():
@@ -257,7 +264,7 @@ def _fetch(self, url, *args, **kwargs):
     def info(self, callback):
         def _really_callback(response):
             if response.code == 200:
-                callback(json.loads(response.body))
+                callback(TrombiDict(json.loads(response.body)))
             else:
                 callback(_error_response(response))