diff --git a/ib-clearance.py b/ib-clearance.py index f41528e..3168767 100755 --- a/ib-clearance.py +++ b/ib-clearance.py @@ -49,16 +49,20 @@ def get_data(conn): id = int(input("\nPlease enter student ID: ")) cur = conn.cursor() - cur.execute("SELECT id, student_class, other FROM students") - ids = cur.fetchall() # ids is list of tuples. + cur.execute(""" + SELECT id, student_class, other + FROM students + WHERE id = ? + """, (id,)) + data = cur.fetchone() - for tup in ids: - if id == tup[0]: - lis = list(tup) - lis[2] = lis[2].split() - return lis + if data is not None: + data = list(data) # SQLite returns tuples. + data[2] = data[2].split() else: raise ValueError + + return data except ValueError: print("\nInvalid ID, try again.") return get_data(conn)