Добавление метода в существующий экземпляр объекта


643

Я читал, что возможно добавить метод к существующему объекту (то есть, не в определении класса) в Python.

Я понимаю, что это не всегда хорошо. Но как можно это сделать?

Ответы:


921

В Python есть разница между функциями и связанными методами.

>>> def foo():
...     print "foo"
...
>>> class A:
...     def bar( self ):
...         print "bar"
...
>>> a = A()
>>> foo
<function foo at 0x00A98D70>
>>> a.bar
<bound method A.bar of <__main__.A instance at 0x00A9BC88>>
>>>

Связанные методы были «привязаны» (насколько описательно) к экземпляру, и этот экземпляр будет передан в качестве первого аргумента при каждом вызове метода.

Однако вызываемые элементы, являющиеся атрибутами класса (в отличие от экземпляра), по-прежнему не связаны, поэтому вы можете изменить определение класса в любое время:

>>> def fooFighters( self ):
...     print "fooFighters"
...
>>> A.fooFighters = fooFighters
>>> a2 = A()
>>> a2.fooFighters
<bound method A.fooFighters of <__main__.A instance at 0x00A9BEB8>>
>>> a2.fooFighters()
fooFighters

Ранее определенные экземпляры также обновляются (если они сами не переопределяют атрибут):

>>> a.fooFighters()
fooFighters

Проблема возникает, когда вы хотите присоединить метод к одному экземпляру:

>>> def barFighters( self ):
...     print "barFighters"
...
>>> a.barFighters = barFighters
>>> a.barFighters()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: barFighters() takes exactly 1 argument (0 given)

Функция не привязывается автоматически, когда она прикреплена непосредственно к экземпляру:

>>> a.barFighters
<function barFighters at 0x00A98EF0>

Чтобы связать его, мы можем использовать функцию MethodType в модуле типов :

>>> import types
>>> a.barFighters = types.MethodType( barFighters, a )
>>> a.barFighters
<bound method ?.barFighters of <__main__.A instance at 0x00A9BC88>>
>>> a.barFighters()
barFighters

На этот раз другие экземпляры класса не были затронуты:

>>> a2.barFighters()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: A instance has no attribute 'barFighters'

Дополнительную информацию можно найти, прочитав о дескрипторах и программировании метаклассов .


65
Вместо того, чтобы вручную создавать a MethodType, вызовите протокол дескриптора вручную, и пусть функция создаст ваш экземпляр: создаст barFighters.__get__(a)связанный метод для barFightersпривязанного к a.
Мартин Питерс

2
@MartijnPieters любые преимущества использования descriptor protocolпротив создания, MethodTypeвозможно, быть немного более читабельным.
EndermanAPM

17
@EndermanAPM: Несколько: скорее всего, он продолжит работать точно так же, как и при доступе к атрибуту в экземпляре. Он будет работать classmethodи staticmethodи другие дескрипторы тоже. Это позволяет избежать загромождения пространства имен еще одним импортом.
Мартин Питерс

34
Полный код для предложенного дескрипторного подходаa.barFighters = barFighters.__get__(a)
eqzx

98

Новый модуль устарел с python 2.6 и удален в 3.0, используйте типы

см. http://docs.python.org/library/new.html

В приведенном ниже примере я намеренно удалил возвращаемое значение из patch_me()функции. Я думаю, что возвращаемое значение может заставить поверить, что patch возвращает новый объект, который не соответствует действительности - он изменяет входящий. Вероятно, это может способствовать более дисциплинированному использованию обезьяноподготовки.

import types

class A(object):#but seems to work for old style objects too
    pass

def patch_me(target):
    def method(target,x):
        print "x=",x
        print "called from", target
    target.method = types.MethodType(method,target)
    #add more if needed

a = A()
print a
#out: <__main__.A object at 0x2b73ac88bfd0>  
patch_me(a)    #patch instance
a.method(5)
#out: x= 5
#out: called from <__main__.A object at 0x2b73ac88bfd0>
patch_me(A)
A.method(6)        #can patch class too
#out: x= 6
#out: called from <class '__main__.A'>

