Upload conversationDB.py

Updated DB manager methods

conversationDB.py

@@ -13,41 +13,90 @@ mysql_port = 3306  # the default MySQL port
 mysql_user = 'root'
 mysql_password = 'naP2tion'
 mysql_db = 'warbot'
+ssh_port = 22
+dbTableName = 'conversations' # messages data table
 
-# create an SSH client and load the private key
-ssh_client = paramiko.SSHClient()
-ssh_client.load_system_host_keys()
-ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-private_key = paramiko.RSAKey.from_private_key_file(ssh_key_path)
-
-# start the SSH tunnel using sshtunnel
-with SSHTunnelForwarder(
-        (ssh_host, 22),
-        ssh_username=ssh_user,
-        ssh_pkey=private_key,
-        remote_bind_address=(mysql_host, mysql_port)) as tunnel:
-    
-    # connect to the MySQL server through the SSH tunnel
-    mysql_conn = pymysql.connect(
-        host='localhost',
-        port=tunnel.local_bind_port,
-        user=mysql_user,
-        password=mysql_password,
-        db=mysql_db
-    )
-
-    # send a query to the MySQL database and print the response table
-    with mysql_conn.cursor() as cursor:
-        query = 'SELECT * FROM conversations WHERE username = "user1";'
-        cursor.execute(query)
-        rows = cursor.fetchall()
-        for row in rows:
-            print(row)
-
-    # close the MySQL connection
-    mysql_conn.close()
-
-# close the SSH client
-ssh_client.close()
-
-# query = 'SELECT * FROM conversations WHERE username = "user1";'
+class DataBase():
+    # manages mySQL connection and send-receive
+    def __init__(db, ssh_key_path = ssh_key_path, ssh_host=ssh_host, ssh_port=ssh_port, ssh_user=ssh_user,
+                 mysql_host=mysql_host,mysql_port=mysql_port,mysql_user=mysql_user,mysql_password=mysql_password,mysql_db=mysql_db):
+        # create an SSH client and load the private key
+        db.ssh_client = paramiko.SSHClient()
+        db.ssh_client.load_system_host_keys()
+        db.ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+        db.private_key = paramiko.RSAKey.from_private_key_file(ssh_key_path)
+        db.ssh_key_path = ssh_key_path
+        db.ssh_host=ssh_host
+        db.ssh_port=ssh_port
+        db.ssh_user=ssh_user
+        db.mysql_host=mysql_host
+        db.mysql_port=mysql_port
+        db.mysql_user=mysql_user
+        db.mysql_password=mysql_password
+        db.mysql_db=mysql_db
+    def getmessages(db,username = ""):
+        # start the SSH tunnel using sshtunnel
+        with SSHTunnelForwarder(
+                (db.ssh_host, db.ssh_port),
+                ssh_username=db.ssh_user,
+                ssh_pkey=db.private_key,
+                remote_bind_address=(db.mysql_host, db.mysql_port)) as tunnel:
+            # connect to the MySQL server through the SSH tunnel
+            mysql_conn = pymysql.connect(
+                host='localhost', # will be due to the SSH tunneling issue
+                port=tunnel.local_bind_port,
+                user=db.mysql_user,
+                password=db.mysql_password,
+                db=db.mysql_db
+            )
+
+            # send a query to the MySQL database and print the response table
+            with mysql_conn.cursor() as cursor:
+                query = f'SELECT * FROM {dbTableName} WHERE username = "{username}";'
+                cursor.execute(query)
+                rows = cursor.fetchall()
+
+                messages = [message[2] for message in rows]
+
+            # close the MySQL connection
+            mysql_conn.close()
+
+        # close the SSH client
+        db.ssh_client.close()
+        return messages
+
+    def setmessages(db,username, message_text):
+        # start the SSH tunnel using sshtunnel
+        with SSHTunnelForwarder(
+                (db.ssh_host, db.ssh_port),
+                ssh_username=db.ssh_user,
+                ssh_pkey=db.private_key,
+                remote_bind_address=(db.mysql_host, db.mysql_port)) as tunnel:
+            # connect to the MySQL server through the SSH tunnel
+            mysql_conn = pymysql.connect(
+                host='localhost', # will be due to the SSH tunneling issue
+                port=tunnel.local_bind_port,
+                user=db.mysql_user,
+                password=db.mysql_password,
+                db=db.mysql_db
+            )
+
+            # send a query to the MySQL database and print the response table
+            with mysql_conn.cursor() as cursor:
+                query = f"INSERT INTO {dbTableName} (username,message_text) VALUES ('{username}','{message_text}');"
+                cursor.execute(query)
+                mysql_conn.commit()
+
+            # close the MySQL connection
+            mysql_conn.close()
+
+        # close the SSH client
+        db.ssh_client.close()
+
+if __name__ == '__main__':
+    username = 'user2'
+
+    db = DataBase()
+    db.setmessages(username='user2',message_text='some message')
+    messages = db.getmessages(username)
+    print(messages)