summaryrefslogtreecommitdiff
path: root/Utilities/db_queries.py
blob: db4a08e0caffd4628a4fa0b53845638a08845ce5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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