Rewrite get_data, unnecessary Python data processing

This commit is contained in:
Abdulkadir Furkan Şanlı 2019-01-15 12:43:51 +01:00
parent cab6fdff4f
commit 201d2d4a65
No known key found for this signature in database
GPG Key ID: 7823BD18E6F95D73

View File

@ -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)