diff --git a/user_module.py b/user_module.py index 680896c..5e239a3 100644 --- a/user_module.py +++ b/user_module.py @@ -99,7 +99,17 @@ class User(): def __init__(self, username:str = "user"): super().__init__() self.name = username - self.flags = [] + self.flags = {} + + def setFlag(self, name, flagType): + flag:UserFlag = UserFlag(name, flagType) + self.flags[name] = flag + + def getFlag(self, name): + return self.flags[name] + + def deleteFlag(self, name): + self.flags.pop(name, None) class UserMessage(): def __init__(self, user = "User", message = ""): @@ -114,7 +124,7 @@ class UserMessage(): class UserFlagTypes(Enum): REACTIVE = 1 -class UserFlags(): +class UserFlag(): def __init__(self, flagName = "User", flagType:UserFlagTypes = None): super().__init__() self.name = flagName