Example #1
0
def add_schedule2(class_id):
    if current_user.admin:
        form = ScheduleForm()
        form.monday1.choices = DB.subject_list(class_id)
        form.monday2.choices = DB.subject_list(class_id)
        form.monday3.choices = DB.subject_list(class_id)
        form.monday4.choices = DB.subject_list(class_id)
        form.monday5.choices = DB.subject_list(class_id)
        form.monday6.choices = DB.subject_list(class_id)

        form.tuesday1.choices = DB.subject_list(class_id)
        form.tuesday2.choices = DB.subject_list(class_id)
        form.tuesday3.choices = DB.subject_list(class_id)
        form.tuesday4.choices = DB.subject_list(class_id)
        form.tuesday5.choices = DB.subject_list(class_id)
        form.tuesday6.choices = DB.subject_list(class_id)

        form.wednesday1.choices = DB.subject_list(class_id)
        form.wednesday2.choices = DB.subject_list(class_id)
        form.wednesday3.choices = DB.subject_list(class_id)
        form.wednesday4.choices = DB.subject_list(class_id)
        form.wednesday5.choices = DB.subject_list(class_id)
        form.wednesday6.choices = DB.subject_list(class_id)

        form.thursday1.choices = DB.subject_list(class_id)
        form.thursday2.choices = DB.subject_list(class_id)
        form.thursday3.choices = DB.subject_list(class_id)
        form.thursday4.choices = DB.subject_list(class_id)
        form.thursday5.choices = DB.subject_list(class_id)
        form.thursday6.choices = DB.subject_list(class_id)

        form.friday1.choices = DB.subject_list(class_id)
        form.friday2.choices = DB.subject_list(class_id)
        form.friday3.choices = DB.subject_list(class_id)
        form.friday4.choices = DB.subject_list(class_id)
        form.friday5.choices = DB.subject_list(class_id)
        form.friday6.choices = DB.subject_list(class_id)

        form.saturday1.choices = DB.subject_list(class_id)
        form.saturday2.choices = DB.subject_list(class_id)
        form.saturday3.choices = DB.subject_list(class_id)
        form.saturday4.choices = DB.subject_list(class_id)
        form.saturday5.choices = DB.subject_list(class_id)
        form.saturday6.choices = DB.subject_list(class_id)
        schedule_id = DB.get_schedule_id(class_id)
        if form.submit.data:
            DB.add_schedule_days(schedule_id, "monday", form.monday1.data, 1)
            DB.add_schedule_days(schedule_id, "monday", form.monday2.data, 2)
            DB.add_schedule_days(schedule_id, "monday", form.monday3.data, 3)
            DB.add_schedule_days(schedule_id, "monday", form.monday4.data, 4)
            DB.add_schedule_days(schedule_id, "monday", form.monday5.data, 5)
            DB.add_schedule_days(schedule_id, "monday", form.monday6.data, 6)

            DB.add_schedule_days(schedule_id, "tuesday", form.tuesday1.data, 1)
            DB.add_schedule_days(schedule_id, "tuesday", form.tuesday2.data, 2)
            DB.add_schedule_days(schedule_id, "tuesday", form.tuesday3.data, 3)
            DB.add_schedule_days(schedule_id, "tuesday", form.tuesday4.data, 4)
            DB.add_schedule_days(schedule_id, "tuesday", form.tuesday5.data, 5)
            DB.add_schedule_days(schedule_id, "tuesday", form.tuesday6.data, 6)

            DB.add_schedule_days(schedule_id, "wednesday",
                                 form.wednesday1.data, 1)
            DB.add_schedule_days(schedule_id, "wednesday",
                                 form.wednesday2.data, 2)
            DB.add_schedule_days(schedule_id, "wednesday",
                                 form.wednesday3.data, 3)
            DB.add_schedule_days(schedule_id, "wednesday",
                                 form.wednesday4.data, 4)
            DB.add_schedule_days(schedule_id, "wednesday",
                                 form.wednesday5.data, 5)
            DB.add_schedule_days(schedule_id, "wednesday",
                                 form.wednesday6.data, 6)

            DB.add_schedule_days(schedule_id, "thursday", form.thursday1.data,
                                 1)
            DB.add_schedule_days(schedule_id, "thursday", form.thursday2.data,
                                 2)
            DB.add_schedule_days(schedule_id, "thursday", form.thursday3.data,
                                 3)
            DB.add_schedule_days(schedule_id, "thursday", form.thursday4.data,
                                 4)
            DB.add_schedule_days(schedule_id, "thursday", form.thursday5.data,
                                 5)
            DB.add_schedule_days(schedule_id, "thursday", form.thursday6.data,
                                 6)

            DB.add_schedule_days(schedule_id, "friday", form.friday1.data, 1)
            DB.add_schedule_days(schedule_id, "friday", form.friday2.data, 2)
            DB.add_schedule_days(schedule_id, "friday", form.friday3.data, 3)
            DB.add_schedule_days(schedule_id, "friday", form.friday4.data, 4)
            DB.add_schedule_days(schedule_id, "friday", form.friday5.data, 5)
            DB.add_schedule_days(schedule_id, "friday", form.friday6.data, 6)

            DB.add_schedule_days(schedule_id, "saturday", form.saturday1.data,
                                 1)
            DB.add_schedule_days(schedule_id, "saturday", form.saturday2.data,
                                 2)
            DB.add_schedule_days(schedule_id, "saturday", form.saturday3.data,
                                 3)
            DB.add_schedule_days(schedule_id, "saturday", form.saturday4.data,
                                 4)
            DB.add_schedule_days(schedule_id, "saturday", form.saturday5.data,
                                 5)
            DB.add_schedule_days(schedule_id, "saturday", form.saturday6.data,
                                 6)
            return redirect(url_for("index"))
        return render_template("add_schedule2.html", form=form)
    else:
        return redirect(url_for("index"))