1
Как это будет работать, если добавляемый метод должен ссылаться на себя? Моя первая попытка приводит к синтаксической ошибке, но добавление self в определении метода, похоже, не работает. импортирует типы классов A (объект): #, но, похоже, работает и для объектов старого стиля ax = 'ax' pass def patch_me (target): метод def (target, x): print (self.ax) print ("x =" , x) print ("call from", target) target.method = types.MethodType (method, target) # добавить еще при необходимости a = A () print (a.ax)
WesR

85

Предисловие - примечание о совместимости: другие ответы могут работать только в Python 2 - этот ответ должен прекрасно работать в Python 2 и 3. Если вы пишете только Python 3, вы можете явно отказаться от наследования object, но в противном случае код должен остаться прежним ,

Добавление метода в существующий экземпляр объекта

Я читал, что возможно добавить метод к существующему объекту (например, не в определении класса) в Python.

Я понимаю, что это не всегда хорошее решение сделать это. Но как можно это сделать?

Да, это возможно - но не рекомендуется

Я не рекомендую это. Это плохая идея. Не делай этого.

Вот несколько причин:

  • Вы будете добавлять связанный объект к каждому экземпляру, с которым вы это делаете. Если вы делаете это много, вы, вероятно, будете тратить много памяти. Связанные методы обычно создаются только на короткое время их вызова, и затем они перестают существовать, когда автоматически собирается мусор. Если вы сделаете это вручную, у вас будет привязка имени, ссылающаяся на связанный метод - что предотвратит его сборку мусора при использовании.
  • Экземпляры объектов данного типа обычно имеют свои методы для всех объектов этого типа. Если вы добавите методы в другом месте, некоторые экземпляры будут иметь эти методы, а другие нет. Программисты этого не ожидают, и вы рискуете нарушить правило наименьшего удивления .
  • Так как есть и другие веские причины не делать этого, вы также дадите себе плохую репутацию, если будете это делать.

Таким образом, я предлагаю вам не делать этого, если у вас нет действительно веской причины. Гораздо лучше определить правильный метод в определении класса или, менее предпочтительно, напрямую связать его с классом, например, так:

Foo.sample_method = sample_method

Однако, поскольку это поучительно, я покажу вам несколько способов сделать это.

Как это можно сделать

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

class Foo(object):
    '''An empty class to demonstrate adding a method to an instance'''

Создайте экземпляр:

foo = Foo()

Создайте метод для добавления к нему:

def sample_method(self, bar, baz):
    print(bar + baz)

Метод naught (0) - использовать метод дескриптора, __get__

Пунктирный поиск функций вызывает __get__метод функции с экземпляром, связывая объект с методом и создавая, таким образом, «связанный метод».

foo.sample_method = sample_method.__get__(foo)

и сейчас:

>>> foo.sample_method(1,2)
3

Метод первый - types.MethodType

Сначала импортируем типы, из которых мы получим конструктор метода:

import types

Теперь мы добавляем метод к экземпляру. Для этого нам понадобится конструктор MethodType из typesмодуля (который мы импортировали выше).

Сигнатура аргумента для types.MethodType (function, instance, class):

foo.sample_method = types.MethodType(sample_method, foo, Foo)

и использование:

>>> foo.sample_method(1,2)
3

Способ второй: лексическое связывание

Сначала мы создаем функцию-оболочку, которая связывает метод с экземпляром:

def bind(instance, method):
    def binding_scope_fn(*args, **kwargs): 
        return method(instance, *args, **kwargs)
    return binding_scope_fn

Применение:

>>> foo.sample_method = bind(foo, sample_method)    
>>> foo.sample_method(1,2)
3

Метод третий: functools.partial

Неполная функция применяет первый аргумент (ы) к функции (и, необязательно, аргументы с ключевыми словами) и может быть позже вызвана с остальными аргументами (и с переопределением аргументов с ключевыми словами). Таким образом:

>>> from functools import partial
>>> foo.sample_method = partial(sample_method, foo)
>>> foo.sample_method(1,2)
3    

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

Несвязанная функция как атрибут объекта - почему это не работает:

Если мы попытаемся добавить образец_метода таким же образом, как мы могли бы добавить его к классу, он не связан с экземпляром и не принимает неявное «я» в качестве первого аргумента.

