iLO-manager first main commitgit status

backend/database.py

@@ -0,0 +1,86 @@
+import sqlite3
+import bcrypt
+
+def initialize_db():
+    conn = sqlite3.connect('servers.db')
+    cursor = conn.cursor()
+
+    # Create table for servers
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS servers (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            ip_address TEXT NOT NULL,
+            name TEXT NOT NULL
+        )
+    ''')
+
+    # Create table for users
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT NOT NULL UNIQUE,
+            password TEXT NOT NULL
+        )
+    ''')
+
+    # Create table for actions
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS actions (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            name TEXT NOT NULL UNIQUE,
+            endpoint TEXT NOT NULL
+        )
+    ''')
+
+    conn.commit()
+    conn.close()
+
+def add_user(username, password):
+    hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+    conn = sqlite3.connect('servers.db')
+    cursor = conn.cursor()
+    cursor.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, hashed_password))
+    conn.commit()
+    conn.close()
+
+def check_user(username, password):
+    conn = sqlite3.connect('servers.db')
+    cursor = conn.cursor()
+    cursor.execute("SELECT password FROM users WHERE username = ?", (username,))
+    user = cursor.fetchone()
+
+    if user and bcrypt.checkpw(password.encode('utf-8'), user[0].encode('utf-8')):
+        return True
+    return False
+
+def add_server(ip_address, name):
+    conn = sqlite3.connect('servers.db')
+    cursor = conn.cursor()
+    cursor.execute("INSERT INTO servers (ip_address, name) VALUES (?, ?)", (ip_address, name))
+    conn.commit()
+    conn.close()
+
+def get_servers():
+    conn = sqlite3.connect('servers.db')
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM servers")
+    data = cursor.fetchall()
+    conn.close()
+    return data
+
+def add_action(name, endpoint):
+    conn = sqlite3.connect('servers.db')
+    cursor = conn.cursor()
+    cursor.execute("INSERT INTO actions (name, endpoint) VALUES (?, ?)", (name, endpoint))
+    conn.commit()
+    conn.close()
+
+
+def get_actions():
+    conn = sqlite3.connect('servers.db')
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM actions")
+    data = cursor.fetchall()
+    conn.close()
+    return data
+