1.4 什么是猴子补丁?

也许你没用过猴子补丁，但是你必须知道，在Python/Ruby 这类脚本语言中，有一种用法叫 Monkey Patch 。

这名字挺好玩的哈。在网上，关于这个名称的来源，大致有如下两种说法：

1、第一种说法。这个用法，原本叫“Guerrilla Patch”，Guerrilla 意为「游击队」，理解起来，就是这个补丁，哪里需要它，它就会出现在哪里，和运行状态打补丁这个功能可以说很形象了。后来老外发现，Guerrilla 和 Gorilla 读音是几乎一样的，说着说着就说成了Gorilla Patch，Gorilla 是大猩猩的意思，有点吓人，那干脆就叫 Monkey Patch 吧。

2、在英文中，monkeying about这个词意为 顽皮的，而猴子补丁这种用法 将原来的代码弄乱了，跟猴子一样调皮，有异曲同工之妙，所以这才将这种用法叫 做猴子补丁。

1.4.1 什么是猴子补丁

在程序运行时给代码加补丁的方法被称为 Monkey Patch 。

用法大概就是如下面这样

from SomeOtherProduct.SomeModule import SomeClass

def speak(self):
    print（"ok!"）

SomeClass.speak = speak

它从代码中可以看出，它可以在需要的时候（运行时），给外部模块（内建，第三方模块，自定义模块） 添加/替换 方法。

  • 在运行时替换方法、属性等

  • 在不修改第三方代码的情况下增加原来不支持的功能

  • 在运行时为内存中的对象增加patch而不是在磁盘的源代码中增加

1.4.2 实例讲解

这里举一个添加方法的例子

import pandas as pd
def just_foo_cols(self):
    """Get a list of column names containing the string 'foo'

    """
    return [x for x in self.columns if 'foo' in x]

pd.DataFrame.just_foo_cols = just_foo_cols # monkey-patch the DataFrame class
df = pd.DataFrame([list(range(4))], columns=["A","foo","foozball","bar"])
df.just_foo_cols()
del pd.DataFrame.just_foo_cols # you can also remove the new method

在openstack中的例子

image0

还有就是gevent中也有用到。

1.4.3 如何使用猴子补丁？

为实例绑定对象(扩展__slot__)

import types
import logging


def get_logger():
    logger = logging.getLogger(__name__)
    logger.setLevel(logging.INFO)
    handler = logging.FileHandler('monkey_patch.log')
    handler.setLevel(logging.INFO)

    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    handler.setFormatter(formatter)

    logger.addHandler(handler)
    return logger


LOG = get_logger()


def info(self, msg, *args, **kwargs):
    print(msg)

    INFO = 20
    if self.isEnabledFor(INFO):
        self._log(INFO, msg, args, **kwargs)

# 重点
LOG.info = types.MethodType(info, LOG)

LOG.info('hello, world')

给函数添加装饰器

func = importutils.import_class("%s.%s" % (module, key))
setattr(sys.modules[module], key,
        decorator("%s.%s" % (module, key), func))
关注公众号，获取最新干货！

关注公众号，获取最新干货！