Update database.py

database.py

@@ -44,4 +44,36 @@ def get_user_role(email):
     cursor.execute("SELECT role FROM users WHERE email=?", (email,))
     row = cursor.fetchone()
     conn.close()
-    return row[0] if row else None
+    return row[0] if row else None
+
+def get_sessions(class_id):
+    conn = get_conn()
+    cursor = conn.cursor()
+    cursor.execute("SELECT id, date, is_open FROM sessions WHERE class_id=? ORDER BY date DESC", (class_id,))
+    rows = cursor.fetchall()
+    conn.close()
+    return rows
+
+def toggle_session(class_id, is_open):
+    from datetime import datetime
+    date = datetime.now().strftime("%Y-%m-%d %H:%M")
+    conn = get_conn()
+    cursor = conn.cursor()
+    cursor.execute("INSERT INTO sessions (class_id, date, is_open) VALUES (?, ?, ?)", 
+                   (class_id, date, int(is_open)))
+    conn.commit()
+    conn.close()
+    return f"Session {'opened' if is_open else 'closed'} for {class_id} on {date}"
+
+def add_student(name, email, password, class_id):
+    conn = get_conn()
+    cursor = conn.cursor()
+    try:
+        cursor.execute("INSERT INTO users (name, email, password, role, class_id) VALUES (?, ?, ?, 'student', ?)", 
+                       (name, email, password, class_id))
+        conn.commit()
+    except sqlite3.IntegrityError:
+        return "Email already exists."
+    finally:
+        conn.close()
+    return "Student added successfully."