Create a sheet for the hunt when creating a new hunt

[turbot] / turbot_lambda / turbot_lambda.py

diff --git a/turbot_lambda/turbot_lambda.py b/turbot_lambda/turbot_lambda.py
index 9d878af8c7e05a92b58846e96a20c9c54fd682c5..ab670d34c74c60476c90036fd0629834beb96703 100644 (file)
--- a/turbot_lambda/turbot_lambda.py
+++ b/turbot_lambda/turbot_lambda.py
@@ -1,10 +1,14 @@
 from urllib.parse import parse_qs
 from slack import WebClient
+import base64
 import boto3
 import requests
 import json
 import os
+import pickle
 from types import SimpleNamespace
+from google.auth.transport.requests import Request
+from googleapiclient.discovery import build
 
 import turbot.actions
 import turbot.commands
@@ -23,11 +27,34 @@ response = ssm.get_parameter(Name='SLACK_BOT_TOKEN', WithDecryption=True)
 slack_bot_token = response['Parameter']['Value']
 slack_client = WebClient(slack_bot_token)
 
+response = ssm.get_parameter(Name='GSHEETS_PICKLE_BASE64', WithDecryption=True)
+gsheets_pickle_base64 = response['Parameter']['Value']
+gsheets_pickle = base64.b64decode(gsheets_pickle_base64)
+gsheets_creds = pickle.loads(gsheets_pickle)
+if gsheets_creds:
+    if gsheets_creds.valid:
+        print("Loaded valid GSheets credentials from SSM")
+    else:
+        gsheets_creds.refresh(Request())
+        gsheets_pickle = pickle.dumps(gsheets_creds)
+        gsheets_pickle_base64 = base64.b64encode(gsheets_pickle)
+        print("Storing refreshed GSheets credentials into SSM")
+        ssm.put_parameter(Name='GSHEETS_PICKLE_BASE64',
+                          Type='SecureString',
+                          Value=gsheets_pickle_base64,
+                          Overwrite=True)
+service = build('sheets',
+                'v4',
+                credentials=gsheets_creds,
+                cache_discovery=False)
+sheets = service.spreadsheets()
+
 db = boto3.resource('dynamodb')
 
 turb = SimpleNamespace()
 turb.slack_client = slack_client
 turb.db = db
+turb.sheets = sheets
 
 def error(message):
     """Generate an error response for a Slack request