Dekorator do ustawiania atrybutów funkcji

Question

Chcę, aby różne funkcje były wykonywane tylko wtedy, gdy zalogowany użytkownik ma wymagany poziom uprawnień.

Aby moje życie było bardziej złożone, po prostu chcę użyć dekoratorów. Poniżej uwidoczniam ustawienie atrybutu permission dla funkcji "dekorowanych" - jak pokazano poniżej.

def permission(permission_required): 
    def wrapper(func): 
     def inner(*args, **kwargs): 
      setattr(func, 'permission_required', permission_required) 
      return func(*args, **kwargs) 
     return inner 
    return wrapper 

@permission('user') 
def do_x(arg1, arg2): 

    ... 

@permission('admin') 
def do_y(arg1, arg2): 
    ... 

Ale kiedy zrobić:

fn = do_x 
if logged_in_user.access_level == fn.permission_required: 
    ... 

otrzymuję błąd 'function' object has no attribute 'permission_required'

Czego mi brakuje?

Answer 1

Ustawiasz atrybut w wewnętrznej (opakowaniu) funkcji. Nie trzeba funkcję otoki wcale:

def permission(permission_required): 
    def decorator(func): 
     func.permission_required = permission_required 
     return func 
    return decorator 

Twój dekorator musi powrócić coś że będzie zastąpić pierwotną funkcję. Oryginalna funkcja sama w sobie (z dodanym atrybutem) zrobi to dobrze, ponieważ wszystko, co chcesz zrobić, to dodać do niej atrybut.

Jeśli trzeba jeszcze owijkę, a następnie ustawić atrybut na funkcję owijarki Zamiast:

def permission(permission_required): 
    def decorator(func): 
     def wrapper(*args, **kwargs): 
      # only use a wrapper if you need extra code to be run here 
      return func(*args, **kwargs) 
     wrapper.permission_required = permission_required 
     return wrapper 
    return decorator 

Wszakże wymieniasz owinięte funkcji z owijki zwracanej przez dekoratora, tak to jest obiekt, na który będziesz szukać atrybutu.

Answer 2

Twój dekorator powinien powrócić funkcję, która może zastąpić do_x lub do_y, nie zwraca wynik realizacji wynosi do_x lub do_y Można modity ozdobić jak poniżej:

def permission(permission_required): 
    def wrapper(func): 
     def inner(): 
      setattr(func, 'permission_required', permission_required) 
      return func 
     return inner() 
    return wrapper 

oczywiście masz inny krótki rozwiązanie :

def permission(permission_required): 
    def wrapper(func): 
     setattr(func, 'permission_required', permission_required) 
     return func 
    return wrapper