Fix: Add list of observations

1 files changed, 6 insertions, 2 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]