Блог

У меня есть программа инвентаризации, варианты 1-2 работают, как и вариант 4. Текущая проблема — вариант 3. Он правильно обновляет количество товаров, если обновление является добавлением. Но если количество является вычитанием, оно не вычитается, а вместо этого добавляется. Чтобы дублировать мою ошибку типа 1, введите Apple, кол-во 10 (добавление товара / кол-во в инвентарь), введите 98, затем 3 (Обновление), введите -4, введите 98, затем 4 (повторно введите имя товара, и он напечатает имя товара и неправильное количество).

Как мне это исправить?

Полный код:

 import os


name = []

qty = []

class Foo():
    def __init__(self, name, qty):
        self.name = name
        self.qty = qty

def menuDisplay():
    print ('=============================')
    print ('= Inventory Management Menu =')
    print ('=============================')
    print ('(1) Add New Item to Inventory')
    print ('(2) Remove Item from Inventory')
    print ('(3) Update Inventory')
    print ('(4) Search Item in Inventory')
    print ('(5) Print Inventory Report')
    print ('(99) Quit')
    CHOICE = int(input("Enter choice: "))
    menuSelection(CHOICE)

def menuSelection(CHOICE):

    if CHOICE == 1:
        print('Adding Inventory')
        print('================')
        new_name = input('Enter the name of the item: ')
        name.append(new_name)
        new_qty = int(input("Enter the quantity of the item: "))
        qty.append(new_qty)
        CHOICE = int(input('Enter 98 to continue or 99 to exit: '))
    if CHOICE == 98:
        menuDisplay()
    elif CHOICE == 99:
        exit()
    elif CHOICE == 2:
        print('Removing Inventory')
        print('==================')
        removing = input('Enter the item name to remove from inventory: ')
        indexdel = name.index(removing)
        name.pop(indexdel)
        qty.pop(indexdel)
        CHOICE = int(input('Enter 98 to continue or 99 to exit: '))
    if CHOICE == 98:
        menuDisplay()
    elif CHOICE == 99:
        exit()
    elif CHOICE == 3:
        print('Updating Inventory')
        print('==================')
        item = input('Enter the item to update: ')
        update = int(input("Enter the updated quantity. Enter 5 for additional or -5 for less: "))
        if update <= -1:
            qty[name.index(item)] -= update
            print("Update made")
            CHOICE = int(input('Enter 98 to continue or 99 to exit: '))
            if CHOICE == 98:
                menuDisplay()
            elif CHOICE == 99:
                exit()
        elif update >= 0:
            qty[name.index(item)]  = update
            print("Update Made")
            CHOICE = int(input('Enter 98 to continue or 99 to exit: '))
            if CHOICE == 98:
                menuDisplay()
            elif CHOICE == 99:
                exit()
    elif CHOICE == 4:
        print('Searching Inventory')
        print('===================')
        search = input('Enter the name of the item: ')
        pos = name.index(search) if search in name else -1
        if (pos >= 0):
            print ('Item:     ', name[pos])
            print ('Quantity: ', qty[pos])
            print ('----------')
        else:
            print("Item not in inventory")
    CHOICE = int(input('Enter 98 to continue or 99 to exit: '))
    if CHOICE == 98:
            menuDisplay()
    elif CHOICE == 99:
        exit()
    elif CHOICE == 5:
        print('Current Inventory')
        print('=================')
        input('Enter the name of the item: ')
        CHOICE = int(input('Enter 98 to continue or 99 to exit: '))
    if CHOICE == 98:
        menuDisplay()
    elif CHOICE == 99:
        exit()
        printInventory()
    elif CHOICE == 99:
        exit()

menuDisplay()
  

Поскольку update значение отрицательное, вычитание его из qty[name.index(item)] увеличит количество, поскольку вычитание отрицательного числа совпадает с добавлением. Просто добавьте его вместо этого, как вы делаете для положительного случая.