如何修改用戶輸入請求中的鍵值

我正在制作一個基本的應用程式，讓用戶添加、洗掉和編輯他們擁有的股票。我正在努力做到這一點，這樣我就可以讓用戶“添加”或“洗掉”他們決定并鍵入它，輸入會捕獲他們的回應。我不使用 if user_input = 'add' 執行此操作，然后 elif user_input == 'remove' 執行此操作。解決這個問題的最佳方法是什么。

這是我編輯庫存部分的代碼；

    def edit_inventory(self):
        item_name = input('Enter item name: ')

        try:
            user_input = input('Type \'ADD\' or \'REMOVE\' to change quantity: ').lower()
            if user_input not in ['add', 'remove']:
                raise ValueError
            quantity = int(input(f'Enter how many you wish to {user_input}: '))
            if user_input == 'add':
                for item_name in self.stock.keys():
                    self.stock[item_name]  = quantity
            elif user_input == 'remove':
                for item_name in self.stock.keys():
                    self.stock[item_name] -= quantity

        except KeyError:
            print('Item not in stock. Check current stock for current inventory.')
        except ValueError:
            print('Invalid response.')

我想解決 if 和 elif 以提高效率。

uj5u.com熱心網友回復：

一種方法是使用調度表，如下所示：

    def add_item(self, item: str, qty: int) -> None:
        self.stock[item]  = qty

    def remove_item(self, item: str, qty: int) -> None:
        self.stock[item] -= qty

    def edit_inventory(self) -> None:
        cmd_table = {
            'add': self.add_item,
            'remove': self.remove_item,
        }
        item = input('Enter item name: ')
        cmd = input("Type 'ADD' or 'REMOVE' to change quantity: ").lower()
        qty = int(input(f'Enter how many you wish to {cmd}: '))

        def invalid_cmd(_item: str, _qty: int) -> None:
            print("Invalid response.")
        try:
            cmd_table.get(cmd, default=invalid_cmd)(item, qty)
        except KeyError:
            print(
                'Item not in stock.',
                'Check current stock for current inventory.'
            )

這樣，如果您有大量命令，則每個命令只需要字典中的一個條目，而不是一堆if/elif. 您還可以免費獲得有關無效條目的合理行為；您可以使用[]并獲取一個KeyError，或者您可以使用.get()并提供一個默認函式，或者您可以測驗if cmd in cmd_table.

如果您非常具體地只修改專案的數量，您還可以將 int 操作放入表中并使用這些操作而不是定義方法：

    def edit_inventory(self) -> None:
        cmd_table = {
            'add': int.__add__,
            'remove': int.__sub__,
        }
        item = input('Enter item name: ')
        cmd = input("Type 'ADD' or 'REMOVE' to change quantity: ").lower()
        qty = int(input(f'Enter how many you wish to {cmd}: '))

        def invalid_cmd(_curr: int, _qty: int) -> None:
            print("Invalid response.")
        try:
            cmd_func = cmd_table.get(cmd, default=invalid_cmd)
            self.stock[item] = cmd_func(self.stock[item], qty)
        except KeyError:
            print(
                'Item not in stock.',
                'Check current stock for current inventory.'
            )

uj5u.com熱心網友回復：

If 提出與 samwise 相同的想法，但采用課堂方法。所以，無論如何我都會發布它

def do_this():
    print("this")


def do_that():
    print("that")

class Command1(dict):
    def __call__(self, command: str):
        self[command]()

command = input("your command: ")
executor = Command1()
executor["add"] = do_this
executor(command)

class Command2:
    my_commands = {"add": do_this, "remove": do_that}
    def __call__(self, command: str):
        self.my_commands[command]()

command = input("your command: ")
executor = Command2()
executor(command)

這樣做的好處是，當您的資料在類中時，您只需添加__call__方法并擴展介面即可。

標籤：

上一篇：如何通過將兩個串列合并在一起來創建字典串列，其中鍵是重復的

下一篇：如何在dart中映射地圖的回傳