#======================================================================
# Save data in a database
#======================================================================
import pymysql as mydb                      #Python 3x Mysql database driver
# import cx_Oracle as oradb                 #if using Oracle database driver

firstname  = input("Enter your first name: ")           #prompt for user entry 
lastname   = input("Enter your last name: ")
address    = input("Enter your address: ")
flavor     = input("Enter your ice cream flavor: ")
topping    = input("Enter your ice cream topping: ")
creditCard = input("Enter your credit card: ")

def saveDB( ):                                          #save data to database

    sql = F""" INSERT INTO cust_order 
               VALUES(null, '{firstname}', '{lastname}', '{address}', 
                      '{flavor}', '{topping}', '{creditCard}', 1 ) """

#   print(sql)                                          #debugging to ensure SQL code is OK

    try:    
        conn  = mydb.connect(host='localhost',user='demo2',password='demo2',database='demo2') 
#       conn  = oradb.connect( 'demo2/demo2@localhost:1521/orcl' )   

        cursor = conn.cursor( )                         #create a cursor
        cursor.execute(sql);                            #execute the sql
        conn.commit( )

    except mydb.Error as e:                             #catch MySql DB errors
        (errorNum, errorMsg) = e.args
        msg = 'Database Error - ' + str(errorNum) + errorMsg
        return  
#   except oradb.DatabaseError as e:                    #catch Oracle DB errors
#       msg = 'Database Error - ' + e
#       return  

    print("Your order has been saved successfully")
 
    cursor.close( )                                     #close the cursor/buffer
    conn.close( )                                       #close the connection

#=====================================================================
#  Main code
#=====================================================================
saveDB( )                                               #save into the database