Update main.py

rmccorm4 · web-flow · commit d2e3feeaa1cf · 2019-01-17T11:03:34.000-08:00
Back to `six` library
diff --git a/drive/driveapp/main.py b/drive/driveapp/main.py
@@ -8,10 +8,10 @@
 from __future__ import print_function
 import pprint
 import httplib2
+import six
 from googleapiclient.discovery import build
 import googleapiclient.http
 import oauth2client.client
-from builtins import input
 
 # OAuth 2.0 scope that will be authorized.
 # Check https://developers.google.com/drive/scopes for all available scopes.
@@ -33,8 +33,8 @@
 flow.redirect_uri = oauth2client.client.OOB_CALLBACK_URN
 authorize_url = flow.step1_get_authorize_url()
 print('Go to the following link in your browser: ' + authorize_url)
-# Should support both Python2 and Python3 with 'from builtins import input'
-code = input('Enter verification code: ').strip()
+# `six` library supports Python2 and Python3 without redefining builtin input()
+code = six.moves.input('Enter verification code: ').strip()
 credentials = flow.step2_exchange(code)
 
 # Create an authorized Drive API client.
