misc update

Author: chenqi

PageSpeedInsights/job.py

@@ -7,11 +7,14 @@
 PROJECT_ID = os.getenv("GCP_PROJECT_ID")
 TOPIC_NAME = "psi-single"
 
+URL_DICT = ["https://m.ctrip.com/webapp/flight/schedule/detail.html"]
+
 def run(event, context):
     publisher = pubsub_v1.PublisherClient()
     topic = publisher.topic_path(PROJECT_ID, TOPIC_NAME)
-    data = b'https://m.ctrip.com/webapp/flight/schedule/detail.html'
-    publisher.publish(topic, data)
+    for url in URL_DICT:
+        data = url.encode('utf-8')
+        publisher.publish(topic, data)
     return 'OK'
 
 def test_job():

PageSpeedInsights/psi.py

@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 '''PageSpeed Insights Single + Google Cloud Functions'''
 import os
+import base64
 import requests
 
 # Access Token, generated from GCP Console Credentials page.
@@ -27,22 +28,19 @@ def run(url):
         print(response.json())
     except requests.RequestException as _e:
         print(_e)
-    return ('OK', 200)
+    return 'OK'
 
 
 def run_pubsub(event, context):
-    import base64
     pubsub_message = base64.urlsafe_b64decode(event['data']).decode('utf-8')
-    run(pubsub_message)
-    return 'OK'
+    return run(pubsub_message)
 
 
 def test_run_http(test_url):
     run(test_url)
 
 
 def test_run_pubsub(test_url):
-    import base64
     event = {"data": base64.urlsafe_b64encode(test_url.encode('utf-8'))}
     context = None
     run_pubsub(event, context)

PageSpeedInsights/requirements.txt

@@ -1,3 +1,3 @@
-# Function dependencies, for example:
+# Function dependencies
 requests==2.21.0
 google-cloud-pubsub==0.40.0