>>> foo.sample_method = sample_method
>>> foo.sample_method(1,2)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: sample_method() takes exactly 3 arguments (2 given)

Мы можем заставить работать несвязанную функцию, явно передавая экземпляр (или что-нибудь еще, поскольку этот метод фактически не использует selfпеременную аргумента), но это не будет соответствовать ожидаемой сигнатуре других экземпляров (если мы будем использовать патчи для обезьян) этот экземпляр):

>>> foo.sample_method(foo, 1, 2)
3

Вывод

Теперь вы знаете несколько способов , которыми Вы могли бы сделать это, но со всей серьезностью - не делайте этого.


1
Отказ от ответственности является то , что мне было интересно , о. Определение методов - это просто функции, вложенные в определение класса.
Atcold

1
@ Atcold Я подробно изложил причины, чтобы избежать этого во введении.
Аарон Холл

__get__Метод также необходим класс в качестве следующего параметра: sample_method.__get__(foo, Foo).
Аидас Бендорайтис

2
@AidasBendoraitis Я бы не сказал, что это «нужно», это необязательный параметр, который предоставляется при применении протокола дескриптора - но функции python не используют аргумент: github.com/python/cpython/blob/master/Objects/funcobject .c # L581
Аарон Холл

Мой комментарий основан на этой ссылке: python-reference.readthedocs.io/en/latest/docs/dunderdsc/… То, что я сейчас вижу из официальных документов, это необязательно: docs.python.org/3/howto/descriptor.html# протокол-дескриптор
Аидас Бендорайтис

35

Я думаю, что приведенные выше ответы упустили ключевой момент.

Давайте иметь класс с методом:

class A(object):
    def m(self):
        pass

Теперь давайте поиграем с этим в ipython:

In [2]: A.m
Out[2]: <unbound method A.m>

Итак, м () какой - то образом становится несвязанным методом А . Но так ли это на самом деле?

In [5]: A.__dict__['m']
Out[5]: <function m at 0xa66b8b4>

Оказывается, что m () - это просто функция, ссылка на которую добавляется в словарь классов A - в этом нет никакой магии. Тогда почему Am дает нам несвязанный метод? Это потому, что точка не переводится в простой поиск по словарю. Это де-факто вызов класса A .__ __.__ getattribute __ (A, 'm'):

In [11]: class MetaA(type):
   ....:     def __getattribute__(self, attr_name):
   ....:         print str(self), '-', attr_name

In [12]: class A(object):
   ....:     __metaclass__ = MetaA

In [23]: A.m
<class '__main__.A'> - m
<class '__main__.A'> - m

Теперь я не совсем уверен, почему последняя строка печатается дважды, но все же ясно, что там происходит.

Теперь то, что по умолчанию делает __getattribute__, это то, что он проверяет, является ли атрибут так называемым дескриптором или нет, то есть реализует ли он специальный метод __get__. Если он реализует этот метод, то возвращаемый результат является результатом вызова этого метода __get__. Возвращаясь к первой версии нашего класса А , вот что у нас есть:

In [28]: A.__dict__['m'].__get__(None, A)
Out[28]: <unbound method A.m>

А поскольку функции Python реализуют протокол дескриптора, если они вызываются от имени объекта, они связывают себя с этим объектом в своем методе __get__.

Итак, как добавить метод к существующему объекту? Предполагая, что вы не против исправления класса, это так просто, как:

B.m = m

Тогда Bm «становится» несвязанным методом благодаря магии дескриптора.

И если вы хотите добавить метод только к одному объекту, то вы должны сами эмулировать механизм, используя types.MethodType:

b.m = types.MethodType(m, b)

Кстати:

In [2]: A.m
Out[2]: <unbound method A.m>

In [59]: type(A.m)
Out[59]: <type 'instancemethod'>

In [60]: type(b.m)
Out[60]: <type 'instancemethod'>

In [61]: types.MethodType
Out[61]: <type 'instancemethod'>

20

В Python исправление обезьяны обычно работает путем перезаписи сигнатуры класса или функции вашей собственной. Ниже приведен пример из Zope Wiki :

