Rewrite get_data, unnecessary Python data processing
This commit is contained in:
parent
cab6fdff4f
commit
201d2d4a65
@ -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)
|
||||
|
Reference in New Issue
Block a user