Restaurant_management_backend.py

from config import connect

from datetime import datetime

# Function to create a new customer
def create_customer():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            name = input("Enter customer name: ")
            address = input("Enter customer address: ")
            phone_no = input("Enter customer phone number (optional, press Enter to skip): ").strip() or None
            birthday = input("Enter customer birthday (YYYY-MM-DD): ")
            cursor.execute("INSERT INTO customers (name, address, phone_no, birthday) VALUES (%s, %s, %s, %s)",
                           (name, address, phone_no, birthday))
            conn.commit()
            print("Customer created successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while creating customer.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()


# Function to reserve a table
def reserve_table():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            customer_id = int(input("Enter customer ID: "))
            table_id = int(input("Enter table ID to reserve: "))
            date = input("Enter reservation date (YYYY-MM-DD): ")
            cursor.execute("INSERT INTO reservation (customer_id, table_id, date) VALUES (%s, %s, %s)",
                           (customer_id, table_id, date))
            conn.commit()
            print("Table reserved successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while reserving the table.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to remove a table reservation
def remove_reservation():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            reservation_id = int(input("Enter reservation ID to remove: "))
            cursor.execute("DELETE FROM reservation WHERE reservation_id = %s", (reservation_id,))
            conn.commit()
            print("Table reservation removed successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while removing the reservation.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to make an order
def make_order():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            customer_id = int(input("Enter customer ID: "))
            delivery_id_input = input("Enter delivery ID (or leave blank for null): ")
            if delivery_id_input.strip() == "":
                delivery_id = None
            else:
                delivery_id = int(delivery_id_input)
            table_id = int(input("Enter table ID for the order: "))
            is_delivery = bool(input("Enter 1 if delivery otherwise 0: "))
            item_name = (input("Enter item name: "))
            quantity = int(input("Enter quantity: "))
            if (item_name != 'Pizza' or item_name != 'Burger' or item_name != 'Salad') :
                print('item not included in menu')
                return
            else :
                cursor.execute("INSERT INTO orders (customer_id, table_id,deliv_id,is_delivery, order_status,item_name, quantity) VALUES (%s, %s, %s, %s, %s, %s, %s)",
                (customer_id, table_id, delivery_id,is_delivery, 'Preparing', item_name, quantity))
                conn.commit()
                print("Order placed successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while placing the order.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to remove an order
def remove_order():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            order_id = int(input("Enter order ID to remove: "))
            cursor.execute("DELETE FROM orders WHERE order_id = %s", (order_id,))
            conn.commit()
            print("Order removed successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while removing the order.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to change order status
def change_order_status():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            order_id = int(input("Enter order ID to change status: "))
            new_status = input("Enter new status: ")
            cursor.execute("UPDATE orders SET order_status = %s WHERE order_id = %s", (new_status, order_id))
            conn.commit()
            print("Order status changed successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while changing order status.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to make a delivery order
def make_delivery_order():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            address = input("Enter delivery address: ")
            c_id = int(input("Enter customer id: "))
            cursor.execute("INSERT INTO delivery (address,customer_id) VALUES (%s, %s) RETURNING delivery_id", (address,c_id))
            delivery_id = cursor.fetchone()[0]
            print("Delivery order placed successfully. Delivery ID:", delivery_id)
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while placing delivery order.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to remove a delivery order
def remove_delivery_order():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            delivery_id = int(input("Enter delivery ID to remove: "))
            cursor.execute("DELETE FROM delivery WHERE delivery_id = %s", (delivery_id,))
            conn.commit()
            print("Delivery order removed successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while removing delivery order.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to make a transaction
def make_transaction():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            c_id = int(input("Enter customer id: "))
            amount = float(input("Enter transaction amount: "))
            payment_method = input("Enter payment method: ")
            date = input("Enter transaction date (YYYY-MM-DD): ")
            tips = float(input("Enter tips amount (if any): "))
            cursor.execute("INSERT INTO transactions (amount, payment_method, date, tips, customer_id) VALUES (%s, %s, %s, %s, %s)",
                           (amount, payment_method, date, tips, c_id))
            conn.commit()
            print("Transaction recorded successfully.")
        except psycopg2.Error as e:
            conn.rollback()
            print("Error occurred while recording the transaction.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()
                       
# Function to display customer information
def display_customer_info():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            cursor.callproc("get_customer_info")
            customer_info = cursor.fetchall()
            print("\nCustomer Information:")
            for row in customer_info:
                print("Customer ID:", row[0])
                print("Customer Name:", row[1])
                print("Address:", row[2])
                print("Phone Number:", row[3])
                print("Birthday:", row[4])
                print('\n')
        except psycopg2.Error as e:
            print("Error occurred while fetching customer information.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to display transactions
def display_transactions():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            cursor.callproc("get_transactions")
            transactions = cursor.fetchall()
            for transaction in transactions:
                print("Transaction ID:", transaction[0])
                print("Customer ID:", transaction[5])
                print("Amount:", transaction[1])
                print("Payment Method:", transaction[2])
                print("Time:", transaction[3])
                print("Tips:", transaction[4])
                print('\n')
        except psycopg2.Error as e:
            print("Error occurred while fetching transactions.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to display delivery information
def display_delivery():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            cursor.callproc("get_delivery")
            deliveries = cursor.fetchall()
            for delivery in deliveries:
                print("Delivery ID:", delivery[0])
                print("Customer ID:", delivery[1])
                print("Address:", delivery[2])
                print('\n')
        except psycopg2.Error as e:
            print("Error occurred while fetching deliveries.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to display reservations
def display_reservations():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            cursor.callproc("get_reservations")
            reservations = cursor.fetchall()
            for reservation in reservations:
                print("Reservation ID:", reservation[0])
                print("Customer ID:", reservation[1])
                print("Table ID:", reservation[2])
                print("Date:", reservation[3])
                print("Size:", reservation[4])
                print('\n')
        except psycopg2.Error as e:
            print("Error occurred while fetching reservations.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to display menu
def get_menu():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            cursor.callproc("get_menu")
            menu = cursor.fetchall()
            print("\nMENU:")
            for m in menu:
                print("Item name:", m[0])
                print("Description:", m[1])
                print("Price:", m[2])
                print('\n')
        except psycopg2.Error as e:
            print("Error occurred while fetching reservations.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()

# Function to display orders
def display_orders():
    conn = connect()
    if conn:
        try:
            cursor = conn.cursor()
            cursor.callproc("get_orders")
            orders = cursor.fetchall()
            for order in orders:
                print("Order ID:", order[0])
                print("Customer ID:", order[1])
                print("Table ID:", order[2])
                print("Delivery ID:", order[3])
                print("Is Delivery:", order[4])
                print("Order Status:", order[5])
                print("Item name:", order[6])
                print("Quantity:", order[7])
                print("\n")
        except psycopg2.Error as e:
            print("Error occurred while fetching orders.")
            print(e)
        finally:
            if cursor:
                cursor.close()
            conn.close()


# Main function to provide options to the user
def main():
    while True:
        print("\nSelect an option:")
        print("1. Reserve a table")
        print("2. Remove table reservation")
        print("3. Make an order")
        print("4. Remove an order")
        print("5. Change order status")
        print("6. Make a delivery order")
        print("7. Remove a delivery order")
        print("9. Make a transaction")
        print("0. Exit")
        print("A. Create a new customer")
        print("B. Display options")

        choice = input("Enter your choice: ")
        if choice == "1":   
            reserve_table()
        elif choice == "2":
            remove_reservation()
        elif choice == "3":
            make_order()
        elif choice == "4":
            remove_order()
        elif choice == "5":
            change_order_status()
        elif choice == "6":
            make_delivery_order()
        elif choice == "7":
            remove_delivery_order()
        elif choice == "9":
            make_transaction()
        elif choice.upper() == "A":
            create_customer()
        elif choice.upper() == "B":
            print("\nSelect an option to display:")
            print("1. Customer information")
            print("2. Transactions")
            print("3. Deliveries")
            print("4. Reservations")
            print("5. Orders")
            print("6. Menu")
            print("0. Back")
            choice = input("Enter your choice: ")
            if choice == "1":
                display_customer_info()
            elif choice == "2":
                display_transactions()
            elif choice == "3":
                display_delivery()
            elif choice == "4":
                display_reservations()
            elif choice == "5":
                display_orders()
            elif choice == "6":
                get_menu()
            elif choice == "0":
                break
            else:
                print("Invalid choice. Please try again.")
                
        elif choice == "0":
            break
        else:
            print("Invalid choice. Please try again.")

if __name__ == "__main__":
    main()