Fix: Add list of observations

2 files changed, 8 insertions, 4 deletions

diff --git a/src/db.py b/src/db.py
index 0c7e998..fa201aa 100755
--- a/src/db.py
+++ b/src/db.py
@@ -37,15 +37,19 @@ class DictDB():
         pass
 
     def add(self, data, batch_write=False):
-        key = str(self.unique_key())
+        keys = []
 
         if type(data) is list:
             for item in data:
+                key = str(self.unique_key())
                 self.couchdb[key] = item
+                keys.append(key)
         else:
+            key = str(self.unique_key())
             self.couchdb[key] = data
+            keys.append(key)
 
-        return key
+        return keys
 
     def get(self, key):
         return self.couchdb[key]
diff --git a/src/wsgi.py b/src/wsgi.py
index ef1e36f..e99ac74 100755
--- a/src/wsgi.py
+++ b/src/wsgi.py
@@ -99,8 +99,8 @@ class EPAdd(CollectorResource):
             resp.text = CollectorResource.parse_error(decodedin)
             return
 
-        key = self._db.add(json_data)
-        resp.text = repr(key) + '\n'
+        keys = self._db.add(json_data)
+        resp.text = repr(keys) + '\n'
 
 
 def init(url_res_map, addr='', port=8000):