""" gspread.spreadsheet ~~~~~~~~~~~~~~ This module contains common spreadsheets' models. """ from .exceptions import WorksheetNotFound from .urls import ( DRIVE_FILES_API_V3_URL, SPREADSHEET_BATCH_UPDATE_URL, SPREADSHEET_DRIVE_URL, SPREADSHEET_SHEETS_COPY_TO_URL, SPREADSHEET_URL, SPREADSHEET_VALUES_APPEND_URL, SPREADSHEET_VALUES_BATCH_CLEAR_URL, SPREADSHEET_VALUES_BATCH_UPDATE_URL, SPREADSHEET_VALUES_BATCH_URL, SPREADSHEET_VALUES_CLEAR_URL, SPREADSHEET_VALUES_URL, ) from .utils import ExportFormat, finditem, quote from .worksheet import Worksheet class Spreadsheet: """The class that represents a spreadsheet.""" def __init__(self, client, properties): self.client = client self._properties = properties metadata = self.fetch_sheet_metadata() self._properties.update(metadata["properties"]) @property def id(self): """Spreadsheet ID.""" return self._properties["id"] @property def title(self): """Spreadsheet title.""" return self._properties["title"] @property def url(self): """Spreadsheet URL.""" return SPREADSHEET_DRIVE_URL % self.id @property def creationTime(self): """Spreadsheet Creation time.""" try: return self._properties["createdTime"] except KeyError: # Filter the list using the name to reduce the request size # Filter the item using the unique ID to ensure we update the exacte same item metadata = finditem( lambda x: x["id"] == self.id, self.client.list_spreadsheet_files(self.title), ) self._properties.update(metadata) return self._properties["createdTime"] @property def lastUpdateTime(self): """Spreadsheet last updated time.""" try: return self._properties["modifiedTime"] except KeyError: # Filter the list using the name to reduce the request size # Filter the item using the unique ID to ensure we update the exacte same item metadata = finditem( lambda x: x["id"] == self.id, self.client.list_spreadsheet_files(self.title), ) self._properties.update(metadata) return self._properties["modifiedTime"] @property def updated(self): """.. deprecated:: 2.0 This feature is not supported in Sheets API v4. """ import warnings warnings.warn( "Spreadsheet.updated() is deprecated, " "this feature is not supported in Sheets API v4", DeprecationWarning, stacklevel=2, ) @property def timezone(self): """Spreadsheet timeZone""" return self._properties["timeZone"] @property def locale(self): """Spreadsheet locale""" return self._properties["locale"] @property def sheet1(self): """Shortcut property for getting the first worksheet.""" return self.get_worksheet(0) def __iter__(self): yield from self.worksheets() def __repr__(self): return "<{} {} id:{}>".format( self.__class__.__name__, repr(self.title), self.id, ) def batch_update(self, body): """Lower-level method that directly calls `spreadsheets/:batchUpdate `_. :param dict body: `Request body `_. :returns: `Response body `_. :rtype: dict .. versionadded:: 3.0 """ r = self.client.request( "post", SPREADSHEET_BATCH_UPDATE_URL % self.id, json=body ) return r.json() def values_append(self, range, params, body): """Lower-level method that directly calls `spreadsheets//values:append `_. :param str range: The `A1 notation `_ of a range to search for a logical table of data. Values will be appended after the last row of the table. :param dict params: `Query parameters `_. :param dict body: `Request body`_. :returns: `Response body`_. :rtype: dict .. versionadded:: 3.0 """ url = SPREADSHEET_VALUES_APPEND_URL % (self.id, quote(range)) r = self.client.request("post", url, params=params, json=body) return r.json() def values_clear(self, range): """Lower-level method that directly calls `spreadsheets//values:clear `_. :param str range: The `A1 notation `_ of the values to clear. :returns: `Response body`_. :rtype: dict .. versionadded:: 3.0 """ url = SPREADSHEET_VALUES_CLEAR_URL % (self.id, quote(range)) r = self.client.request("post", url) return r.json() def values_batch_clear(self, params=None, body=None): url = SPREADSHEET_VALUES_BATCH_CLEAR_URL % self.id r = self.client.request("post", url, params=params, json=body) return r.json() def values_get(self, range, params=None): """Lower-level method that directly calls `spreadsheets//values/ `_. :param str range: The `A1 notation `_ of the values to retrieve. :param dict params: (optional) `Query parameters`_. :returns: `Response body`_. :rtype: dict .. versionadded:: 3.0 """ url = SPREADSHEET_VALUES_URL % (self.id, quote(range)) r = self.client.request("get", url, params=params) return r.json() def values_batch_get(self, ranges, params=None): """Lower-level method that directly calls `spreadsheets//values:batchGet `_. :param ranges: List of ranges in the `A1 notation `_ of the values to retrieve. :param dict params: (optional) `Query parameters`_. :returns: `Response body`_. :rtype: dict """ if params is None: params = {} params.update(ranges=ranges) url = SPREADSHEET_VALUES_BATCH_URL % (self.id) r = self.client.request("get", url, params=params) return r.json() def values_update(self, range, params=None, body=None): """Lower-level method that directly calls `spreadsheets//values/`_. :param str range: The `A1 notation `_ of the values to update. :param dict params: (optional) `Query parameters`_. :param dict body: (optional) `Request body`_. :returns: `Response body`_. :rtype: dict Example:: sh.values_update( 'Sheet1!A2', params={ 'valueInputOption': 'USER_ENTERED' }, body={ 'values': [[1, 2, 3]] } ) .. versionadded:: 3.0 """ url = SPREADSHEET_VALUES_URL % (self.id, quote(range)) r = self.client.request("put", url, params=params, json=body) return r.json() def values_batch_update(self, params=None, body=None): url = SPREADSHEET_VALUES_BATCH_UPDATE_URL % self.id r = self.client.request("post", url, params=params, json=body) return r.json() def _spreadsheets_get(self, params=None): """A method stub that directly calls `spreadsheets.get `_.""" url = SPREADSHEET_URL % self.id r = self.client.request("get", url, params=params) return r.json() def _spreadsheets_sheets_copy_to(self, sheet_id, destination_spreadsheet_id): """Lower-level method that directly calls `spreadsheets.sheets.copyTo `_.""" url = SPREADSHEET_SHEETS_COPY_TO_URL % (self.id, sheet_id) body = {"destinationSpreadsheetId": destination_spreadsheet_id} r = self.client.request("post", url, json=body) return r.json() def fetch_sheet_metadata(self, params=None): if params is None: params = {"includeGridData": "false"} url = SPREADSHEET_URL % self.id r = self.client.request("get", url, params=params) return r.json() def get_worksheet(self, index): """Returns a worksheet with specified `index`. :param index: An index of a worksheet. Indexes start from zero. :type index: int :returns: an instance of :class:`gspread.worksheet.Worksheet`. :raises: :class:`~gspread.exceptions.WorksheetNotFound`: if can't find the worksheet Example. To get third worksheet of a spreadsheet: >>> sht = client.open('My fancy spreadsheet') >>> worksheet = sht.get_worksheet(2) """ sheet_data = self.fetch_sheet_metadata() try: properties = sheet_data["sheets"][index]["properties"] return Worksheet(self, properties) except (KeyError, IndexError): raise WorksheetNotFound("index {} not found".format(index)) def get_worksheet_by_id(self, id): """Returns a worksheet with specified `worksheet id`. :param id: The id of a worksheet. it can be seen in the url as the value of the parameter 'gid'. :type id: int :returns: an instance of :class:`gspread.worksheet.Worksheet`. :raises: :class:`~gspread.exceptions.WorksheetNotFound`: if can't find the worksheet Example. To get the worksheet 123456 of a spreadsheet: >>> sht = client.open('My fancy spreadsheet') >>> worksheet = sht.get_worksheet_by_id(123456) """ sheet_data = self.fetch_sheet_metadata() try: item = finditem( lambda x: x["properties"]["sheetId"] == id, sheet_data["sheets"], ) return Worksheet(self, item["properties"]) except (StopIteration, KeyError): raise WorksheetNotFound("id {} not found".format(id)) def worksheets(self): """Returns a list of all :class:`worksheets ` in a spreadsheet. """ sheet_data = self.fetch_sheet_metadata() return [Worksheet(self, x["properties"]) for x in sheet_data["sheets"]] def worksheet(self, title): """Returns a worksheet with specified `title`. :param title: A title of a worksheet. If there're multiple worksheets with the same title, first one will be returned. :type title: str :returns: an instance of :class:`gspread.worksheet.Worksheet`. :raises: WorksheetNotFound: if can't find the worksheet Example. Getting worksheet named 'Annual bonuses' >>> sht = client.open('Sample one') >>> worksheet = sht.worksheet('Annual bonuses') """ sheet_data = self.fetch_sheet_metadata() try: item = finditem( lambda x: x["properties"]["title"] == title, sheet_data["sheets"], ) return Worksheet(self, item["properties"]) except (StopIteration, KeyError): raise WorksheetNotFound(title) def add_worksheet(self, title, rows, cols, index=None): """Adds a new worksheet to a spreadsheet. :param title: A title of a new worksheet. :type title: str :param rows: Number of rows. :type rows: int :param cols: Number of columns. :type cols: int :param index: Position of the sheet. :type index: int :returns: a newly created :class:`worksheets `. """ body = { "requests": [ { "addSheet": { "properties": { "title": title, "sheetType": "GRID", "gridProperties": { "rowCount": rows, "columnCount": cols, }, } } } ] } if index is not None: body["requests"][0]["addSheet"]["properties"]["index"] = index data = self.batch_update(body) properties = data["replies"][0]["addSheet"]["properties"] worksheet = Worksheet(self, properties) return worksheet def duplicate_sheet( self, source_sheet_id, insert_sheet_index=None, new_sheet_id=None, new_sheet_name=None, ): """Duplicates the contents of a sheet. :param int source_sheet_id: The sheet ID to duplicate. :param int insert_sheet_index: (optional) The zero-based index where the new sheet should be inserted. The index of all sheets after this are incremented. :param int new_sheet_id: (optional) The ID of the new sheet. If not set, an ID is chosen. If set, the ID must not conflict with any existing sheet ID. If set, it must be non-negative. :param str new_sheet_name: (optional) The name of the new sheet. If empty, a new name is chosen for you. :returns: a newly created :class:`gspread.worksheet.Worksheet` .. versionadded:: 3.1 """ body = { "requests": [ { "duplicateSheet": { "sourceSheetId": source_sheet_id, "insertSheetIndex": insert_sheet_index, "newSheetId": new_sheet_id, "newSheetName": new_sheet_name, } } ] } data = self.batch_update(body) properties = data["replies"][0]["duplicateSheet"]["properties"] worksheet = Worksheet(self, properties) return worksheet def del_worksheet(self, worksheet): """Deletes a worksheet from a spreadsheet. :param worksheet: The worksheet to be deleted. :type worksheet: :class:`~gspread.worksheet.Worksheet` """ body = {"requests": [{"deleteSheet": {"sheetId": worksheet.id}}]} return self.batch_update(body) def reorder_worksheets(self, worksheets_in_desired_order): """Updates the ``index`` property of each Worksheet to reflect its index in the provided sequence of Worksheets. :param worksheets_in_desired_order: Iterable of Worksheet objects in desired order. Note: If you omit some of the Spreadsheet's existing Worksheet objects from the provided sequence, those Worksheets will be appended to the end of the sequence in the order that they appear in the list returned by :meth:`gspread.spreadsheet.Spreadsheet.worksheets`. .. versionadded:: 3.4 """ idx_map = {} for idx, w in enumerate(worksheets_in_desired_order): idx_map[w.id] = idx for w in self.worksheets(): if w.id in idx_map: continue idx += 1 idx_map[w.id] = idx body = { "requests": [ { "updateSheetProperties": { "properties": {"sheetId": key, "index": val}, "fields": "index", } } for key, val in idx_map.items() ] } return self.batch_update(body) def share( self, email_address, perm_type, role, notify=True, email_message=None, with_link=False, ): """Share the spreadsheet with other accounts. :param value: user or group e-mail address, domain name or None for 'default' type. :type value: str, None :param perm_type: The account type. Allowed values are: ``user``, ``group``, ``domain``, ``anyone``. :type perm_type: str :param role: The primary role for this user. Allowed values are: ``owner``, ``writer``, ``reader``. :type role: str :param notify: (optional) Whether to send an email to the target user/domain. :type notify: bool :param email_message: (optional) The email to be sent if notify=True :type email_message: str :param with_link: (optional) Whether the link is required for this permission :type with_link: bool Example:: # Give Otto a write permission on this spreadsheet sh.share('otto@example.com', perm_type='user', role='writer') # Give Otto's family a read permission on this spreadsheet sh.share('otto-familly@example.com', perm_type='group', role='reader') """ self.client.insert_permission( self.id, value=email_address, perm_type=perm_type, role=role, notify=notify, email_message=email_message, with_link=with_link, ) def export(self, format=ExportFormat.PDF): """Export the spreadsheet in the given format. :param str file_id: A key of a spreadsheet to export :param str format: The format of the resulting file. Possible values are: ``ExportFormat.PDF``, ``ExportFormat.EXCEL``, ``ExportFormat.CSV``, ``ExportFormat.OPEN_OFFICE_SHEET``, ``ExportFormat.TSV``, and ``ExportFormat.ZIPPED_HTML``. See `ExportFormat`_ in the Drive API. Default value is ``ExportFormat.PDF``. :returns bytes: The content of the exported file. .. _ExportFormat: https://developers.google.com/drive/api/guides/ref-export-formats """ return self.client.export(self.id, format) def list_permissions(self): """Lists the spreadsheet's permissions.""" return self.client.list_permissions(self.id) def remove_permissions(self, value, role="any"): """Remove permissions from a user or domain. :param value: User or domain to remove permissions from :type value: str :param role: (optional) Permission to remove. Defaults to all permissions. :type role: str Example:: # Remove Otto's write permission for this spreadsheet sh.remove_permissions('otto@example.com', role='writer') # Remove all Otto's permissions for this spreadsheet sh.remove_permissions('otto@example.com') """ permission_list = self.client.list_permissions(self.id) key = "emailAddress" if "@" in value else "domain" filtered_id_list = [ p["id"] for p in permission_list if p.get(key) == value and (p["role"] == role or role == "any") ] for permission_id in filtered_id_list: self.client.remove_permission(self.id, permission_id) return filtered_id_list def transfer_ownership(self, permission_id): """Transfer the ownership of this file to a new user. It is necessary to first create the permission with the new owner's email address, get the permission ID then use this method to transfer the ownership. .. note:: You can list all permissions using :meth:`gspread.spreadsheet.Spreadsheet.list_permissions`. .. warning:: You can only transfer ownership to a new user, you cannot transfer ownership to a group or a domain email address. """ url = "{}/{}/permissions/{}".format( DRIVE_FILES_API_V3_URL, self.id, permission_id ) payload = { "role": "writer", # new owner must be writer in order to accept ownership by editing permissions "pendingOwner": True, } return self.client.request("patch", url, json=payload) def accept_ownership(self, permission_id): """Accept the pending ownership request on that file. It is necessary to edit the permission with the pending ownership. .. note:: You can only accept ownership transfer for the user currently being used. """ url = "{}/{}/permissions/{}".format( DRIVE_FILES_API_V3_URL, self.id, permission_id, ) payload = { "role": "owner", } params = { "transferOwnership": True, } return self.client.request("patch", url, json=payload, params=params) def named_range(self, named_range): """return a list of :class:`gspread.cell.Cell` objects from the specified named range. :param named_range: A string with a named range value to fetch. :type named_range: str """ # the function `range` does all necessary actions to get a named range. # This is only here to provide better user experience. return self.sheet1.range(named_range) def list_named_ranges(self): """Lists the spreadsheet's named ranges.""" return self.fetch_sheet_metadata(params={"fields": "namedRanges"}).get( "namedRanges", [] ) def update_title(self, title): """Renames the spreadsheet. :param str title: A new title. """ body = { "requests": [ { "updateSpreadsheetProperties": { "properties": {"title": title}, "fields": "title", } } ] } response = self.batch_update(body) self._properties["title"] = title return response def update_timezone(self, timezone): """Updates the current spreadsheet timezone. Can be any timezone in CLDR format such as "America/New_York" or a custom time zone such as GMT-07:00. """ body = { "requests": [ { "updateSpreadsheetProperties": { "properties": {"timeZone": timezone}, "fields": "timeZone", }, }, ] } return self.batch_update(body) def update_locale(self, locale): """Update the locale of the spreadsheet. Can be any of the ISO 639-1 language codes, such as: de, fr, en, ... Or an ISO 639-2 if no ISO 639-1 exists. Or a combination of the ISO language code and country code, such as en_US, de_CH, fr_FR, ... .. note:: Note: when updating this field, not all locales/languages are supported. """ body = { "requests": [ { "updateSpreadsheetProperties": { "properties": {"locale": locale}, "fields": "locale", }, }, ] } return self.batch_update(body) def list_protected_ranges(self, sheetid): """Lists the spreadsheet's protected named ranges""" sheets = self.fetch_sheet_metadata( params={"fields": "sheets.properties,sheets.protectedRanges"} )["sheets"] try: sheet = finditem( lambda sheet: sheet["properties"]["sheetId"] == sheetid, sheets ) except StopIteration: raise WorksheetNotFound("worksheet id {} not found".format(sheetid)) return sheet.get("protectedRanges", [])