From 91c31e9646dc300c30ac258106cb5b320efcc9a3 Mon Sep 17 00:00:00 2001
From: Alex Orid <TheCuriousNerd@users.noreply.github.com>
Date: Fri, 16 Apr 2021 02:59:17 -0400
Subject: [PATCH] User Flags

---
 user_module.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

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