Я пытаюсь написать класс, который будет перехватывать любой общий вызов функции, а затем выполнять с ним некоторые операции.
В качестве примера попробую использовать ответ на аналогичный вопрос от здесь.
Однако он выдает ошибку AttributeError: 'method' object has no attribute '__name__'
, когда я пытаюсь изменить имя метода, назначив атрибут __name__
.
Используемый мной код, адаптированный из пример выше:
class A():
def __init__(self):
self.x = 1 # set some attribute
def __getattr__(self,attr):
return self.__get_global_handler(attr)
def __get_global_handler(self, name):
# Do anything that you need to do before simulating the method call
handler = self.__global_handler
handler.__name__ = name # Change the method's name
return handler
def __global_handler(self, *args, **kwargs):
# Do something with these arguments
print("I am an imaginary method with name %s" % self.__global_handler.__name__)
print("My arguments are: " + str(args))
print("My keyword arguments are: " + str(kwargs))
def real_method(self, *args, **kwargs):
print("I am a method that you actually defined")
print("My name is %s" % self.real_method.__name__)
print("My arguments are: " + str(args))
print("My keyword arguments are: " + str(kwargs))
Он отлично работает, если я прокомментирую строку, которая меняет имя метода (handler.__name__ = name # Change the method's name
):
>>> a.imaginary_method
<bound method A.__global_handler of <__main__.A object at 0x00000150E64FC2B0>>
>>> a.imaginary_method(1, 2, x=3, y=4)
I am an imaginary method with name __global_handler
My arguments are: (1, 2)
My keyword arguments are: {'x': 3, 'y': 4}
Однако, если я раскомментирую строку (чтобы принудительно изменить имя), я получаю:
>>> a.imaginary_method
[...]
AttributeError: 'method' object has no attribute '__name__'
Я ожидал получить что-то вроде:
>>> a.imaginary_method
<bound method A.imaginary_method of <__main__.A object at 0x00000150E64FC2B0>>
>>> a.imaginary_method(1, 2, x=3, y=4)
I am an imaginary method with name imaginary_method
My arguments are: (1, 2)
My keyword arguments are: {'x': 3, 'y': 4}
Итак, есть ли способ изменить имя метода на лету? Большое спасибо!
Причина, по которой handler.__name__ = name
не работает, заключается в том, что handler
является связанный метод, то есть это объект, который инкапсулирует ссылку на функцию и экземпляр, который будет принимать значение self
. Вы можете подтвердить это, распечатав его; он говорит "связанный метод":
>>> a.imaginary_method
<bound method A.__global_handler of <__main__.A object at 0x00000150E64FC2B0>>
Вы можете получить доступ к базовой функции через атрибут __func__
и изменить ее имя:
def __get_global_handler(self, name):
handler = self.__global_handler
handler.__func__.__name__ = name # Change the method's name
handler.__func__.__qualname__ = __class__.__qualname__ + '.' + name
return handler
Однако проблема с этим подходом заключается в том, что все «воображаемые методы» на самом деле относятся к одному и тому же методу __global_handler
. Если вы попытаетесь создать два метода с разными именами, вы увидите, что на самом деле они оба имеют одно и то же имя (потому что это один и тот же метод):
a = A()
foo = a.foo
bar = a.bar
print(foo) # <bound method A.bar of <__main__.A object at 0x00000212AFCC7198>>
print(bar) # <bound method A.bar of <__main__.A object at 0x00000212AFCC7198>>
Таким образом, лучшим решением будет каждый раз создавать новую функцию:
class MethodFactory:
def __getattr__(self, name):
def func(*args, **kwargs):
print("I am an imaginary method with name", name)
print("My arguments are:", args)
print("My keyword arguments are:", kwargs)
func.__name__ = name
func.__qualname__ = __class__.__qualname__ + '.' + name
return func
a = MethodFactory()
foo = a.foo
bar = a.bar
print(foo) # <function MethodFactory.foo at 0x00000238BDADC268>
print(bar) # <function MethodFactory.bar at 0x00000238BF8AB730>
Как видите, побочным эффектом этой реализации является то, что foo
и bar
являются обычными функциями, а не связанными методами. Если по какой-то причине вам нужно вернуть связанные методы, вы можете вручную преобразовать функцию в связанный метод, вызвав ее метод __get__
:
class MethodFactory:
def __getattr__(self, name):
def func(self, *args, **kwargs):
print("I am an imaginary method with name", name)
print("My arguments are:", args)
print("My keyword arguments are:", kwargs)
func.__name__ = name
func.__qualname__ = __class__.__qualname__ + '.' + name
return func.__get__(self, type(self))
a = MethodFactory()
foo = a.foo
bar = a.bar
print(foo) # <bound method MethodFactory.foo of <__main__.MethodFactory object at 0x00000137BF0C29E8>>
print(bar) # <bound method MethodFactory.bar of <__main__.MethodFactory object at 0x00000137BF0C29E8>>