processcustomer.cgi

#!/usr/local/bin/python3

import cgi
from pizza_kitchen_helpers import *

def process_form(form_data):
    first_name = form_data.getvalue("first")
    last_name = form_data.getvalue("last")
    street = form_data.getvalue("street")
    town = form_data.getvalue("town")
    post_code = form_data.getvalue("postcode")
    email = form_data.getvalue("email")
    return first_name, last_name, street, town, post_code, email
    
def insert_customer(db, cursor,first_name, last_name, street, town, post_code, email):
    sql = """insert into customer 
          (first_name, last_name, street_address, town, post_code, email_address) 
          values ('{0}','{1}','{2}','{3}','{4}','{5}')
          """.format(first_name, last_name, street, town, post_code, email)
    cursor.execute(sql)
    db.commit()

def select_customer(db,cursor,first_name, last_name, post_code):
    sql = """select customer_id
             from customer
             where first_name = '{0}' and last_name = '{1}' and post_code = '{2}'
          """.format(first_name, last_name, post_code)
    cursor.execute(sql)
    customer_id = cursor.fetchall()
    return customer_id[0][0]
    


if __name__ == "__main__":
    try:
        html_top("New Customer Added")
        form_data = cgi.FieldStorage()
        first_name, last_name, street, town, post_code, email = process_form(form_data)
        db,cursor = connect_pizza_database()
        insert_customer(db, cursor,first_name, last_name, street, town, post_code, email)
        customer_id = select_customer(db,cursor,first_name, last_name, post_code)
        print("<h1>Pizza Kitchen</h1>")
        print("<h2>Customer Confirmation</h2>")
        print("<p>Thank you for registering with us. Your customer number is:</p>")
        print("<p><h1>{0}</h1></p>".format(customer_id))
        cursor.close()
        html_tail()
    except:
        cgi.print_exception()