from SomeOtherProduct.SomeModule import SomeClass
def speak(self):
   return "ook ook eee eee eee!"
SomeClass.speak = speak

Этот код перезапишет / создаст метод с именем talk в классе. В недавнем посте Джеффа Этвуда по исправлению обезьян . Он показывает пример в C # 3.0, который является текущим языком, который я использую для работы.


6
Но это влияет на все экземпляры класса, а не только на один.
glglgl

14

Вы можете использовать лямбду, чтобы привязать метод к экземпляру:

def run(self):
    print self._instanceString

class A(object):
    def __init__(self):
        self._instanceString = "This is instance string"

a = A()
a.run = lambda: run(a)
a.run()

Вывод:

This is instance string

9

Существует как минимум два способа присоединить метод к экземпляру без types.MethodType:

>>> class A:
...  def m(self):
...   print 'im m, invoked with: ', self

>>> a = A()
>>> a.m()
im m, invoked with:  <__main__.A instance at 0x973ec6c>
>>> a.m
<bound method A.m of <__main__.A instance at 0x973ec6c>>
>>> 
>>> def foo(firstargument):
...  print 'im foo, invoked with: ', firstargument

>>> foo
<function foo at 0x978548c>

1:

>>> a.foo = foo.__get__(a, A) # or foo.__get__(a, type(a))
>>> a.foo()
im foo, invoked with:  <__main__.A instance at 0x973ec6c>
>>> a.foo
<bound method A.foo of <__main__.A instance at 0x973ec6c>>

2:

>>> instancemethod = type(A.m)
>>> instancemethod
<type 'instancemethod'>
>>> a.foo2 = instancemethod(foo, a, type(a))
>>> a.foo2()
im foo, invoked with:  <__main__.A instance at 0x973ec6c>
>>> a.foo2
<bound method instance.foo of <__main__.A instance at 0x973ec6c>>

Полезные ссылки:
Модель данных - вызов дескрипторов
Descriptor HowTo Guide - вызов дескрипторов


7

То, что вы ищете, setattrя верю. Используйте это, чтобы установить атрибут объекта.

>>> def printme(s): print repr(s)
>>> class A: pass
>>> setattr(A,'printme',printme)
>>> a = A()
>>> a.printme() # s becomes the implicit 'self' variable
< __ main __ . A instance at 0xABCDEFG>

8
Это исправление класса A, а не экземпляра a.
Итан Фурман

5
Есть ли причина использовать setattr(A,'printme',printme)вместо просто A.printme = printme?
Тобиас Кинцлер

1
Это имеет смысл, если вы создаете имя метода во время выполнения.
rr-

6

Так как этот вопрос задан для не-Python версий, вот JavaScript:

a.methodname = function () { console.log("Yay, a new method!") }

5

Консолидация вики-ответов Джейсона Пратта и сообщества с обзором результатов различных методов привязки:

Особенно обратите внимание , как добавление функции привязки в качестве метода класса работ , но ссылки сферы неверна.

#!/usr/bin/python -u
import types
import inspect

## dynamically adding methods to a unique instance of a class


# get a list of a class's method type attributes
def listattr(c):
    for m in [(n, v) for n, v in inspect.getmembers(c, inspect.ismethod) if isinstance(v,types.MethodType)]:
        print m[0], m[1]

# externally bind a function as a method of an instance of a class
def ADDMETHOD(c, method, name):
    c.__dict__[name] = types.MethodType(method, c)

class C():
    r = 10 # class attribute variable to test bound scope

    def __init__(self):
        pass

    #internally bind a function as a method of self's class -- note that this one has issues!
    def addmethod(self, method, name):
        self.__dict__[name] = types.MethodType( method, self.__class__ )

    # predfined function to compare with
    def f0(self, x):
        print 'f0\tx = %d\tr = %d' % ( x, self.r)

a = C() # created before modified instnace
b = C() # modified instnace


def f1(self, x): # bind internally
    print 'f1\tx = %d\tr = %d' % ( x, self.r )
def f2( self, x): # add to class instance's .__dict__ as method type
    print 'f2\tx = %d\tr = %d' % ( x, self.r )
