Added custom prefix & info

core/database.py

@@ -7,6 +7,7 @@ class Servers:
     def __init__(self, server=None):
         self.server = server
         self.col = client["BytesBump"]["servers"]
+        self.prefixes = client["BytesBump"]["prefixes"]
 
     def get(self):
         return self.col.find_one({"_id":self.server})
@@ -43,4 +44,26 @@ class Servers:
             self.col.delete_one({'_id': self.server})
         else:
             self.col.delete_one(checks)
-    
+
+    """
+    Prefix management
+    """
+
+    @property
+    def hasPrefix(self):
+        r = self.prefixes.find_one({'_id':self.server})
+        if r: return True
+        return False
+
+    @property
+    def deletePrefix(self):
+        self.prefixes.delete_one({'_id':self.server})
+
+    def setPrefix(self, prefix):
+        if self.hasPrefix: self.prefixes.update_one({'_id':self.server}, {'$set':{'prefix':prefix}})
+        else: self.prefixes.insert_one({'_id':self.server, 'prefix':prefix})
+
+    def getPrefix(self):
+        prefix = self.prefixes.find_one({'_id':self.server})
+        if prefix: return prefix["prefix"]
+        else: return Data("config").yaml_read()["prefix"]