def add_student(): form = AddStudentForm() if form.validate_on_submit(): teacher = Teacher.query.filter_by(name=form.teacher_name.data).first() if teacher is None: abort(404) student = Student() student.name = form.name.data student.teacher = teacher student.age = form.age.data db.session.add(student) db.session.commit() return redirect(url_for('.students')) return render_template('add_student.html', form=form)
def setUp(self): self.app = create_app('test') self.app_context = self.app.app_context() self.app_context.push() db.create_all() t = Teacher() t.name = 'zhangliu' t.age = 35 db.session.add(t) db.session.commit() s = Student() s.name = 'wangwu' s.age = 20 s.teacher = t db.session.add(s) db.session.commit()
def handle(self, *args, **options): faker = Faker() for _ in range(options['len']): book = Book() book.title = uuid.uuid4() book.save() subject, _ = Subject.objects.get_or_create(title='HTML') student = Student() student.name = faker.first_name() student.surname = faker.last_name() student.age = faker.random_number(digits=None) student.address = faker.address() student.birthday = faker.date() student.description = faker.text() student.email = faker.email() student.book = book student.subject = subject student.save() teacher, _ = Teacher.objects.get_or_create(name=faker.name()) teacher.students.add(student) teacher.save()