Start work on setup

Author: damianpostaru

client/setup.py

@@ -0,0 +1,25 @@
+from app import db
+from models.dessert import Dessert
+from models.drink import Drink
+from models.mysql_model import save_new_pizza, save_new_drink, save_new_dessert
+from models.pizza import Pizza
+
+
+def setup():
+    emmentaler = {"name": "emmentaler", "vegetarian": True, "price": 1}
+    mozzarella = {"name": "mozzarella", "vegetarian": True, "price": 1}
+    gorgonzola = {"name": "gorgonzola", "vegetarian": True, "price": 1}
+    goat_cheese = {"name": "goat_cheese", "vegetarian": True, "price": 1.50}
+
+    if not db.session.query(Pizza).first():
+        four_cheese = save_new_pizza("Four Cheese", [emmentaler, mozzarella, gorgonzola, goat_cheese])
+
+    if not db.session.query(Drink).first():
+        cola = save_new_drink("Cola", 1.99)
+        sprite = save_new_drink("Sprite", 1.99)
+        fanta = save_new_drink("Fanta", 1.99)
+        beer = save_new_drink("Beer", 3.99)
+
+    if not db.session.query(Dessert).first():
+        chocolate_lavacake = save_new_dessert("Chocolate Lavacake", 2.99)
+        chocolate_chip_cookie = save_new_dessert("Chocolate Chip Cookie", 1.99)

main.py

@@ -2,8 +2,10 @@
 from flask import make_response, render_template
 from flask import request
 
+from client.setup import setup
 from models.mysql_model import *
 
+setup()
 
 @app.route("/test")
 def get_test():

models/mysql_model.py

@@ -1,19 +1,19 @@
 from datetime import datetime, timedelta
 
-from flask_apscheduler import APScheduler
 from apscheduler.triggers.date import DateTrigger
+from flask_apscheduler import APScheduler
 
 from app import db, app
-from models.ingredient import Ingredient
-from models.pizza import Pizza
+from models.address import Address
+from models.customer import Customer
 from models.dessert import Dessert
 from models.drink import Drink
-from models.menu_item import MenuItem
 from models.driver import Driver
-from models.address import Address
-from models.customer import Customer
+from models.ingredient import Ingredient
+from models.menu_item import MenuItem
 from models.order import Order
 from models.order_item import OrderItem
+from models.pizza import Pizza
 
 scheduler = APScheduler()
 scheduler.init_app(app)
@@ -22,7 +22,7 @@
 
 def save_new_pizza(name, ingredients):
     ingredients_list = []
-    calculated_price = 1.0
+    calculated_price = 0
     veg_flag = True
     for ingredient in ingredients:
         calculated_price = calculated_price + ingredient["price"]
@@ -34,6 +34,7 @@ def save_new_pizza(name, ingredients):
         else:
             ingredients_list.append(
                 Ingredient(name=ingredient["name"], vegetarian=ingredient["vegetarian"], price=ingredient["price"]))
+    calculated_price = calculated_price + 0.4 * calculated_price
     new_pizza = Pizza(name=name, vegetarian=veg_flag, price=calculated_price, ingredients=ingredients_list)
     db.session.add(new_pizza)
     db.session.commit()