我试图在python中包装/猴子补丁模块.我正在尝试开发一种干净的实现方法,不会干扰任何现有的代码.
问题
给定一个从MODULE导入一些CLASS的脚本
from MODULE import CLASS
我想用另一个_MODULE_替换MODULE. _MODULE_是原始MODULE的补丁.我能看到的最干净的界面如下.
from overlay import MODULE # Switches MODULE for _MODULEfrom 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