from models import Location, Product
from app import db


menuitems = [
    'Spicy Chicken',
    'Advocado Chick',
    'Indian Summer',
    'Olive Garden',
    'Advocado Spring',
    'Spicy Mexican',
    'Beefcake',
    'Iron Man',
    'Fitalian',
    'Captain',
    'Sea Breeze',
    'Vegan Market',
    'Sunset Beach',
    'Hot Tofu',
    'Vegan Advocado Spring'
]

pricedict = {'Small': 799, 'Medium': 999, 'Large': 1199}


def add():
    simpizza = Location()
    simpizza.configure("Fitchen", "?", "?", "https://www.fitchen.be/")
    db.session.add(simpizza)

    for menuitem in menuitems:
        for size, price in pricedict.items():
            for container in ['bowl', 'wrap']:
                name = '%s %s in %s' % (size, menuitem, container)
                entry = Product()
                entry.configure(simpizza, name, price)
                db.session.add(entry)