-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathseed.py
40 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
39
"""Load children data into database."""
from model import Child, Location, connect_to_db, db
from server import app
from crud import create_user
import os
import crud
import model
import server
os.system('dropdb children')
os.system('createdb children')
model.connect_to_db(server.app)
model.db.create_all()
def get_children():
"""Load children from dataset into database."""
with open("data/finaldata.csv", encoding='utf-8-sig') as children_data:
for r in enumerate(children_data):
data = r[1].split(',')
child_id, age_2021, date_missing, lname, fname, missing_age, city, county, state, gender, ethnicity, latitude, longitude = data
child_instance = crud.create_child(fname=fname, lname=lname, ethnicity=ethnicity, date_missing=date_missing, missing_age=int(missing_age), age_2021=int(age_2021))
location_instance = crud.create_location(child_id=child_instance.child_id, state=state, city=city, county=county)
# print(location_instance)
# print(location_instance.child_id)
if __name__ == '__main__':
connect_to_db(app)
db.create_all()
get_children()