2019-06-17 20:10:03 +02:00
|
|
|
from __future__ import print_function
|
|
|
|
|
import pickle
|
|
|
|
|
import os.path
|
|
|
|
|
from googleapiclient.discovery import build
|
|
|
|
|
from google_auth_oauthlib.flow import InstalledAppFlow
|
|
|
|
|
from google.auth.transport.requests import Request
|
|
|
|
|
|
|
|
|
|
# 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.
|
|
|
|
|
SPREADSHEET_ID = '1aICR-owH6CENOJ7l5gnCZTYo9z1b-ddzi0fx0cKkvfI'
|
|
|
|
|
RANGES = ['A12:C29', 'H12:J29', 'O20:Q23']
|
|
|
|
|
|
|
|
|
|
def load_character(character):
|
|
|
|
|
"""Shows basic usage of the Sheets API.
|
|
|
|
|
Prints values from a sample spreadsheet.
|
|
|
|
|
"""
|
|
|
|
|
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.
|
|
|
|
|
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()
|
|
|
|
|
values = []
|
|
|
|
|
ranges = ['{}!{}'.format(character, r) for r in RANGES]
|
|
|
|
|
result = sheet.values().batchGet(spreadsheetId=SPREADSHEET_ID,
|
|
|
|
|
ranges=ranges).execute()
|
|
|
|
|
skills = dict()
|
|
|
|
|
for r in result.get('valueRanges', []):
|
|
|
|
|
for v in r.get('values', []):
|
|
|
|
|
if v:
|
|
|
|
|
skills[v[0]] = int(v[2])
|
|
|
|
|
|
|
|
|
|
return skills
|
|
|
|
|
|
