在 Python 中定义要从类中的其他方法调用的 main_menu 函数

Question

我有一个类，它有几个方法可以从用户输入中分配属性，还有三个方法可以使用输入添加、删除或更新嵌套字典。 我想添加一个 main_menu 函数，以便用户可以访问添加、删除和更新方法，然后选择继续添加/删除/更新字典，或者返回主菜单。

当我尝试创建一个 main_menu 函数时，我收到 NameError: name 'command' is not defined。如果 main_menu 不是函数，则程序将按预期循环，但是一旦我尝试将其转换为函数，就会出现错误。我尝试了不同级别的缩进，但我是 Python 新手，不知道还能尝试什么。

class MyClass:
    def __init__(self):
        self.x = 0
        self.y = 0
        self.z = 0

    def get_x(self):
        #code to get x from user input
    def get_y(self):
        #code to get y from user
    def get_z(self):
        #code to get z from user

    def add_info(self):
        store_info = {}
        id = 0
        while command = '1':
            new_info = {}
            new_id = len(store_info) + 1
            store_info[new_id] = new_info

            x = h.get_x()
            new_info['x'] = x
            y = h.get_y()
            new_info['y'] = y
            z = h.get_z()
            new_info['z'] = z

            print('Your info has been updated.\n', store_info)
            choice = input('To add more info, type 1. To return to the main menu, type 2')
            if choice == '1':
                continue
            elif choice == '2':
                 main_menu()
            else:
                print('The End')
                break

    def delete_info(self):
        #code to delete, with an option at the end to return to main_menu

    def update_info(self):
        #code to update, with option for main_menu

def main_menu():
    main_menu_option = """Type 1 to add.
Type 2 to delete.
Type 3 to update.
Type any other key to quit.\n"""

    h = MyClass()
    command = input(main_menu_option)
    if command == '1':
        h.add_info()
    elif command == '2':
        h.delete_info()
    elif command == '3':
        h.update_info()
    else:
        print('Good bye.')

main_menu()

当我运行程序时，我得到主菜单并输入 1，但随后收到 NameError for command。 在我尝试将 main_menu 设为函数之前，我可以访问 add 方法将信息添加到嵌套字典中。

Answer 1

在 Python 中，方法/函数只能访问其作用域或父作用域中的变量。例如：

command = 1  # A

def foo():
    print(command)

def bar():
    command = 2  # B
    print(command)

foo()
bar()
print(command)

打印出1，然后是2，然后是1。这是因为当我们调用 foo 时，python 会查看局部变量并意识到那里没有 command 变量，因此它查看上面的范围并在标签 A 处看到 command 变量。然后当我们调用 @987654328 @我们将一个名为command（标签B）的变量添加到bar的局部变量中，然后打印它，注意python不会在这里查看全局变量，所以它打印2而不更改初始@987654332 @变量（标签A）当我们最终在脚本末尾print(command)时可以看到。

正因为如此，您的程序失败了，您的 add_info 方法正在尝试访问一个名为 command 的变量，但是在其局部变量或其全局范围内不存在任何变量。解决此问题的方法是将command 添加到add_info 局部变量，方法是将其作为参数从main_menu 传递给方法。

# in main_menu
if command == '1':
    h.add_info(command)

# in add_info
def add_info(self, command):
    ...