processorder1.cgi

#!/usr/local/bin/python3

import cgi
from pizza_kitchen_helpers import *

def process_form(form_data):
    num_pizzas = int(form_data.getvalue("pizza"))
    return num_pizzas
    
def get_available_pizzas(db,cursor):
    sql = "select * from pizza"
    cursor.execute(sql)
    pizzas = cursor.fetchall()
    return pizzas
    
def get_available_sizes(db,cursor):
    sql = "select * from size"
    cursor.execute(sql)
    sizes = cursor.fetchall()
    return sizes

def create_pizza_drop_down(pizzas,name):
    drop_down = """<select name="pizza{0}">""".format(name)
    for each in pizzas:
        drop_down += """<option value="{0}">{1}</option>""".format(each[0],each[1])
    drop_down += "</select>"
    return drop_down
    
def create_size_drop_down(sizes,name):
    drop_down = """<select name="size{0}">""".format(name)
    for each in sizes:
        drop_down += """<option value="{0}">{1}</option>""".format(each[0],each[0])
    drop_down += "</select>"
    return drop_down
    
def create_form(num_pizzas,pizzas,sizes):
    form = """<form method="post" action="processorder2.cgi"/>"""
    for each in range(num_pizzas):
        form += create_pizza_drop_down(pizzas,each)
        form += create_size_drop_down(sizes,each)
        form += "<br/>"
    form += """<input type="submit" name="submit" value="Summarise Order"/>"""
    form += """<input type="hidden" name="quantity" value="{0}"/>""".format(num_pizzas)
    form += "</form>"
    return form

if __name__ == "__main__":
    try:
        html_top("Select Pizzas")
        form_data = cgi.FieldStorage()
        num_pizzas = process_form(form_data)
        db,cursor = connect_pizza_database()
        pizzas = get_available_pizzas(db,cursor)
        sizes = get_available_sizes(db,cursor)
        form = create_form(num_pizzas,pizzas,sizes)
        print("<h1>Pizza Kitchen</h1>")
        print("<h2>Choose Pizza Types and Sizes</h2>")
        print(form)
        cursor.close()
        html_tail()
    except:
        cgi.print_exception()