added reading character from google sheet and fixed some bugs

surrounding rolling skills from the character sheet
This commit is contained in:
daniel
2019-06-17 20:10:03 +02:00
parent ccc9c44ebd
commit 8e29452fe4
5 changed files with 87 additions and 22 deletions

View File

@@ -0,0 +1,53 @@
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

View File

@@ -1,18 +1,26 @@
import chat.response as response
import users.sheet_reader as sheet_reader
class UserManager:
def __init__(self):
self.users = dict()
self.skill_map = dict()
self.skill_dict = dict()
def add_user(self, name, sid):
self.users[sid] = name
print("ADDED USER, USERS NOW IS:", self.users)
return name
if name not in self.skill_dict:
try:
self.skill_dict[name] = sheet_reader.load_character(name)
loaded = True
except Exception as e:
print(e)
loaded = False
else:
loaded = True
return name, loaded
def remove_user(self, sid):
print("REMOVING USER:", sid)
if sid in self.users:
return self.users.pop(sid)
@@ -22,3 +30,7 @@ class UserManager:
def get_users(self):
return self.users.values()
def get_skills(self, name):
if name in self.skill_dict:
skills = self.skill_dict[name]
return '\n'.join(skills.keys())