Rename each sheet we copy from the template

[turbot] / turbot / sheets.py

diff --git a/turbot/sheets.py b/turbot/sheets.py
index 9f8b7c3e0d38eabb0fd08332b684ea39789fea66..ff8b1423f3385c8bdb700ccbbbb14aa3c7c5d7a5 100644 (file)
--- a/turbot/sheets.py
+++ b/turbot/sheets.py
@@ -74,16 +74,36 @@ def sheets_create_for_puzzle(turb, puzzle):
 
     for sheet in response["sheets"]:
         if sheet["properties"]["title"] in PUZZLE_TEMPLATE_SHEETS:
-            turb.sheets.sheets().copyTo(spreadsheetId=PUZZLE_TEMPLATE_ID,
-                                        sheetId=sheet["properties"]["sheetId"],
-                                        body={
-                                            "destinationSpreadsheetId":
-                                            new_sheet['id']
-                                        }).execute()
+            res = turb.sheets.sheets().copyTo(
+                spreadsheetId=PUZZLE_TEMPLATE_ID,
+                sheetId=sheet["properties"]["sheetId"],
+                body={
+                    "destinationSpreadsheetId": new_sheet['id']
+                }).execute()
+            rename_sheet(turb, new_sheet['id'], res['sheetId'],
+                         sheet["properties"]["title"])
 
     return new_sheet
 
-def renameSheet(turb, url, newName):
+def rename_sheet(turb, spreadsheet_id, sheet_id, name):
+
+    body = {
+        'requests': [{
+            'updateSheetProperties': {
+                'properties': {
+                    'sheetId': sheet_id,
+                    'title': name
+                },
+                'fields': 'title'
+            }
+        }]
+    }
+
+    turb.sheets.batchUpdate(spreadsheetId=spreadsheet_id,
+                            body=body
+                            ).execute()
+
+def rename_spreadsheet(turb, url, newName):
     id = extractIdFromSheetUrl(url)
     requests = []
     requests.append({