added reading character from google sheet and fixed some bugs
surrounding rolling skills from the character sheet
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
@@ -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())
|
||||
|
||||
Reference in New Issue
Block a user