Merge pull request #513 from mangalap123/develop

#454 Subscribe to updates field

Author: jlubken

setup.py

@@ -33,6 +33,8 @@
         "pyyaml",
         "selenium",
         "streamlit",
+        "gspread",
+        "oauth2client"
     ],
     classifiers=[
         "Programming Language :: Python :: 3",

src/penn_chime/presentation.py

@@ -5,8 +5,10 @@
 
 import altair as alt
 import numpy as np
+import os
+import json
 import pandas as pd
-
+import penn_chime.spreadsheet as sp
 from .constants import (
     CHANGE_DATE,
     DOCS_URL,
@@ -172,6 +174,8 @@ def display_sidebar(st, d: Parameters) -> Parameters:
     # it's kindof like ember or angular if you are familiar with those
 
     st_obj = st.sidebar
+    # used_widget_key = st.get_last_used_widget_key ( )
+
     current_hospitalized_input = NumberInput(
         st_obj,
         "Currently hospitalized COVID-19 patients",
@@ -349,6 +353,8 @@ def display_sidebar(st, d: Parameters) -> Parameters:
         max_y_axis = max_y_axis_input()
 
     current_date = current_date_input()
+    #Subscribe implementation
+    subscribe(st_obj)
 
     return Parameters(
         current_hospitalized=current_hospitalized,
@@ -374,6 +380,66 @@ def display_sidebar(st, d: Parameters) -> Parameters:
             days=ventilated_days),
     )
 
+#Read the environment variables and cteate json key object to use with ServiceAccountCredentials
+def readGoogleApiSecrets():
+    client_secret = {}
+    os.getenv
+    type = os.getenv ('GAPI_CRED_TYPE').strip()
+    print (type)
+    client_secret['type'] = type,
+    client_secret['project_id'] = os.getenv ('GAPI_CRED_PROJECT_ID'),
+    client_secret['private_key_id'] = os.getenv ('GAPI_CRED_PRIVATE_KEY_ID'),
+    client_secret['private_key'] = os.getenv ('GAPI_CRED_PRIVATE_KEY'),
+    client_secret['client_email'] = os.getenv ('GAPI_CRED_CLIENT_EMAIL'),
+    client_secret['client_id'] = os.getenv ('GAPI_CRED_CLIENT_ID'),
+    client_secret['auth_uri'] = os.getenv ('GAPI_CRED_AUTH_URI'),
+    client_secret['token_uri'] = os.getenv ('GAPI_CRED_TOKEN_URI'),
+    client_secret['auth_provider_x509_cert_url'] =  os.getenv ('GAPI_CRED_AUTH_PROVIDER_X509_CERT_URL'),
+    client_secret['client_x509_cert_url'] = os.getenv ('GAPI_CRED_CLIENT_X509_CERT_URI'),
+    json_data = json.dumps (client_secret)
+    print(json_data)
+    return json_data
+
+def readGoogleApiSecretsDict():
+    type = os.getenv ('GAPI_CRED_TYPE')
+    project_id = os.getenv ('GAPI_CRED_PROJECT_ID')
+    private_key_id =  os.getenv ('GAPI_CRED_PRIVATE_KEY_ID')
+    private_key = os.getenv ('GAPI_CRED_PRIVATE_KEY')
+    client_email = os.getenv ('GAPI_CRED_CLIENT_EMAIL')
+    client_id = os.getenv ('GAPI_CRED_CLIENT_ID')
+    auth_uri = os.getenv ('GAPI_CRED_AUTH_URI')
+    token_uri = os.getenv ('GAPI_CRED_TOKEN_URI')
+    auth_provider_x509_cert_url = os.getenv ('GAPI_CRED_AUTH_PROVIDER_X509_CERT_URL')
+    client_x509_cert_url = os.getenv ('GAPI_CRED_CLIENT_X509_CERT_URI')
+
+    secret = {
+        'type' : type,
+        'project_id' : project_id,
+        'private_key_id' : private_key_id,
+        'private_key':private_key,
+        'client_email': client_email,
+        'client_id': client_id,
+        'auth_uri': auth_uri,
+        'token_uri': token_uri,
+        'auth_provider_x509_cert_url':auth_provider_x509_cert_url,
+        'client_x509_cert_url':client_x509_cert_url
+    }
+    return secret
+
+def subscribe(st_obj):
+    st_obj.subheader ("Subscribe")
+    email = st_obj.text_input (label="Enter Email", value="", key="na_lower_1")
+    name = st_obj.text_input (label="Enter Name", value="", key="na_upper_1")
+    affiliation = st_obj.text_input (label="Enter Affiliation", value="", key="na_upper_2")
+    if st_obj.button (label="Submit", key="ta_submit_1"):
+        row = [email, name, affiliation]
+        send_subscription_to_google_sheet(st_obj, row)
+
+def send_subscription_to_google_sheet(st_obj, row):
+    json_secret = readGoogleApiSecretsDict()
+    #print(json_secret)
+    spr = sp.spreadsheet (st_obj, json_secret)
+    spr.writeToSheet("CHIME Form Submissions", row)
 
 def display_footer(st):
     st.subheader("References & Acknowledgements")

src/penn_chime/spreadsheet.py

@@ -0,0 +1,45 @@
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+
+class spreadsheet:
+    def __init__(self, st_obj, secret):
+        self.st_obj = st_obj
+        # use creds to create a client to interact with the Google Drive API
+        self.scope = ['https://spreadsheets.google.com/feeds', 'https://www.googleapis.com/auth/drive']
+        #secret = 'client_secret.json'
+        self.creds = ServiceAccountCredentials.from_json(secret)
+        #self.creds = ServiceAccountCredentials.from_json_keyfile_name(secret, self.scope)
+        self.client = gspread.authorize(self.creds)
+
+
+    def createsheet(self, sheetname, header):
+        sheet = None
+        try:
+            self.client.create(sheetname)
+        except Exception as e:
+            print("Caught exception " + e.__str__())
+        try:
+            sheet = self.client.open (sheetname).sheet1
+            index = 0
+            sheet.insert_row (header, index)
+        except Exception as e:
+            print("Caught exception " + e.__str__())
+        return sheet
+
+    def writeToSheet(self, sheetname, row):
+        try:
+            sheet = self.client.open (sheetname).sheet1
+            list_of_hashes = sheet.get_all_records ( )
+            currentrow = sheet.row_count
+            sheet.insert_row (row, currentrow)
+        except Exception as e:
+            print("Caught exception " + e.__str__())
+
+    def getAllRows(self, sheetname):
+        allrows = None
+        try:
+            sheet = self.client.open (sheetname).sheet1
+            allrows = sheet.get_all_records ( )
+        except Exception as e:
+            print("Caught exception " + e.__str__())
+        return allrows