python - 是否可以检测函数中声明的局部变量的数量？

Question

在 Python 测试夹具中，是否可以统计一个函数在其主体中声明了多少个局部变量？

def foo():
    a = 1
    b = 2

Test.assertEqual(countLocals(foo), 2)

或者，有没有办法查看函数是否声明了任何变量？

def foo():
    a = 1
    b = 2

def bar():
    pass

Test.assertEqual(hasLocals(foo), True)
Test.assertEqual(hasLocals(bar), False)

我正在考虑的用例与验证用户提交的代码有关。

Answer 1

是的，关联的代码对象说明了 co_nlocals 属性中的所有本地名称:

foo.__code__.co_nlocals

演示:

>>> def foo():
...     a = 1
...     b = 2
...
>>> foo.__code__.co_nlocals
2

参见 Datamodel documentation :

User-defined functions

[...]

__code__ The code object representing the compiled function body.

Code objects

[...]

Special read-only attributes: [...] co_nlocals is the number of local variables used by the function (including arguments); [...]