Как работает в данном примере декорирование?

Question

[Ohotnikx]

Нарыл на статье хабра примерно похожий код на тему декораторов:

def decor(func): 
    def _wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        return result
    return _wrapper 

class MyClass:
    def complex_calculation(self):
        return 42

MyClass = decor(MyClass)

my_obj = MyClass()

print(my_obj.complex_calculation())

Я пытался в нем разобраться но стопорился об одно: в строке my_obj = MyClass() вроде как создается экземпляр класса MyClass. Однако строкой выше происходит декорирование и MyClass становится ссылкой на функцию _wrapper. Т.е. по сути, в этой строке должен в качестве аргумента приниматься класс, чтобы result стал экземпляром класса, но класс как аргумент передавался только при вызове decor() строкой еще выше и (наверное) не играл никакой роли. Заранее спасибо!

Comments

[Влад Григорьев]

работает точно также как и обычное декорирование, ибо @decor есть тоже самое что decor(_class)
ну и чудес не бывает, все для чего то но нужно как правило.
Ну и что б понятно учится надо учится, не по статьям, а по учебникам

[Сергей П]

Влад Григорьев, ться=)

Answer 1

[Максим Припадчев]

Не очень понятно, а что не понятно. Передается объект класса в функцию, ты зачем назвал параметр функции func а передаешь туда объект класса. Вот здесь result = func(*args, **kwargs) ты инициализируешь экземпляр класса, затем возвращаешь его да и все, MyClass у тебя стала функцией которая возвращает экземпляр класса. Ну добавь там чего нибудь например.

def f():
    return 100
    
def decor(func): 
    def _wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        result.complex_calculation = f
        return result
    return _wrapper 

class MyClass:
    def complex_calculation(self):
        return 42

MyClass = decor(MyClass)

my_obj = MyClass()

print(my_obj.complex_calculation())

Твой метод вернет, не 42, а 100. Или объяви метод какой-нибудь которого нет.

Comments

[Влад Григорьев]

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

[Максим Припадчев]

Влад Григорьев, print(type(MyClass)) - function

[Влад Григорьев]

Максим Припадчев, и шо, а функция перестал быть классом?

def func():
    pass
type(func)
<class 'function'>

[Максим Припадчев]

Влад Григорьев, Не ну в этом смысле нам вообще надо отказать от слова функция тогда потому что все является объектом.

[Влад Григорьев]

Максим Припадчев, да и если не докапываться. класс остался классом, то что на него навешали декоратор, не отменяет факта, что он остался классом.
а если докапываться, то почти все будет являться объектом.

[Максим Припадчев]

Влад Григорьев, ты чего хочешь я не пойму мы что говорим не функция а экземпляр объекта класса функция?

[Влад Григорьев]

Максим Припадчев, MyClass = decor(MyClass) - в данной строчке, мы просто декорируем класс, никакого экземпляра тут не появляется, это тоже самое, что и повесить декоратор над классом.

[Максим Припадчев]

Влад Григорьев, decor(MyClass) вот этот объект function. MyClass становится указателем на него. Я имел ввиду вот это и это очевидно из написанного.

[Максим Припадчев]

Влад Григорьев, да мы можем использовать иной синтакс, но в этой строчке MyClass у него становится ссылкой на объект типа function.

[Ohotnikx]

В виду своего незнания дальше я возможно буду нести бред, за что извиняюсь, но вижу я код так: после декорирования MyClass = decor(MyClass) MyClass стала функцией и ее функционал такой же, как и у _wrapper, т.е.

result = func(*args, **kwargs) result.complex_calculation = f return result

И как вы и сказали, она возвращает инициализированный экземпляр класса, что и следует из строки ниже. Только мне не совсем ясно, как она возвращает экземпляр, если в аргументы функции сам класс не передается? Ибо MyClass передавался в качестве аркамента как класс только в MyClass = decor(MyClass).

[Влад Григорьев]

Максим Припадчев, если душнить то душнить до конца, или не душнить вовсе.
В питоне есть модуль dataclasses и декоратор dataclass. который в сущности делает, ровно тоже что и описали вы, но при этом в документации говорится, что возвращает класс обогащенный методами.
Я нигде не встречал, что б в питоне при использование декораторов, начинали говорить что они меняют свой тип. Хотя так оно и есть.
А там где рассматривают что мерняется тип, то уже рассматривают всю цепочку вызовов целиком, а не только последние 3

[Максим Припадчев]

Ohotnikx, Не очень мне понятно, о чем ты спрашиваешь.
1. есть функция decor, которая возвращает внутреннею функцию.

2. есть внутренняя функция _wrapper которая возвращает экземпляр класса MyClass

3. сначала имя "MyClass' указывает на объект класса MyClass (Когда ты пишешь class MyClass: pass) он будет исполнен это объект, который при его объявлении, а не при вызове. Попробуй например

class BadClass:
    a = 1/0

тебе не надо инициализировать экземпляр класса, ты получишь ошибку. Потому что код будет выполнен, в отличии от функции например (там только при вызове).

4. после вот это строчки MyClass = decor(MyClass) имя 'MyClass' это указатель на объект внутренней функции. Иными словами 'MyClass = _wrapper'
затем ты вызываешь MyClass(), а вызывается _wrapper то есть когда ты пишешь MyClass() это тоже самое _wrapper(*args, **kwargs). _wrapper в свою очередь имеет доступ к объект класса который мы объявляли изначально, вот в этой части decor(MyClass) мы передаем объект класса. То есть он доступен для _wrapper в расширенной но не глобальной области видимости. Вот здесь ты инициализируешь экземпляр func(*args, **kwargs), я тебе еще замечание насчет имени func сделал, так и запутаться не долго, у тебя происходит result = MyClass(). То есть создание экземпляра твоего класса, затем ты потенциально декорируешь его "увеличиваешь функциональность например" и возвращаешь. Все.

Ты не обязан вот в этой строчке MyClass = decor(MyClass) давать имя ссылке на объект функции _wrapper 'MyClass'. это сделано, что бы вызов не отличался. Для понимания ну сделай например так.

My_Decorated_Class_builder = decor(MyClass)

decorated_class_instances = My_Decorated_Class_builder()
normal_class_instance = MyClass()
print(decorated_class_instances.complex_calculation())
print(normal_class_instance.complex_calculation())

Что еще не понятно?

[Максим Припадчев]

Влад Григорьев, Душнить это ты пытался, я даже не заострял внимание на твоих комментариях, не заостряю и сейчас. Перечитай свои комментарии на свежую голову. Если же ты все еще настроен продолжать дискуссию. Тогда конкретно и по существу, в чем я не прав, где я ошибся в ответе?

[Сергей П]

Влад Григорьев,

и шо, а функция перестал быть классом?

Вот ключевая ошибка.
Функция - это не класс, а инстанс класса `function`.
После декорирования класса мы получаем именно функцию. По факту же происходит эффект, идентичный декорированию привязанного метода __call__ нашего класса.
Так что если докапываться, то корректнее сказать, что декорируется именно функция, и на выходе тоже получается функция, которая возвращает идентичный результат.
Но зачем играть словами и докапываться, когда нужно просто понимать что происходит, и всё?