def delete_student():
form = DeleteForm()
if request.method == 'POST' and form.validate():
student = Students(idNum=form.id_number.data)
if student.exist():
student.delete()
flash("Deleted student with id number '{}'".format(
form.id_number.data), category='success')
return redirect(url_for('students'))
else:
flash("Student Does not exist.", category='danger')
return redirect(url_for('delete_student'))
else:
return render_template('delete_student.html', form=form)
def add_student():
form = StudentForm()
if request.method == 'POST' and form.validate():
student = Students(form.firstname.data.title(), form.lastname.data.title(),
form.gender.data, form.id_number.data, form.course.data)
if student.exist() is False:
student.add()
flash("Student Added Successfully with ID number '{}'".format(
form.id_number.data), category='success')
return redirect(url_for('students'))
else:
flash("Student Already Exist!", category='danger')
return redirect(url_for('add_student'))
else:
return render_template('add_student.html', form=form)
def edit_student(student_id):
form = StudentForm()
db = Students(idNum=student_id)
student = db.search('idNum', student_id)
if request.method == 'POST' and form.validate():
newId = Students(idNum=form.id_number.data)
if(newId.exist() and student_id != form.id_number.data):
flash('ID number already taken please choose another ID NUMBER',
category='danger')
return redirect(url_for('edit_student', student_id=db.idNum))
else:
db.update(form.firstname.data, form.lastname.data,
form.gender.data, form.id_number.data, form.course.data)
flash('Successfully Edited Student', category='success')
return redirect(url_for('students'))
else:
for attribute in student:
form.firstname.data = attribute[0]
form.lastname.data = attribute[1]
form.gender.data = attribute[2]
form.id_number.data = attribute[3]
form.course.data = attribute[4]
return render_template('edit_student.html', form=form)