Some changes

data/DatabaseConnection.py

@@ -0,0 +1,100 @@
+import mysql.connector
+
+class WrongMinecraftName(Exception):
+    pass
+class SQLInsertError(Exception):
+    pass
+class UnLinkError(Exception):
+    pass
+class PlayerNotLinked(Exception):
+    pass
+
+class DatabaseConnection:
+    # databases -> worldcraft and worldcraft_discord
+    def __init__(self, database="worldcraft_discord"):
+        if database == "worldcraft_discord":
+            self.mydb = mysql.connector.connect(
+                host="192.168.1.251",
+                user="worldcraft_discord",
+                password="aquev5vcwhLwTdRt",
+                database=database
+            )
+        elif database == "s13_worldcraft":
+            self.mydb = mysql.connector.connect(
+                host="192.168.1.251",
+                user="u13_f2zpWZpIKY",
+                password="@fZQ6Uu3+U^WH1i2JNemgTC7",
+                database=database
+            )
+
+        self.cursor = self.mydb.cursor()
+
+    def get_db(self):
+        return self.mydb
+
+    def get_cursor(self):
+        return self.cursor
+
+    def close(self):
+        self.mydb.close()
+
+# this will save the uuid of the player and discord id
+class PlayerDBLinker:
+    def __init__(self):
+        self.tablename = "playerlinks"
+        self.discorddbconn = DatabaseConnection(database="worldcraft_discord")
+        self.serverdbconn = DatabaseConnection(database="s13_worldcraft")
+        # check if table exists
+        cursor = self.discorddbconn.get_cursor()
+        cursor.execute("create table IF NOT EXISTS {} (id int NOT NULL PRIMARY KEY AUTO_INCREMENT, minecraftUUID varchar(36) NOT NULL, discordid TINYTEXT NOT NULL)".format(self.tablename))
+
+    def checkPlayerLink(self, minecraftUUID, discordid):
+        cursor = self.discorddbconn.get_cursor()
+        cursor.execute("select count(*) from {} where minecraftUUID='{}' AND discordid='{}'".format(self.tablename, minecraftUUID, discordid))
+        res = cursor.fetchone()
+        return res[0] == 1
+
+    def minecraftUUIDused(self, minecraftUUID):
+        cursor = self.discorddbconn.get_cursor()
+        cursor.execute("select count(*) from {} where minecraftUUID='{}'".format(self.tablename, minecraftUUID))
+        res = cursor.fetchone()
+        return res[0] >= 1
+
+    def discordidused(self, discordid):
+        cursor = self.discorddbconn.get_cursor()
+        cursor.execute("select count(*) from {} where discordid='{}'".format(self.tablename, discordid))
+        res = cursor.fetchone()
+        return res[0] >= 1
+
+    def linkPlayer(self, minecraftUUID, discordid):
+        cursor = self.discorddbconn.get_cursor()
+        sql = "insert into {} (minecraftUUID, discordid) VALUES (%s, %s)".format(self.tablename)
+        val = (minecraftUUID, discordid)
+
+        try:
+            cursor.execute(sql, val)
+            self.discorddbconn.get_db().commit()
+        except:
+            raise SQLInsertError()
+
+
+    def unlinkPlayer(self, minecraftname, discordid):
+        # get uuid from server database -> check if in linkerdatabase
+        serverdb_cursor = self.serverdbconn.get_cursor()
+        discorddb_cursor = self.discorddbconn.get_cursor()
+        sql = "SELECT playerUUID FROM playerprofiles WHERE playerName = '{}' LIMIT 1".format(minecraftname)
+        serverdb_cursor.execute(sql)
+        res = serverdb_cursor.fetchone()
+        if res == None:
+            raise WrongMinecraftName()
+        playeruuid = res[0]
+        if self.checkPlayerLink(playeruuid, discordid):
+            sql2 = f"DELETE FROM {self.tablename} WHERE minecraftUUID = '{playeruuid}' AND discordid = '{discordid}'"
+            discorddb_cursor.execute(sql2)
+            res2 = self.discorddbconn.get_db().commit()
+        else:
+            raise PlayerNotLinked()
+
+    def closeconnections(self):
+        self.discorddbconn.close()
+        self.serverdbconn.close()