-
Notifications
You must be signed in to change notification settings - Fork 2
/
seed_db.py
33 lines (26 loc) · 839 Bytes
/
seed_db.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
"""Script to seed database"""
# Import os and libraries
import os
import json
import csv
# Import dabase model, crud, and server
import crud
import model
import server
# Drop and create db
os.system('dropdb recipes')
os.system('createdb recipes')
# Connect db to Flask app
model.connect_to_db(server.app)
# Create tables from classes inherited from db.model
model.db.create_all()
# read and open ingredients csv
file = open('data/top-1k-ingredients.csv', newline='')
data = csv.reader(file, delimiter=';')
# each row is a list of 2, name and id
ingredients_data = {row[1]:row[0] for row in data}
# make id as key and name as value
# loop over ingredients dictionary
for ingredient in ingredients_data:
# create an ingredient and save to db
crud.create_ingredient(ingredient_id=ingredient, name=ingredients_data[ingredient])