def f3( self, x): # assign to class as method type
    print 'f3\tx = %d\tr = %d' % ( x, self.r )
def f4( self, x): # add to class instance's .__dict__ using a general function
    print 'f4\tx = %d\tr = %d' % ( x, self.r )


b.addmethod(f1, 'f1')
b.__dict__['f2'] = types.MethodType( f2, b)
b.f3 = types.MethodType( f3, b)
ADDMETHOD(b, f4, 'f4')


b.f0(0) # OUT: f0   x = 0   r = 10
b.f1(1) # OUT: f1   x = 1   r = 10
b.f2(2) # OUT: f2   x = 2   r = 10
b.f3(3) # OUT: f3   x = 3   r = 10
b.f4(4) # OUT: f4   x = 4   r = 10


k = 2
print 'changing b.r from {0} to {1}'.format(b.r, k)
b.r = k
print 'new b.r = {0}'.format(b.r)

b.f0(0) # OUT: f0   x = 0   r = 2
b.f1(1) # OUT: f1   x = 1   r = 10  !!!!!!!!!
b.f2(2) # OUT: f2   x = 2   r = 2
b.f3(3) # OUT: f3   x = 3   r = 2
b.f4(4) # OUT: f4   x = 4   r = 2

c = C() # created after modifying instance

# let's have a look at each instance's method type attributes
print '\nattributes of a:'
listattr(a)
# OUT:
# attributes of a:
# __init__ <bound method C.__init__ of <__main__.C instance at 0x000000000230FD88>>
# addmethod <bound method C.addmethod of <__main__.C instance at 0x000000000230FD88>>
# f0 <bound method C.f0 of <__main__.C instance at 0x000000000230FD88>>

print '\nattributes of b:'
listattr(b)
# OUT:
# attributes of b:
# __init__ <bound method C.__init__ of <__main__.C instance at 0x000000000230FE08>>
# addmethod <bound method C.addmethod of <__main__.C instance at 0x000000000230FE08>>
# f0 <bound method C.f0 of <__main__.C instance at 0x000000000230FE08>>
# f1 <bound method ?.f1 of <class __main__.C at 0x000000000237AB28>>
# f2 <bound method ?.f2 of <__main__.C instance at 0x000000000230FE08>>
# f3 <bound method ?.f3 of <__main__.C instance at 0x000000000230FE08>>
# f4 <bound method ?.f4 of <__main__.C instance at 0x000000000230FE08>>

print '\nattributes of c:'
listattr(c)
# OUT:
# attributes of c:
# __init__ <bound method C.__init__ of <__main__.C instance at 0x0000000002313108>>
# addmethod <bound method C.addmethod of <__main__.C instance at 0x0000000002313108>>
# f0 <bound method C.f0 of <__main__.C instance at 0x0000000002313108>>

Лично я предпочитаю внешний маршрут функции ADDMETHOD, так как он позволяет мне динамически присваивать новые имена методов также внутри итератора.

def y(self, x):
    pass
d = C()
for i in range(1,5):
    ADDMETHOD(d, y, 'f%d' % i)
print '\nattributes of d:'
listattr(d)
# OUT:
# attributes of d:
# __init__ <bound method C.__init__ of <__main__.C instance at 0x0000000002303508>>
# addmethod <bound method C.addmethod of <__main__.C instance at 0x0000000002303508>>
# f0 <bound method C.f0 of <__main__.C instance at 0x0000000002303508>>
# f1 <bound method ?.y of <__main__.C instance at 0x0000000002303508>>
# f2 <bound method ?.y of <__main__.C instance at 0x0000000002303508>>
# f3 <bound method ?.y of <__main__.C instance at 0x0000000002303508>>
# f4 <bound method ?.y of <__main__.C instance at 0x0000000002303508>>

addmethodпереписано следующим образом def addmethod(self, method, name): self.__dict__[name] = types.MethodType( method, self )решает проблему
Энтони Хаткинс

5

Это на самом деле дополнение к ответу "Джейсон Пратт"

Хотя ответ Jasons работает, он работает, только если кто-то хочет добавить функцию в класс. Это не сработало для меня, когда я попытался перезагрузить уже существующий метод из файла исходного кода .py.

