Zdobywając metody klasy Pythona, w jaki sposób przekazać instancję do dekoratora?

Question

To jest python 2.5, to jest GAE też nie, że to ma znaczenie.

Mam następujący kod, jestem dekorowanie metody foo() w bar, przy użyciu klasy dec_check jako dekorator.

class dec_check(object): 

    def __init__(self, f): 
    self.func = f 

    def __call__(self): 
    print 'In dec_check.__init__()' 
    self.func() 

class bar(object): 

    @dec_check 
    def foo(self): 
    print 'In bar.foo()' 

b = bar() 
b.foo() 

Podczas wykonywania tego miałem nadzieję zobaczyć:

In dec_check.__init__() 
In bar.foo() 

Ale ja dostaję „TypeError: foo() takes exactly 1 argument (0 given)” jako .foo(), jako metoda obiektu, bierze siebie jako argument. Zgaduję, że problem polega na tym, że instancja bar faktycznie nie istnieje, gdy wykonuję kod dekoratora.

Jak przekazać instancję klasy bar do klasy dekoratora?

Answer 1

Trzeba zrobić dekorator do descriptor - albo poprzez zapewnienie jej (meta) klasa ma __get__ metody, lub, sposób prostszy, za pomocą dekoratora funkcję zamiast dekorator klasa (ponieważ funkcje są już deskryptorami). Np .:

def dec_check(f): 
    def deco(self): 
    print 'In deco' 
    f(self) 
    return deco 

class bar(object): 
    @dec_check 
    def foo(self): 
    print 'in bar.foo' 

b = bar() 
b.foo() 

to drukuje

In deco 
in bar.foo 

to pożądane.

Answer 2

Dekoratory mogą być nieco trudne w Pythonie. Oto przykład, z którego kiedyś wróciłem. Możesz być w stanie wywnioskować z tego, co próbujesz zrobić.

Django authentication and Ajax - URLs that require login

Answer 3

Odpowiedź Alex wystarczy, gdy funkcja jest wystarczająca. Jednak gdy potrzebujesz klasy, możesz sprawić, by działała, dodając następującą metodę do klasy dekoratora.

def __get__(self, obj, objtype): 
    """Support instance methods.""" 
    import functools 
    return functools.partial(self.__call__, obj) 

Aby to zrozumieć, należy zrozumieć protokół deskryptora. Protokół deskryptora jest mechanizmem wiązania rzeczy z instancją. Składa się z __ get __, __ set __ i __ delete __, które są wywoływane, gdy rzecz jest ustawiona, lub usunięta ze słownika instancji. W tym przypadku, gdy rzecz pochodzi z instancji, wiążemy pierwszy argument jego metody __ call do instancji, używając częściowego. Robi się to automatycznie dla funkcji składowych, gdy klasa jest konstruowana, ale dla funkcji syntetycznego elementu, takiej jak ta, musimy to zrobić jawnie.

Answer 4

jeśli chcesz napisać dekoratora jako klasę, którą możesz zrobić.

from functools import update_wrapper, partial 

MyDecorator(object): 
    def __init__(self, func): 
     update_wrapper(self, func) 
     self.func = func 

    def __get__(self, obj, objtype): 
     """Support instance methods.""" 
     return functools.partial(self.__call__, obj) 

    def __call__(self, obj, *args, **kwargs): 
     print('Logic here') 
     return self.func(obj, *args, **kwargs) 

my_decorator = MyDecorator 

class MyClass(object): 
    @my_decorator 
    def my_method(self): 
     pass