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()
