37 lines
961 B
Python
37 lines
961 B
Python
import chat.response as response
|
|
import users.sheet_reader as sheet_reader
|
|
|
|
|
|
class UserManager:
|
|
def __init__(self):
|
|
self.users = dict()
|
|
self.skill_dict = dict()
|
|
|
|
def add_user(self, name, sid):
|
|
self.users[sid] = 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):
|
|
if sid in self.users:
|
|
return self.users.pop(sid)
|
|
|
|
def has_user(self, name):
|
|
return name in self.users.values()
|
|
|
|
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())
|