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

发布时间：2020-12-16 22:45:47 所属栏目：Python 来源：网络整理

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

我试图在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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!