Rewrite get_data, unnecessary Python data processing
This commit is contained in:
		@@ -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