有什么方法可以更改函数外部定义的变量或函数

Question

假设我有foo.py，如下所示。 这个文件不是我创建的，所以我既不想修改也不想复制它。 换句话说，foo.py 在我安装的某个额外包中。

# foo.py

def bar():
  print('This is bar')

def foo():
  print('something')
  bar()
  print('something')

然后，我想实现 foo_as_baz() 行为为 cmets。

# baz.py

from foo import foo

def baz():
  print('This is baz')

def foo_as_baz():
"""
This function is expected to behave as below
  print('something')
  baz()
  print('something')
"""  
  pass

我尝试了以下一种，但由于名称范围不同，它不起作用。

def foo_as_baz():
  bar = baz # I expect this `baz` affects `bar` function in `foo`
  foo()

Answer 1

郑重声明：不要在普通代码中做低于水平线的事情。这令人困惑，而且在很大程度上毫无意义。

如果你正在做这种事情来测试你的代码，我推荐标准库unittest.mock 包，它提供了一种系统的方法来用假的 Python 对象替换 Python 对象以进行测试。

如果您认为您有一些真正的非测试用例，请详细说明...

---

# foo.py

def bar():
    print("This is bar")

def foo():
    print("something")
    bar()
    print("something")

然后在另一个文件中：

import foo


def baz():
    print("This is baz")


def foo_as_baz():
    original = foo.bar  # save a copy of the original bar()

    foo.bar = baz       # make foo.bar point to baz()

    foo.foo()           

    foo.bar = original  # cleanup by restoring bar()


foo_as_baz()

与您的尝试的主要区别在于我执行了import foo，这让我可以查看实际的模块命名空间foo。我将foo.bar 换成baz，然后调用foo.foo()，然后将foo.bar 切换回原来的bar 函数。

这是因为函数实际上是模块的属性。因此，我们可以重新分配这些属性，就好像它们是任何其他对象的属性一样（因为在 Python 中，包括模块在内的一切都是对象）。

Answer 2

您可以通过稍微修改foo 的工作方式将函数本身作为参数传递给foo

def foo(func=bar):
   print('something')
   func()
   print('something')

def foo_as_baz():
   foo(baz)

通过将bar 添加为foo 的默认参数，您可以保留原始行为