def create_data_course(): role_teacher = Role.objects(name='teacher').first() list_teacher = User.objects(role=role_teacher) role_student = Role.objects(name='student').first() list_student = User.objects(role=role_student) list_shift = Shift.objects.all() Course(id_course="SD-001", name="Word - Excel - Power Point", tuition=120, teacher=list_teacher[0], shift=list_shift[0]).save() Course(id_course="DB-001", name="DataBase System", tuition=199, teacher=list_teacher[1], shift=list_shift[1]).save() Course(id_course="WEB-001", name="Web Interface", tuition=149, teacher=list_teacher[2], shift=list_shift[2]).save() Course(id_course="JAVA-001", name="Java Application", tuition=299, teacher=list_teacher[3], shift=list_shift[0]).save() Course(id_course="RJ-001", name="React - Redux JavaScript", tuition=289, teacher=list_teacher[4], shift=list_shift[2]).save()
def create(): activate = list(range(10)) activate[3] = "active" if current_user.is_authenticated: form = CreateCourseForm() if form.validate_on_submit(): id_course = form.id_course.data course = Course.objects(id_course=id_course).first() if course is None: name = form.name.data start_date = form.start_date.data finish_date = form.finish_date.data tuition = float(form.tuition.data) shift = Shift.objects(name=form.shift.data).first() role_teacher = Role.objects(name='teacher').first() list_teacher = User.objects(role=role_teacher, is_activate=True) for item in list_teacher: if item.first_name + ' ' + item.last_name == form.teacher.data: teacher = item break course = Course(id_course=id_course, name=name, start_date=start_date, finish_date=finish_date, tuition=tuition, shift=shift, teacher=teacher) course.save() flash('Course has been created!', 'success') return redirect(url_for('educate.educate_info', id=course.id)) flash('Course is exist!, please check again', 'danger') return redirect(url_for('educate.create')) return render_template('educate_create.html', title='Educate Create Course', activate=activate, form=form) return redirect(url_for('users.login'))