processorder4.cgi

#!/usr/local/bin/python3

import cgi
from pizza_kitchen_helpers import *

def process_form(form_data):
    quantity = int(form_data.getvalue("quantity"))
    pizzas = []
    for each in range(quantity):
        pizza = form_data.getvalue("pizza{0}".format(each))
        size = form_data.getvalue("size{0}".format(each))
        num = form_data.getvalue("quantity{0}".format(each))
        pizzas.append((pizza,size,num))
    date = form_data.getvalue("date")
    time = form_data.getvalue("time")
    customer = form_data.getvalue("customer")
    return quantity, pizzas, date, time, customer
    
def add_new_order(db,cursor,quantity, pizzas, date, time, customer):
    sql = """insert into customer_order(date,delivery_time,customer_id) values
            ('{0}','{0} {1}:00','{2}')""".format(date,time,customer)
    cursor.execute(sql)
    db.commit()
    
    sql = """select order_id from customer_order where customer_id='{0}' order by order_id desc limit 1""".format(customer)
    cursor.execute(sql)
    order_id = cursor.fetchone()
    order_id = order_id[0]
    
    for each in pizzas:
        sql = """insert into order_item(order_id,pizza_id,inches,quantity) values
                ('{0}','{1}','{2}','{3}')""".format(order_id,each[0],each[1],each[2])
        cursor.execute(sql)
    db.commit()
    
    return order_id
    
def get_customer_details(db,cursor,customer_id):
    sql = "select * from customer where customer_id='{0}'".format(customer_id)
    cursor.execute(sql)
    customer_details = cursor.fetchone()
    return customer_details
        
def create_confirmation(quantity, pizzas, cost, date, time, customer_details):
    confirmation = "<p>{0} {1} you are ordering:</p>".format(customer_details[1],customer_details[2])
    confirmation += """
                    <table>
                        <tr>
                            <th>Pizza</th>
                            <th>Size</th>
                            <th>Quantity</th>
                            <th>Cost</th>
                        </tr>
                    """
                    
    for each in pizzas:
        confirmation += """
                        <tr>
                            <td>{0}</td>
                            <td>{1}</td>
                            <td>{2}</td>
                            <td>{3}</td>
                        </tr>
                        """.format(each[0],each[1],each[2],each[3])
    
    confirmation += """
                    <tr>
                        <td></td>
                        <td></td>
                        <th>Total Cost</th>
                        <td>{0}</td>
                    </tr>
                    """.format(cost)
    
    confirmation += "</table>"
    
    confirmation += """
                    <p>
                    To be delivered at {0} on {1} to:
                    <ul>
                        <li>{2}</li>
                        <li>{3}</li>
                        <li>{4}</li>
                    </ul>
                    </p>
                    """.format(time, date, customer_details[3],customer_details[4],customer_details[5])
    return confirmation

def get_pizza_id(db,cursor,pizzaname):
    sql = "select pizza_id from pizza where name='{0}'".format(pizzaname)
    cursor.execute(sql)
    name = cursor.fetchone()
    return name[0]
    
def create_form(db, cursor, quantity, pizzas, date, time, customer_id):
    form = """
            <form method="post" action="processorder4.cgi" />
            <input type="hidden" name="customer" value="{0}"/>
            <input type="hidden" name="date" value="{1}"/>
            <input type="hidden" name="time" value="{2}"/>
            <input type="hidden" name="quantity" value="{3}"/>
            """.format(customer_id,date,time,quantity)
    
    orderitem = 0
    for each in pizzas:
        form += """
                <td><input type="hidden" name="pizza{0}" value="{1}" readonly /></td>
                <td><input type="hidden" name="size{0}" value="{2}" readonly /></td>
                <td><input type="hidden" name="quantity{0}" value="{3}" readonly /></td>
                """.format(orderitem, get_pizza_id(db,cursor,each[0]),each[1],each[2])
        orderitem += 1
    
    form += """
            <br/><input type="submit" name="submit" value="Confirm Order"/>
            </form>
            """
            
    return form 

if __name__ == "__main__":
    try:
        html_top("Order Confirmation")
        form_data = cgi.FieldStorage()
        quantity, pizzas, date, time, customer = process_form(form_data)
        db,cursor = connect_pizza_database()
        order_id = add_new_order(db,cursor,quantity, pizzas, date, time, customer)
        print("<h1>Pizza Kitchen</h1>")
        print("<h2>Order Confirmation</h2>")
        print("<p>Thank you for your order! Your order number is: {0}</p>".format(order_id))
        print("""<p><a href="new_order.html">Another Order?</a></p>""")
    except:
        cgi.print_exception()