Update main.py

rmccorm4 · web-flow · commit fb4f52253f65 · 2019-01-17T10:53:25.000-08:00
Update to follow proper linting/style by using `six` library, because it is against standards to redefine builtin `input()`
diff --git a/drive/driveapp/main.py b/drive/driveapp/main.py
@@ -6,18 +6,13 @@
 """
 
 from __future__ import print_function
+import six
 import pprint
 import httplib2
 from googleapiclient.discovery import build
 import googleapiclient.http
 import oauth2client.client
 
-#Add user input support for both Python2 and Python3
-try: 
-    input = raw_input # Python2
-except:
-    pass # Python3
-
 # OAuth 2.0 scope that will be authorized.
 # Check https://developers.google.com/drive/scopes for all available scopes.
 OAUTH2_SCOPE = 'https://www.googleapis.com/auth/drive'
@@ -38,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)
-# If you can't use Python 3.x, please change "input" to "raw_input"
-code = input('Enter verification code: ').strip()
+# Add user input support for both Python2 and Python3
+code = six.moves.input('Enter verification code: ').strip()
 credentials = flow.step2_exchange(code)
 
 # Create an authorized Drive API client.
