#454 Added sending data to googlesheet

#454 Added sending data to googlesheet

Author: mangala915

src/penn_chime/presentation.py

@@ -355,16 +355,9 @@ def display_sidebar(st, d: Parameters) -> Parameters:
         max_y_axis = max_y_axis_input()
 
     current_date = current_date_input()
-    ####ToDo
     #Subscribe implementation
-    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"):
-        send_subscription_to_google_sheet (st_obj, email, name, affiliation)
+    subscribe(st_obj)
 
-    #####ToDo
     return Parameters(
         current_hospitalized=current_hospitalized,
         hospitalized=Disposition(hospitalized_rate, hospitalized_days),
@@ -382,14 +375,18 @@ def display_sidebar(st, d: Parameters) -> Parameters:
         population=population,
     )
 
-#ToDo
-def send_subscription_to_google_sheet(st_obj,email, name, affiliation):
-    print ("send email:" + email + " name:" + name + " affiliation:" + affiliation + " to google sheet")
-    spr = sp.spreadsheet (st_obj, 'penn_chime/client_secret.json')
-    header = ["ContactEmail","Name","Affiliation"]
-    sheet = spr.createsheet("SheetnameToDo", header)
-    row = [email,name,affiliation]
-    spr.writeToSheet(sheet, row)
+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):
+    spr = sp.spreadsheet (st_obj, 'client_secret.json')
+    spr.writeToSheet("CHIME Form Submissions", row)
 
 def write_definitions(st):
     st.subheader("Guidance on Selecting Inputs")
@@ -400,7 +397,6 @@ def write_definitions(st):
         )
     )
 
-
 def write_footer(st):
     st.subheader("References & Acknowledgements")
     st.markdown(

src/penn_chime/spreadsheet.py

@@ -2,11 +2,10 @@
 from oauth2client.service_account import ServiceAccountCredentials
 
 class spreadsheet:
-    def __init__(self, st_obj, secret, kwargs):
+    def __init__(self, st_obj, secret):
         self.st_obj = st_obj
-        self.kwargs = kwargs
         # use creds to create a client to interact with the Google Drive API
-        self.scope = ['https://spreadsheets.google.com/feeds']
+        self.scope = ['https://spreadsheets.google.com/feeds', 'https://www.googleapis.com/auth/drive']
         #secret = 'client_secret.json'
         self.creds = ServiceAccountCredentials.from_json_keyfile_name(secret, self.scope)
         self.client = gspread.authorize(self.creds)
@@ -26,12 +25,20 @@ def createsheet(self, sheetname, header):
             print("Caught exception " + e.__str__())
         return sheet
 
-    def writeToSheet(self, sheet, row):
+    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