implementing FieldFilter for lighthouse endpoint

Author: maceto

functions/lighthouse/libs/queries.py

@@ -1,27 +1,28 @@
 import os
-import json
 from google.cloud import firestore
+from google.cloud.firestore_v1.base_query import FieldFilter
 from .result import Result 
 from .utils import convert_to_array
 
 DB = firestore.Client(project=os.environ.get('PROJECT'), database=os.environ.get('DATABASE'))
+TABLE = 'lighthouse'
 
 def list_data(params):
 
   technology_array = convert_to_array(params['technology'])
   data = []
 
   for technology in technology_array:
-    query = DB.collection(u'lighthouse')
+    query = DB.collection(TABLE)
 
     if 'start' in params:
-      query = query.where('date', '>=', params['start'])
+      query = query.where(filter=FieldFilter('date', '>=', params['start']))
     if 'end' in params:
-      query = query.where('date', '<=', params['end'])
+      query = query.where(filter=FieldFilter('date', '<=', params['end']))
 
-    query = query.where('geo', '==', params['geo'])
-    query = query.where('rank', '==', params['rank'])
-    query = query.where('technology', '==', technology)
+    query = query.where(filter=FieldFilter('geo', '==', params['geo']))
+    query = query.where(filter=FieldFilter('rank', '==', params['rank']))
+    query = query.where(filter=FieldFilter('technology', '==', technology))
 
     documents = query.stream()