我试图在python中包装/猴子补丁模块.我正在尝试开发一种干净的实现方法,不会干扰任何现有的代码.
问题
给定一个从MODULE导入一些CLASS的脚本
from MODULE import CLASS
我想用另一个_MODULE_替换MODULE. _MODULE_是原始MODULE的补丁.我能看到的最干净的界面如下.
from overlay import MODULE # Switches MODULE for _MODULE
from MODULE import CLASS # Original import now uses _MODULE_
这基本上是猴子修补模块,就像猴子补丁类,函数和方法一样.我相信如果这样做是正确的,那么可以用项目特定的方式一致地修补代码.
实现这个的最佳方法是什么？
最佳答案
>>> import wrapt
>>> @wrapt.when_imported('collections')
... def hook(collections):
... OldOrderedDict = collections.OrderedDict
... class MyOrderedDict(OldOrderedDict):
... def monkey(self):
... print('ook ook')
... collections.OrderedDict = MyOrderedDict
...
>>> from collections import OrderedDict
>>> OrderedDict().monkey()
ook ook