Switch sheets/quickstart to from oauth2client to google-auth

slokhorst · slokhorst · commit 1987467baa9b · 2019-01-11T22:07:02.000+01:00
diff --git a/sheets/quickstart/README.md b/sheets/quickstart/README.md
@@ -8,7 +8,7 @@ requests to the Google Sheets API.
 ## Install
 
 ```
-pip install --upgrade google-api-python-client oauth2client
+pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib pickle
 ```
 
 
diff --git a/sheets/quickstart/quickstart.py b/sheets/quickstart/quickstart.py
@@ -14,11 +14,13 @@
 
 # [START sheets_quickstart]
 from __future__ import print_function
+import pickle
+import os.path
 from googleapiclient.discovery import build
-from httplib2 import Http
-from oauth2client import file, client, tools
+from google_auth_oauthlib.flow import InstalledAppFlow
+from google.auth.transport.requests import Request
 
-# If modifying these scopes, delete the file token.json.
+# If modifying these scopes, delete the file token.pickle.
 SCOPES = 'https://www.googleapis.com/auth/spreadsheets.readonly'
 
 # The ID and range of a sample spreadsheet.
@@ -29,15 +31,26 @@ def main():
     """Shows basic usage of the Sheets API.
     Prints values from a sample spreadsheet.
     """
-    # The file token.json stores the user's access and refresh tokens, and is
+    creds = None
+    # The file token.pickle stores the user's access and refresh tokens, and is
     # created automatically when the authorization flow completes for the first
     # time.
-    store = file.Storage('token.json')
-    creds = store.get()
-    if not creds or creds.invalid:
-        flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
-        creds = tools.run_flow(flow, store)
-    service = build('sheets', 'v4', http=creds.authorize(Http()))
+    if os.path.exists('token.pickle'):
+        with open('token.pickle', 'rb') as token:
+            creds = pickle.load(token)
+    # If there are no (valid) credentials available, let the user log in.
+    if not creds or not creds.valid:
+        if creds and creds.expired and creds.refresh_token:
+            creds.refresh(Request())
+        else:
+            flow = InstalledAppFlow.from_client_secrets_file(
+                'credentials.json', SCOPES)
+            creds = flow.run_local_server()
+        # Save the credentials for the next run
+        with open('token.pickle', 'wb') as token:
+            pickle.dump(creds, token)
+
+    service = build('sheets', 'v4', credentials=creds)
 
     # Call the Sheets API
     sheet = service.spreadsheets()
