/
seed.py
38 lines (26 loc) · 1.15 KB
/
seed.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
"""Seed database with sample data from CSV Files."""
from csv import DictReader
from app import db
from models.quiz import Question, Quiz
db.drop_all()
db.create_all()
with open('generator/general_5.csv') as questions:
db.session.bulk_insert_mappings(Question, DictReader(questions))
with open('generator/rose_3.csv') as questions:
db.session.bulk_insert_mappings(Question, DictReader(questions))
with open('generator/aster_3.csv') as questions:
db.session.bulk_insert_mappings(Question, DictReader(questions))
with open('generator/mint_3.csv') as questions:
db.session.bulk_insert_mappings(Question, DictReader(questions))
with open('generator/pea_3.csv') as questions:
db.session.bulk_insert_mappings(Question, DictReader(questions))
questions = Question.query.all()
family = ['general']*5 + ['Rose family']*3 + ['Aster family']*3
family = family + ['Mint family']*3 + ['Pea family']*3
for i in range(len(family)):
new_quiz = Quiz(family=family[i], num_questions=10)
for j in range(10):
questions[10*i + j].family = family[i]
new_quiz.questions.append(questions[i*10 + j])
db.session.add(new_quiz)
db.session.commit()