Как продлить экземпляр класса

Question

MyClass определен в module.py. Мы не можем изменить его. Но мы знаем, что определение класса выглядит следующим образом:

class MyClass: 
    def method(self, msg): 
     print 'from method:', msg 

Я начинаю свой скрипт с импортом модуля, а затем объявить экземпляр объекта:

import module  
foo = module.MyClass() 

Тогда я написать свою собственную функцию:

def function(msg): 
    print 'from function:', msg 

Теперь, каждый раз, когда используется foo.method(''), я хочу позвонить function(), поэтому он печатает одно и то же сообщение.

Будет ли эта ситуация называться monkey patching? Как достичь того, что необходимо?

Answer 1

Да, это называется обезьяна-латание.

Это в основном оформление, но выполняется вручную после того, как класс уже определен.

from functools import wraps 

def wrapper(f): 
    @wraps(f) 
    def wrapped(*args, **kwargs): 
     myFunction() 
     return f(*args, **kwargs) 
    return wrapped 

MyClass.printThis = wrapper(MyClass.printThis) 

Это затронет все экземпляры MyClass, даже те, которые были созданы до применения патча.

Если вам не нужно, чтобы динамически изменять поведение во время выполнения, избежать обезьяны заплат и предпочитают использовать наследование настроить поведение, как это было предложено в комментариях. Это менее хаки.

Answer 2

Законченное с этим решением, пока там лучше один отвечал ...

class MyClass: 
    def method(self, msg): 
     print 'from method:', msg 

def function(msg, callback): 
    print 'from function:', msg 
    callback(msg) 

foo = MyClass() 
foo.function = function 
foo.function(msg='message', callback=foo.method) 

Answer 3

Вы могли бы подкласс, а также:

class MyClass: 
    def method(self, msg): 
     print 'from method:', msg 

def function(msg): 
    print 'from function:', msg 

class MyNewClass(MyClass): 
    def method(self, msg): 
     function(msg) 
     MyClass.method(self, msg) 

И использовать его как:

>>> a = MyNewClass() 
>>> a.method("test") 
from function: test 
from method: test 

Или, если вы хотите, чтобы ваш класс «новый стиль» класс (для Python 2 - судя по вашим высказываниям печати) - просто MyClass наследуют от object, а затем вы можете пользователь super:

class MyClass(object): # object added here 
    def method(self, msg): 
     print 'from method:', msg 

def function(msg): 
    print 'from function:', msg 

class MyNewClass(MyClass): 
    def method(self, msg): 
     function(msg) 
     super(self.__class__, self).method(msg) # super added here 

Answer 4

Это альтернатива wim's answer, которая также включает в себя обезглавливание обезьян. Тем не менее, он выполняет функции, предоставляемые unittest.mock. Преимущество этого подхода состоит в том, что менеджер контекста используется для автоматического применения и удалить патч в пределах ограниченного объема:

from unittest import mock 

# This class would be defined in some third-party library 
class MyClass: 
    def method(self, msg): 
     print('from method:', msg) 


def function(msg): 
    print('from function:', msg) 


old_method = MyClass.method 


def new_method(self, msg): 
    old_method(self, msg) 
    function(msg) 


# The patch is only applied within this scope 
with mock.patch.object(MyClass, 'method', new_method): 
    foo = MyClass() 
    foo.method('message with patched') 

# By this point MyClass is "back to normal" 
print('---') 
foo.method('message with original') 

Выходной

from method: message with patched 
from function: message with patched 
--- 
from method: message with original