Python Overlays:猴子补丁的一个案例

Question

我试图在python中包装/猴子补丁模块.我正在尝试开发一种干净的实现方法,不会干扰任何现有的代码.

问题

给出一个CLASS从a 导入一些的脚本MODULE

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_

这基本上是猴子修补模块,就像猴子补丁类,函数和方法一样.我相信如果这样做是正确的,那么可以用项目特定的方式一致地修补代码.

实现这个的最佳方法是什么？

Answer 1

>>> 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