Skip to content

Instantly share code, notes, and snippets.

@takidog
Created March 11, 2019 13:33
Show Gist options
  • Save takidog/5d6190a0ef39767c2f0c7d939bb36985 to your computer and use it in GitHub Desktop.
Save takidog/5d6190a0ef39767c2f0c7d939bb36985 to your computer and use it in GitHub Desktop.
Oauth
from apiclient import discovery
from oauth2client import client
from oauth2client import tools
from oauth2client.file import Storage
try:
import argparse
flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args()
except ImportError:
flags = None
def get_credentials():
"""Gets valid user credentials from storage.
If nothing has been stored, or if the stored credentials are invalid,
the OAuth2 flow is completed to obtain the new credentials.
Returns:
Credentials, the obtained credential.
"""
home_dir = os.path.expanduser('~')
credential_dir = os.path.join(home_dir, '.credentials')
if not os.path.exists(credential_dir):
os.makedirs(credential_dir)
credential_path = os.path.join(credential_dir,
'drive-python-quickstart.json')
store = Storage(credential_path)
credentials = store.get()
if not credentials or credentials.invalid:
flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES)
flow.user_agent = APPLICATION_NAME
if flags:
credentials = tools.run_flow(flow, store, flags)
else: # Needed only for compatibility with Python 2.6
credentials = tools.run(flow, store)
print('Storing credentials to ' + credential_path)
return credentials
credentials = get_credentials()
http = credentials.authorize(httplib2.Http())
# change your API type e.g. youtube ...
service = discovery.build('drive', 'v2', http=http)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment