1 files changed, 36 insertions, 0 deletions
diff --git a/Utilities/db_queries.py b/Utilities/db_queries.py
new file mode 100644
index 0000000..db4a08e
--- /dev/null
+++ b/Utilities/db_queries.py
@@ -0,0 +1,36 @@
+import sqlite3
+
+
+def update_db(use_latest, ip, port, module_ports):
+  conn = sqlite3.connect('Settings.db')
+  c = conn.cursor()
+
+  if use_latest is False:
+    try:
+      c.execute("""CREATE TABLE settings (
+        ID INTEGER PRIMARY KEY AUTOINCREMENT,
+        ip TEXT,
+        port INTEGER,
+        module_ports TEXT
+      )""")
+      conn.commit()
+    except:
+      c.execute("UPDATE settings SET ip=?, port=?, module_ports=? WHERE ID=1;", (ip, port, module_ports))
+      conn.commit()
+    else:
+      c.execute("INSERT INTO settings (ip, port, module_ports) VALUES (?, ?, ?);", (ip, port, module_ports))
+      conn.commit()
+  
+  conn.close()
+
+
+def get_module_data():
+  conn = sqlite3.connect('Settings.db')
+  c = conn.cursor()
+
+  c.execute("SELECT * FROM settings WHERE ID=1")
+  conn.commit()
+  db_data = c.fetchone()
+  conn.close()
+
+  return db_data
+\ No newline at end of file