Cart.py

# for functions to add items, delete items, and checkout, boolean for checked out?
from datetime import datetime
import pandas as pd
from Inventory import Inventory
from Order import Order
import random

class Cart:
    def __init__(self, username):
        self.Username = username
        self.ISBN = None
        self.quantity = None
        self.CardNumber = str(datetime.today().timestamp())
        self.goodList = pd.DataFrame(columns=['ISBN', 'Quantity', 'Price'], dtype=object)
        self.check=False

    #retrieve card number
    def getCardNumber(self):
        return self.CardNumber

    def AddToCart(self, item, quantity):
        price = Inventory().getPrice(item)

        if (price == -1):
            return "Book does not exist"

        self.ISBN = item
        self.Quantity = quantity
        new_row = pd.Series({"ISBN": item, "Quantity": quantity, "Price": price})
        self.goodList = self.goodList.append(new_row, ignore_index=True)

    def getQuantity(self):
        return self.quantity

    def RemoveFromCart(self, isbn):
        # df.drop(df.loc[df['line_race'] == 0].index, inplace=True)
        self.goodList.drop(self.goodList.loc[self.goodList['ISBN'] == isbn].index, inplace=True)
        self.goodList.reset_index(drop=True, inplace=True)
        # x = np.array(self.goodList)
        # x=x[x[:, 0] != item]
        # self.goodList = x.tolist()

    # makes an new Order and prints a document
    # returns the new order ID which will be added to the user’s orders array
    def checkout(self, inventory, order, customer):
        if (len(self.goodList.index) == 0):
            print("Cart empty - can't checkout")
            return

        carttotal = 0

        #This will calculate the total of the cart.
        for i in range(len(self.goodList.index)):
            isbn = self.goodList.loc[i].ISBN
            quantity = int(self.goodList.loc[i].Quantity)
            price = int(self.goodList.loc[i].Price)
            total = quantity * price
            carttotal = carttotal + total

        print("Cart total: $", carttotal)

        #Below here, we start adding user payment information.
        #This will loop as long as credit card numbers are invalid (aka should only include numbers)
        while True:
            CardNumber = input("\nInput CardNumber: ")
            if CardNumber.isnumeric() == True:
                break
            elif CardNumber.isnumeric() == False:
                print("Invalid Input. Card Number should contain numbers only.")

        cardName = input("Name on card: ")
        billingAddress = input("Please insert your billing address STREET NUMBER AND STREET NAME: ")
        billingCity = input("Please insert billing address CITY: ")
        billingState = input("Please insert billing address STATE: ")
        billingZIP = int(input("Please insert billing address ZIP: "))
        orderNum = random.randint(1, 5000)

        for i in range(len(self.goodList.index)):
            isbn = self.goodList.loc[i].ISBN
            quantity = int(self.goodList.loc[i].Quantity)
            price = int(self.goodList.loc[i].Price)
            total = quantity * price
            order.addNewOrder(CardNumber, cardName, billingAddress, billingCity, billingState, billingZIP, isbn, quantity, price, total, orderNum)

        #If the user would like to store the card information for this order
        print ("Would you like to store this card information to your account?")
        storeCardInfo = input("(Y) Yes\n(N) No\n>>")

        if storeCardInfo.lower() == 'y':
            customer.EditPaymentInfo(cardName, CardNumber, billingAddress, billingCity, billingState, billingZIP)

        self.check = True
        self.goodList = pd.DataFrame(columns=['ISBN', 'Quantity'], dtype=object)
        order.save2file()

        print("\n------Checkout Successful------")


    #If the user deletes something from their cart, this will add it back to inventory -- takes away the reservation.
    def InventoryCheck(self, isbn):
        for i in range(len(self.goodList.index)):
            if (self.goodList.loc[i].ISBN == isbn):
                return self.goodList.loc[i].Quantity

        return 0

    def addback(self, inventory, isbn, quantity):
        for i in range(len(self.goodList.index)):
            if (self.goodList.loc[i].ISBN == isbn):
                inventory.addQuantity(isbn, quantity)

    def AddBackToInventory(self, inventory):

        for i in range(len(self.goodList.index)):
            isbn = self.goodList.loc[i].ISBN
            quantity = int(self.goodList.loc[i].Quantity)
            inventory.addQuantity(isbn, quantity)

    def isCheck(self):
        return self.check

    def CartCheck(self):
        if (len(self.goodList.index) == 0):
            return False
        else:
            return True

    def print(self):
        if len(self.goodList) > 0:
            print("\n#######Cart List################")
            print(self.goodList)
            print('################################')
        else:
            print("\n-------------------------Cart is Empty-----------------------")