Fix sheet creation to copy all sheets from our template

[turbot] / turbot / sheets.py

diff --git a/turbot/sheets.py b/turbot/sheets.py
index fbe9e3d7839d2e52a64d43f471a444efa1869fe3..9f8b7c3e0d38eabb0fd08332b684ea39789fea66 100644 (file)
--- a/turbot/sheets.py
+++ b/turbot/sheets.py
@@ -1,5 +1,6 @@
 PUZZLE_TEMPLATE_ID = "1drSoyrE4gM3JaGweDkOybwXWdKPIDTfUmB1gQCYS3Uw"
-PUZZLE_TEMPLATE_SHEETS = ["Text", "Grid"]
+PUZZLE_TEMPLATE_SHEETS = ["Text", "Square grid", "Hex Grid",
+                          "Formula reference: indexing"]
 
 def sheets_create(turb, name):
     """Create a new sheet with the given name.
@@ -39,14 +40,18 @@ def sheets_create_for_puzzle(turb, puzzle):
     new_sheet = sheets_create(turb, puzzle['name'])
 
     # Insert some useful links into the sheet
+    url_link=''
     if 'url' in puzzle:
         url_text = "Original puzzle is at: {}".format(puzzle['url'])
+        url_link = puzzle['url']
     else:
         url_text = ''
 
+    channel_url_link = ''
     if 'channel_url' in puzzle:
         channel_url_text = "Discussion for this puzzle is at: {}".format(
             puzzle['channel_url'])
+        channel_url_link = puzzle['channel_url']
     else:
         channel_url_text = ''
 
@@ -57,7 +62,10 @@ def sheets_create_for_puzzle(turb, puzzle):
         insertDataOption='INSERT_ROWS',
         body={
             'range': 'A1:A2',
-            'values': [[url_text], [channel_url_text]]
+            'values': [
+                ['=HYPERLINK("'+url_link+'","'+url_text+'")'],
+                ['=HYPERLINK("'+channel_url_link+'","'+channel_url_text+'")']
+            ]
         }).execute()
 
     # Copy some sheets from the Template spreadsheet
@@ -74,3 +82,30 @@ def sheets_create_for_puzzle(turb, puzzle):
                                         }).execute()
 
     return new_sheet
+
+def renameSheet(turb, url, newName):
+    id = extractIdFromSheetUrl(url)
+    requests = []
+    requests.append({
+        'updateSpreadsheetProperties': {
+            'properties': {
+                'title': newName
+            },
+            'fields': 'title'
+        }
+    })
+
+    body = {
+        'requests': requests
+    }
+
+    turb.sheets.batchUpdate(spreadsheetId = id,
+                            body=body
+                            ).execute()
+
+def extractIdFromSheetUrl(url):
+    # Google sheet ids are between the /d/ and /edit in the url, like
+    # https://docs.google.com/spreadsheets/d/1dxHBzjen...-LaXeVPrg/edit#gid=0
+    startIndex = url.find('/d/') + 3
+    endIndex = url.find('/edit')
+    return url[startIndex : endIndex]