splittermond-backend/users/users.py

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())