python 猴子补丁(Monkey Patch)

前言

>>转载<<

python里面什么是猴子补丁(Monkey Patch)?,使用场景有哪些? 猴子补丁主要有以下几个用处:

  • 在运行时替换方法、属性等
  • 在不修改第三方源代码的情况下增加原来不支持的功能,比如自定义json序列化对象
  • 在运行时为内存中的对象增加patch而不是在磁盘的源代码中增加

明确一个观点:猴子补丁(monkey patch)的主要功能就是动态的属性的替换

猴子补丁(Monkey Patch)

属性在运行时的动态替换,叫做猴子补丁(Monkey Patch)。 作用是在运行的时候,动态替换模块方法。先看一个简单的例子 如果有一个模块somemodule.py,在其它代码里面有用到这个类里面的speak方法

class SomeClass(object):

    def __init__(self):
        self.name = "yoyo"

    def speak(self):
        return "hello world"

在不改变原来代码的基础上,可以重新定义一个speck方法,替换原来的

from xxx.somemodule import SomeClass

def new_speak(self):
    return "new hello"

# 替换speck方法
SomeClass.speak = new_speak

替换之后,调用的时候,就会变成新的方法里面内容

some = SomeClass()
print(some.speak())
# 运行结果 new hello

python自定义对象转json串

我自己定义了一个类,如下

class MyDefined(object):

    def __init__(self):
        self.name = "yoyo"
        self.age = 18

    def __repr__(self):
        return 'name={}&age={}'.format(self.name, self.age)

在定义字典的时候,引用了上面类的实例

aa = {
    "a1": True,
    "b1": "hello",
    "c1": [1, 2, 3],
    "d1": MyDefined()
}
print(aa)
# 运行结果:{'a1': True, 'b1': 'hello', 'c1': [1, 2, 3], 'd1': name=yoyo&age=18}

如果直接转json串,会出现异常:TypeError: Object of type ‘MyDefined’ is not JSON serializable

import json

print(json.dumps(aa))

因为MyDefined类是我自己定义的,json库无法解析成对应的字符串,这种情况就需要自己去写一个解析方式

方法一:可以在json.dumps 传一个cls参数

import json

class MyEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, MyDefined):
            return str(obj)
        else:
            return json.JSONEncoder.default(self, obj)
print(json.dumps(aa, cls=MyEncoder))
# 运行结果 {"a1": true, "b1": "hello", "c1": [1, 2, 3], "d1": "name=yoyo&age=18"}

方法二:使用猴子补丁来解决

from json import JSONEncoder

def new_default(self, obj):
    if isinstance(obj, MyDefined):
        return str(obj)
    else:
        return json.JSONEncoder.default(self, obj)

JSONEncoder.default = new_default
print(json.dumps(aa))
# 运行结果 {"a1": true, "b1": "hello", "c1": [1, 2, 3], "d1": "name=yoyo&age=18"}

猴子补丁的应用

还有一个比较实用的例子,很多代码用到 import json,后来发现ujson性能更高, 如果觉得把每个文件的import json 改成 import ujson as json成本较高,或者说想测试一下用ujson替换json是否符合预期,只需要在入口加上:

import json  
import ujson  

def monkey_patch_json():  
    json.__name__ = 'ujson'  
    json.dumps = ujson.dumps  
    json.loads = ujson.loads  

monkey_patch_json()

运行中替换或者添加方法是非常有用的,比如说在单元测试中,有些负责和外界服务通信的函数就需要替换掉,方便测试。

这个技巧不仅很常用,而且在你最终决定要修改代码之前还可以保持代码的可维护性,是一个非常重要的技巧。

还有比如你在pip安装第三方包的时候,使用过程中发现某个方法有一些bug,或者不兼容中文的情况,这时候不要去改源码(虽然改源码能解决,但不利于后期维护,后期你换个电脑,重新pip安装的时候,就忘记之前改哪里了) 像这种情况下,可以自己写个方法去替换第三方包原来的方法,非常实用!

Monkey Patch带了便利的同时也有搞乱源代码优雅的风险

posted @ 2022-03-24 12:35  www.pu  Views(596)  Comments(0Edit  收藏  举报