У меня ушло целую вечность, чтобы найти обходной путь, но хитрость кажется простой ... 1. импортировать код из файла исходного кода 2. и принудительно выполнить перезагрузку 3.rd использовать types.FunctionType (...) для преобразования импортированный и привязанный метод к функции, вы также можете передать текущие глобальные переменные, так как перезагруженный метод будет находиться в другом пространстве имен 4. Теперь вы можете продолжить, как предложено "Jason Pratt", используя types.MethodType (... )

Пример:

# this class resides inside ReloadCodeDemo.py
class A:
    def bar( self ):
        print "bar1"

    def reloadCode(self, methodName):
        ''' use this function to reload any function of class A'''
        import types
        import ReloadCodeDemo as ReloadMod # import the code as module
        reload (ReloadMod) # force a reload of the module
        myM = getattr(ReloadMod.A,methodName) #get reloaded Method
        myTempFunc = types.FunctionType(# convert the method to a simple function
                                myM.im_func.func_code, #the methods code
                                globals(), # globals to use
                                argdefs=myM.im_func.func_defaults # default values for variables if any
                                ) 
        myNewM = types.MethodType(myTempFunc,self,self.__class__) #convert the function to a method
        setattr(self,methodName,myNewM) # add the method to the function

if __name__ == '__main__':
    a = A()
    a.bar()
    # now change your code and save the file
    a.reloadCode('bar') # reloads the file
    a.bar() # now executes the reloaded code

3

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

Использование функции needle()для исправления названного модуля guineapigпроисходит следующим образом:

import gorilla
import guineapig
@gorilla.patch(guineapig)
def needle():
    print("awesome")

Но он также заботится о более интересных случаях использования, как показано в FAQ из документации .

Код доступен на GitHub .


3

Этот вопрос был открыт несколько лет назад, но, эй, есть простой способ симулировать привязку функции к экземпляру класса с помощью декораторов:

def binder (function, instance):
  copy_of_function = type (function) (function.func_code, {})
  copy_of_function.__bind_to__ = instance
  def bound_function (*args, **kwargs):
    return copy_of_function (copy_of_function.__bind_to__, *args, **kwargs)
  return bound_function


class SupaClass (object):
  def __init__ (self):
    self.supaAttribute = 42


def new_method (self):
  print self.supaAttribute


supaInstance = SupaClass ()
supaInstance.supMethod = binder (new_method, supaInstance)

otherInstance = SupaClass ()
otherInstance.supaAttribute = 72
otherInstance.supMethod = binder (new_method, otherInstance)

otherInstance.supMethod ()
supaInstance.supMethod ()

Там, когда вы передаете функцию и экземпляр декоратору связывателя, она создаст новую функцию с тем же объектом кода, что и первый. Затем данный экземпляр класса сохраняется в атрибуте вновь созданной функции. Декоратор возвращает (третью) функцию, вызывающую автоматически скопированную функцию, давая экземпляр в качестве первого параметра.

В заключение вы получаете функцию, имитирующую привязку к экземпляру класса. Оставив исходную функцию без изменений.


2

То, что написал Джейсон Пратт, верно.

>>> class Test(object):
...   def a(self):
...     pass
... 
>>> def b(self):
...   pass
... 
>>> Test.b = b
>>> type(b)
<type 'function'>
>>> type(Test.a)
<type 'instancemethod'>
>>> type(Test.b)
<type 'instancemethod'>

Как видите, Python не считает b () чем-то отличным от a (). В Python все методы - это просто переменные, которые являются функциями.


7
Вы исправляете класс Test, а не его экземпляр.
Итан Фурман

Вы добавляете метод к классу, а не к экземпляру объекта.
TomSawyer

2

Мне кажется странным, что никто не упомянул, что все перечисленные выше методы создают ссылку на цикл между добавленным методом и экземпляром, в результате чего объект остается постоянным до сборки мусора. Был старый трюк, добавляющий дескриптор путем расширения класса объекта:

def addmethod(obj, name, func):
    klass = obj.__class__
    subclass = type(klass.__name__, (klass,), {})
    setattr(subclass, name, func)
    obj.__class__ = subclass

Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.