git.cworth.org Git - turbot/blob - turbot/sheets.py

   1 PUZZLE_TEMPLATE_ID = "1drSoyrE4gM3JaGweDkOybwXWdKPIDTfUmB1gQCYS3Uw"
   2 PUZZLE_TEMPLATE_SHEETS = ["Text", "Grid"]
   3
   4 def sheets_create(turb, name):
   5     """Create a new sheet with the given name.
   6
   7     Returns the dict with 'id' and 'url'of the spreadsheet.
   8     """
   9
  10     # Create a new spreadsheet
  11     spreadsheet_body = {
  12         'properties': {
  13             'title': name
  14         }
  15     }
  16
  17     new_sheet = turb.sheets.create(body=spreadsheet_body).execute()
  18
  19     # Now that we've created a new sheet, we need to also allow anyone
  20     # with the link to the sheet to be able to edit it.
  21     turb.permissions.create(fileId=new_sheet["spreadsheetId"],
  22                             body={'type': 'anyone', 'role': 'writer'},
  23                             fields='id').execute()
  24
  25     return {
  26         'id': new_sheet["spreadsheetId"],
  27         'url': new_sheet["spreadsheetUrl"]
  28     }
  29
  30 def sheets_create_for_puzzle(turb, puzzle):
  31     """Creates a new sheet for a puzzle of the given name
  32
  33     Like sheets_create(), but also copies the puzzle template sheet.
  34
  35     Here, 'puzzle' is a dict that must have a 'name' key and may optionally
  36     have a 'channel_url' or 'url' key."""
  37
  38     # First create the new sheet
  39     new_sheet = sheets_create(turb, puzzle['name'])
  40
  41     # Insert some useful links into the sheet
  42     if 'url' in puzzle:
  43         url_text = "Original puzzle is at: {}".format(puzzle['url'])
  44     else:
  45         url_text = ''
  46
  47     if 'channel_url' in puzzle:
  48         channel_url_text = "Discussion for this puzzle is at: {}".format(
  49             puzzle['channel_url'])
  50     else:
  51         channel_url_text = ''
  52
  53     turb.sheets.values().append(
  54         spreadsheetId=new_sheet['id'],
  55         range='A1:A2',
  56         valueInputOption='USER_ENTERED',
  57         insertDataOption='INSERT_ROWS',
  58         body={
  59             'range': 'A1:A2',
  60             'values': [['=HYPERLINK('+url_text+')'], ['=HYPERLINK('+channel_url_text+')']]
  61         }).execute()
  62
  63     # Copy some sheets from the Template spreadsheet
  64
  65     response = turb.sheets.get(spreadsheetId=PUZZLE_TEMPLATE_ID).execute()
  66
  67     for sheet in response["sheets"]:
  68         if sheet["properties"]["title"] in PUZZLE_TEMPLATE_SHEETS:
  69             turb.sheets.sheets().copyTo(spreadsheetId=PUZZLE_TEMPLATE_ID,
  70                                         sheetId=sheet["properties"]["sheetId"],
  71                                         body={
  72                                             "destinationSpreadsheetId":
  73                                             new_sheet['id']
  74                                         }).execute()
  75
  76     return new_sheet