new endpoint for lighthouse report added

Author: maceto

functions/lighthouse/libs/__init__.py

@@ -0,0 +1,31 @@
+import os
+import json
+from google.cloud import firestore
+from .result import Result 
+
+DB = firestore.Client(project=os.environ.get('PROJECT'))
+
+def list_data(params):
+  ref = DB.collection(u'lighthouse')
+
+  query = ref
+  print("params", params)
+  if 'start' in params:
+    query = query.where('date', '>=', params['start'])
+  if 'end' in params:
+    query = query.where('date', '<=', params['end'])
+  if 'geo' in params:
+    query = query.where('geo', '==', params['geo'])
+  if 'technology' in params:
+    params_array = json.loads(params['app'])
+    query = query.where('technology', 'in', params_array)
+  if 'rank' in params:
+    query = query.where('rank', '==', params['rank'])
+
+  documents = query.stream()
+
+  data = []
+  for doc in documents:
+      data.append(doc.to_dict())
+
+  return Result(result=data)

functions/lighthouse/libs/queries.py

@@ -0,0 +1,20 @@
+
+class Result():
+  def __init__(self, status=None, result=None, errors=[]):
+    self._status = status
+    self.result = result
+    self.errors = errors
+
+  def success(self) -> bool:
+    return not self.failure()
+
+  def failure(self) -> bool:
+    return len(self.errors) > 0
+  
+  @property
+  def status(self):
+    if self._status != None:
+      return self._status
+  
+    return "ok" if self.success else "error"
+    

functions/lighthouse/libs/result.py

@@ -0,0 +1,13 @@
+import json
+
+def output(result):
+  status = 200 if result.success() else 400
+  payload = result.result if result.success() else convert_to_hashes(result.errors)
+  return (json.dumps(payload), status)
+
+def convert_to_hashes(arr):
+    hashes_arr = []
+    for inner_arr in arr:
+        hash_dict = {inner_arr[0]: inner_arr[1]}
+        hashes_arr.append(hash_dict)
+    return hashes_arr

functions/lighthouse/libs/utils.py

@@ -0,0 +1,21 @@
+from .result import Result
+
+class Validator():
+  def __init__(self, params):
+    self.params = params
+    self.errors = []
+    self.normalizer_params = self.normalize(params)
+
+  def validate(self):
+    result = Result(status="ok", result="()")
+
+    # if 'geo' not in self.params:
+    #   self.add_error("geo", "missing geo parameters")
+
+    return Result(errors=self.errors, result=self.params)
+  
+  def add_error(self, key, error):
+    self.errors.append([key, error])
+
+  def normalize(self, params):
+    return ""

functions/lighthouse/libs/validator.py

@@ -0,0 +1,19 @@
+import functions_framework
+from .libs.validator import Validator
+from .libs.utils import output
+from .libs.queries import list_data
+
+@functions_framework.http
+def dispatcher(request):
+  args = request.args.to_dict()
+
+  validator = Validator(params=args)
+  result = validator.validate()
+
+  if result.failure():
+    print("error", result.errors)
+    return output(result)
+  
+  response = list_data(result.result)
+
+  return output(response)

functions/lighthouse/main.py

@@ -0,0 +1,3 @@
+functions-framework
+google-cloud-firestore
+pytest