-
Notifications
You must be signed in to change notification settings - Fork 0
/
manager.py
129 lines (87 loc) · 2.29 KB
/
manager.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
# coding: utf-8
from flask_script import Manager
from flask_migrate import Migrate, MigrateCommand
from app import create_app
from app.models import db
from app.models.student import Student
from app.models.company import Company
from app.models.skill import Skill
app = create_app()
manager = Manager(app)
migrate = Migrate(app, db)
manager.add_command("db", MigrateCommand)
@manager.shell
def make_shell_context():
return dict(
app=app,
db=db,
# add models
Student=Student,
Company=Company,
Skill=Skill
)
@manager.option("-p", "--populate", default="Populating")
def populate(populate):
"""
Populate the database
"""
skills = ["Javascript", "HTML", "CSS", "SASS", "PHP", "Python"]
for name in skills:
skill = Skill(name=name)
db.session.add(skill)
student_1 = Student(
name="Lucia",
last_name="Cardenas",
age=18,
email="lucia.c@gmail.com"
)
student_2 = Student(
name="Maria Gracia",
last_name="Silva",
age=22,
email="mg.s@hotmail.com"
)
db.session.add_all([student_1, student_2])
company = Company(
name="Athelas",
address="Recavarren esquina con pardo",
phone="+51961738608",
website="https://www.athelas.pe"
)
db.session.add(company)
company2 = Company(
name=u"Lucuma Labs",
address="Cerca al faro de miraflores",
phone="+511 681 0041",
website="https://lucumalabs.com/"
)
db.session.add(company2)
company3 = Company(
name=u"Codepicnic",
address="Por todo el mundo",
phone="+48652689522",
website="https://codepicnic.com/"
)
db.session.add(company3)
db.session.commit()
skills = db.session.query(Skill).all()
company.skills.append(skills[0])
company.skills.append(skills[1])
company.skills.append(skills[2])
company.skills.append(skills[3])
company2.skills.append(skills[4])
company2.skills.append(skills[5])
company2.skills.append(skills[1])
company3.skills.append(skills[3])
company3.skills.append(skills[5])
company3.skills.append(skills[1])
student_1.skills.append(skills[0])
student_1.skills.append(skills[1])
student_1.skills.append(skills[2])
student_2.skills.append(skills[0])
student_2.skills.append(skills[1])
student_2.skills.append(skills[2])
student_2.skills.append(skills[3])
db.session.commit()
if __name__ == '__main__':
manager.run()