ADD sshtunnel to mysql_connect.py

2023-07-25 13:01:34 +02:00
parent 0d7b81a3eb
commit 307add1530
13 changed files with 506 additions and 373 deletions
@@ -10,36 +10,47 @@ server = SSHTunnelForwarder(
 )
 server.start()
 print(server.local_bind_port)
-mydb = mysql.connector.connect(
-    host="127.0.0.1",
-    port=server.local_bind_port,
-    user="root",
-    password="N53yBCswuawzBzS445VNAhWVMs3N59Gb9szEsrzXRBzarDqpdETpQeyt5v5CGe",
-    database="Stines-GmbH"
-)
-def get_ip(query):
+def database(query,name,user):
+    mydb = mysql.connector.connect(
+        host="127.0.0.1",
+        port=server.local_bind_port,
+        user="root",
+        password="N53yBCswuawzBzS445VNAhWVMs3N59Gb9szEsrzXRBzarDqpdETpQeyt5v5CGe",
+        database="" + name
+    )
+    mydb.connect()
+    cursor = mydb.cursor()
+    if user:
+        cursor.execute(query,user)
+    else:
+        cursor.execute(query)
+    if "INSERT" in query:
+        mydb.commit()
+    if "SELECT" in query:
+        return cursor.fetchall()
+    if "SHOW" in query :
+        return cursor.fetchall()
+    # mydb.close()
+def get_ip(query,name):
    print("GetIP")
-    mydb.connect()
-    cursor = mydb.cursor()
-    cursor.execute(query)
-    mydb.close()
-    return cursor.fetchall()
+    user = ""
+    return database(query, name,user)
 def get_database():
-    cursor = mydb.cursor()
-    cursor.execute('SHOW DATABASES')
-    return cursor.fetchall()
-def create_database(query):
-    print(query)
-    cursor = mydb.cursor()
-    cursor.execute(query)
-    mydb.commit()
-    mydb.close()
-    return ("Database Created")
-def add_user(query, user):
-    print("add_user")
-    mydb.connect()
-    cursor = mydb.cursor()
-    cursor.execute(query, user)
-    mydb.commit()
-    mydb.close()
-    return ("user Addet")
+    name = ""
+    user = ""
+    query = 'SHOW DATABASES'
+    print(database(query,name,user))
+    return database(query,name,user)
+def create_database(query,name):
+    try:
+        user = ""
+        database(query,name,user)
+        return ("Database Created")
+    except Exception as e:
+        print("Ein Fehler ist aufgetreten:", str(e))
+def add_user(query,name,user):
+    try:
+        database(query,name,user)
+        return ("user Addet")
+    except Exception as e:
+        print("Ein Fehler ist aufgetreten:", str(e))