#454 created spreadsheet handler

#454 created spreadsheet handler

Author: mangala915

src/penn_chime/presentation.py

@@ -6,7 +6,7 @@
 import altair as alt
 import numpy as np
 import pandas as pd
-
+import penn_chime.spreadsheet as sp
 from .constants import (
     CHANGE_DATE,
     DATE_FORMAT,
@@ -355,14 +355,16 @@ 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 (email, name, affiliation)
+        send_subscription_to_google_sheet (st_obj, email, name, affiliation)
 
+    #####ToDo
     return Parameters(
         current_hospitalized=current_hospitalized,
         hospitalized=Disposition(hospitalized_rate, hospitalized_days),
@@ -380,9 +382,14 @@ def display_sidebar(st, d: Parameters) -> Parameters:
         population=population,
     )
 
-def send_subscription_to_google_sheet(email, name, affiliation):
+#ToDo
+def send_subscription_to_google_sheet(st_obj,email, name, affiliation):
     print ("send email:" + email + " name:" + name + " affiliation:" + affiliation + " to google sheet")
-    #implement sending the data to googlesheet
+    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 write_definitions(st):
     st.subheader("Guidance on Selecting Inputs")

src/penn_chime/spreadsheet.py

@@ -0,0 +1,37 @@
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+
+class spreadsheet:
+    def __init__(self, st_obj, secret, kwargs):
+        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']
+        #secret = 'client_secret.json'
+        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, sheet, row):
+        try:
+            currentrow = sheet.row_count
+            sheet.insert_row (row, currentrow)
+        except Exception as e:
+            print("Caught exception " + e.__str__())
+
+
+