AI-Student-Performance/database.py at main · anuskagupta123/AI-Student-Performance · GitHub

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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import sqlite3

def create_db():
    conn = sqlite3.connect("student.db")
    c = conn.cursor()

    c.execute("""
    CREATE TABLE IF NOT EXISTS users(
        username TEXT,
        password TEXT
    )
    """)

    c.execute("""
    CREATE TABLE IF NOT EXISTS predictions(
        username TEXT,
        reading INTEGER,
        writing INTEGER,
        predicted REAL
    )
    """)

    conn.commit()
    conn.close()


def add_user(username,password):
    conn = sqlite3.connect("student.db")
    c = conn.cursor()

    c.execute("INSERT INTO users VALUES (?,?)",(username,password))

    conn.commit()
    conn.close()


def login_user(username,password):
    conn = sqlite3.connect("student.db")
    c = conn.cursor()

    c.execute("SELECT * FROM users WHERE username=? AND password=?",(username,password))

    data = c.fetchall()

    conn.close()

    return data


def save_prediction(username,reading,writing,predicted):
    conn = sqlite3.connect("student.db")
    c = conn.cursor()

    c.execute("INSERT INTO predictions VALUES (?,?,?,?)",
              (username,reading,writing,predicted))

    conn.commit()
    conn.close()