Return the unique key in the response when adding new data.

1 files changed, 9 insertions, 10 deletions

diff --git a/src/db.py b/src/db.py
index 92684c8..1f35cac 100755
--- a/src/db.py
+++ b/src/db.py
@@ -46,18 +46,17 @@ class DictDB():
         return self._ts
 
     def add(self, data, batch_write=False):
-        keys = []
-
+        key = str(self.unique_key())
         if type(data) is list:
             for item in data:
-                item['_id'] = str(self.unique_key())
+                item['_id'] = key
 
                 self.couchdb.save(item)
         else:
-            data['_id'] = str(self.unique_key())
+            data['_id']=key
             self.couchdb.save(data)
 
-        return keys
+        return key
 
     def get(self, key):
         return self.couchdb[key]
@@ -66,19 +65,19 @@ class DictDB():
         pass
 
     def search(self, **kwargs):
-        data = list()
-        selector = dict()
+        data=list()
+        selector=dict()
 
         if kwargs:
-            selector = {
+            selector={
                 "selector": {
                 }
             }
 
             for key in kwargs:
                 if kwargs[key].isnumeric():
-                    kwargs[key] = int(kwargs[key])
-                selector['selector'][key] = {'$eq': kwargs[key]}
+                    kwargs[key]=int(kwargs[key])
+                selector['selector'][key]={'$eq': kwargs[key]}
 
         for doc in self.couchdb.find(selector):
             data.append(doc)