start_day redirects to students if none in the DB

anthony-becerra · Sep 5, 2017 · eccaeb2 · eccaeb2
1 parent 8cfc78a
commit eccaeb2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -152,6 +152,9 @@ def start_day():
     students_att = Attendance.query.filter_by(date_now = date.today()).all()
 
     # checks if not attendace list has been created for the day.
+    if not students:
+        flash('No students in the database yet', 'error')
+        return redirect('/students')
     if not students_att:
         if students:
         # pushes all students into the attendance table, creating