cleaning up

funcs/swfuncs/swchar.py

@@ -0,0 +1,182 @@
+import json
+import string
+
+def getName(user : str):
+    with open("characters.json", "r") as f:
+        data = json.load(f)
+    
+    if user in data:
+        return data[user]["Name"]
+    else:
+        return user
+
+def setUpDict(cmd : dict):
+    if bool(dict):
+        keys = list(cmd)
+        values = list(cmd.values())
+        result = ""
+        if type(values[0]) is dict:
+            return ", ".join(values)
+        else:
+            for x in range(len(keys)):
+                if x%3 != 2:
+                    result += "**" + keys[x] + "**" + ": " + str(values[x]) + " "
+                else:
+                    result += "**" + keys[x] + "**" + ": " + str(values[x]) + "\n"
+            return result
+    else:
+        return "There doesn't seem to be anything here..."
+
+def lookUp(data : dict, key : str, cmd : str = ""):
+    if cmd == " ":
+        cmd = ""
+    elif cmd != "":
+        while cmd[0] == " ":
+            cmd = cmd[1:]
+            if cmd == "":
+                break
+
+    if key in data:
+        if cmd == "":
+            if type(data[key]) is dict:
+                return setUpDict(data[key])
+            else:
+                return data[key]
+        elif cmd[0] == '+':
+            try:
+                cmd = cmd[1:]
+                while cmd[0] == ' ':
+                    cmd = cmd[1:]
+            except:
+                return "Can't do that"
+
+            if type(data[key]) is int:
+                try:
+                    newValue = data[key] + int(cmd)
+                    data[key] = newValue
+                    return data
+                except:
+                    return "Can't add that"
+            else:
+                return "Can't add that"
+        else:
+            while cmd[0] == ' ':
+                cmd = cmd[1:]
+            try:
+                cmd = type(data[key])(cmd)
+                data[key] = cmd
+                return data
+            except:
+                return "Wrong data type"
+    else:
+        return key + " doesn't exist"
+
+def characterSheet(character : dict):
+    divider = "\n----------\n"
+    name = character["Name"]
+    text1 = "Species: "+character["Species"]+"\nCareer: "+character["Career"]
+    text2 = setUpDict(character["Characteristics"])
+    text3 = setUpDict(character["Skills"])
+
+    return name, text1+"\n\n"+text2+divider+text3
+
+def charData(user : str,cmd : str):
+    with open("characters.json", "r") as f:
+        data = json.load(f)
+
+    key = string.capwords(cmd.split(" ")[0])
+    cmd = cmd[len(key):]
+    if cmd == " ":
+        cmd = ""
+    elif cmd != "":
+        while cmd[0] == " ":
+            cmd = cmd[1:]
+            if cmd == "":
+                break
+
+    if user in data:
+        if key in data[user]:
+            if type(data[user][key]) is dict:
+                if cmd == "":
+                    return setUpDict(data[user][key])
+                else:
+                    newKey = string.capwords(cmd.split(" ")[0])
+                    newcmd = cmd[len(newKey):]
+
+                    lookUpResult = lookUp(data[user][key],newKey,newcmd)
+
+                    if type(lookUpResult) is dict:
+                        data[user][key] = lookUpResult
+                        with open("characters.json", "w") as f:
+                            json.dump(data,f,indent = 4)
+                        return "Changed " + data[user]["Name"] + "'s " + key
+                    else:
+                        return lookUpResult
+            else:
+                if cmd == "":
+                    return data[user][key]
+                elif cmd[0] == '+':
+                    try:
+                        cmd = cmd[1:]
+                        while cmd[0] == ' ':
+                            cmd = cmd[1:]
+                    except:
+                        return "Can't do that"
+
+                    if type(data[user][key]) is int:
+                        try:
+                            data[user][key] += int(cmd)
+                            return "Added " + cmd + " to " + user + "'s " + key
+                        except:
+                            return "Can't add that"
+                    else:
+                        return "Can't add that"
+                else:
+                    data[user][key] = cmd
+                    with open("characters.json", "w") as f:
+                            json.dump(data,f,indent = 4)
+                    return "Changed " + data[user]["Name"] + "'s " + key +" to " + cmd
+        else:
+            return "Couldn't find that data. Are you sure you spelled it correctly?"
+    else:
+        return "You don't have a character. You can make one with !swchar"
+
+def parseChar(user : str, cmd : str):
+    with open("characters.json", "r") as f:
+        data = json.load(f)
+
+    if cmd == " ":
+        cmd = ""
+    elif cmd != "":
+        while cmd[0] == " ":
+            cmd = cmd[1:]
+            if cmd == "":
+                break
+    
+    if cmd == "":
+        if user in data:
+            return characterSheet(data[user])
+        else:
+            with open("templates.json", "r") as f:
+                templates = json.load(f)
+            newChar = templates["Character"]
+            data[user] = newChar
+            with open("characters.json", "w") as f:
+                json.dump(data,f,indent = 4)
+                return "", "Character for " + user + " created"
+    else:
+        return "", charData(user,cmd)
+
+def lightsaberChar(user : str):
+    with open("characters.json", "r") as f:
+        data = json.load(f)
+
+    if user in data:
+        return data[user]["Lightsaber Characteristic"]
+
+def userHasChar(user : str):
+    with open("characters.json", "r") as f:
+        data = json.load(f)
+
+    return user in data
+