Python：修改内部方法内的方法局部变量

Posted 2023-02-22

【中文标题】Python：修改内部方法内的方法局部变量

【英文标题】：Python: modify method-local variable inside inner method

【发布时间】：2016-05-23 22:40:12

【问题描述】：

我正在编写一个名为 test_foo 的测试方法（使用 pytest）。我正在测试一个函数foo 的行为，该函数将另一个函数get 作为参数。 foo 迭代调用get，有条件地根据get 的返回值，例如：

def foo(get, param):
    max_num_tries = 3
    curr_num_tries = 0
    response = get(param)
    while curr_num_tries < max_num_tries and response.status_code == BAD_STATUS_CODE:
        response = get(param)
    return response

我正在尝试覆盖get，以便它可以访问它被调用的次数并可以相应地返回不同的值。

这是我目前所拥有的简化版本：

def test_foo():
    tries_so_far = 0

    def get(arg1):
        global tries_so_far
        if tries_so_far < 3:
            tries_so_far += 1
            print("do something special here")
        else:
            print("do something else here")
        return "some return val"

    foo(get, "some arg")

但是，我收到以下错误：

NameError: global name 'tries_so_far' is not defined

如果我在test_foo 之外定义tries_so_far，在模块级别，我会得到预期的行为。但是，我希望 tries_so_far 成为 test_foo 的本地变量。

有没有办法让get 使用全局变量或其他技术对tries_so_far 进行读/写？注意：我无法更改get的参数或返回值。

【问题讨论】：

您似乎没有使用对 tries_so_far 的写入权限。只需删除global，它将能够读取变量。 global 表示“模块级别”，正如您已正确确定的那样。

对不起 - 我忘了把它添加到我的例子中。已更新。

我假设您使用的是打印语句中的 Py3k。如果是这样，您可以更改标签吗？

【参考方案1】：

根据这个问题Why can functions in Python print variables in enclosing scope but cannot use them in assignment? 的公认答案，Python 3 中添加了一条附加语句：nonlocal 可以满足您的需求。它类似于global，但表示要查看封闭范围而不是模块级别。因此，以下修改应该可以让您完全按照自己的意愿行事：

def test_foo():
    tries_so_far = 0

    def get(arg1):
        nonlocal tries_so_far
        if tries_so_far < 3:
            tries_so_far += 1
            print("do something special here")
        else:
            print("do something else here")
        return "some return val"

    foo(get, "some arg")

虽然您的问题与我上面引用的问题并不完全相同，但您应该阅读已接受的答案。它非常好，可能会解决您对此主题的许多未说明的问题。