Added count_documents() implementation to builtin_timeseries

emission/storage/timeseries/builtin_timeseries.py

@@ -440,3 +440,15 @@ def update_data(user_id, key, obj_id, data):
         logging.debug("updating entry %s into timeseries" % new_entry)
         edb.save(ts.get_timeseries_db(key), new_entry)
 
+    def count_data(self, key, extra_query_list):
+        """
+        Returns the total number of documents for the specific key referring to a timeseries db.
+        Additional keys can be passed as an optional list for filtering data.
+        """
+        logging.debug("count_data timeseries called")
+        created_query = self._get_query(key_list=[key], extra_query_list=extra_query_list)
+        result_dataset = self.get_timeseries_db(key)
+        total_entries = result_dataset.count_documents(created_query)
+        return total_entries
+
+

emission/tests/storageTests/TestTimeSeries.py

@@ -81,6 +81,22 @@ def testExtraQueries(self):
         with self.assertRaises(AttributeError):
             list(ts.find_entries(time_query=tq, extra_query_list=[ignored_phones]))
 
+    def testCountData(self):
+        '''
+        Test 1 : Specific key with empty extra_queries
+        key = 'background/location', extra_query_list = []
+        Results in empty query = {}, which matches all documents for a user for that key.
+        Hence should return total count of all documents matching that key.
+        Testing this with sample dataset: "shankari_2015-aug-27"
+        '''
+        ts = esta.TimeSeries.get_time_series(self.testUUID)
+        total_count = ts.count_data("background/location",[])
+        print(total_count)
+        self.assertEqual(total_count, 555)
+        print("Assert Test for Count Data successful!")
+
+
+
 if __name__ == '__main__':
     import emission.tests.common as etc
     etc.configLogging()