Dekorator zum Festlegen von Attributen der Funktion

Question

Ich möchte verschiedene Funktionen nur ausführen, wenn der angemeldete Benutzer die erforderliche Berechtigungsstufe hat.

Um mein Leben komplexer zu machen, möchte ich einfach Dekorateure verwenden. Im Folgenden habe ich versucht, das Attribut permission auf "dekorierte" Funktionen zu setzen - wie unten gezeigt.

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): 
    ... 

Aber wenn ich tue:

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

Ich erhalte einen Fehler 'function' object has no attribute 'permission_required'

Was bin ich?

Answer 1

Sie setzen das Attribut in der inneren (Wrapper) -Funktion. Sie brauchen nicht eine Wrapper-Funktion bei allen:

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

Ihr Dekorateur muss etwas zurück, dass die ursprüngliche Funktion ersetzen werden. Die ursprüngliche Funktion selbst (mit dem hinzugefügten Attribut) ist dafür geeignet, da Sie nur ein Attribut hinzufügen möchten.

Wenn Sie noch einen Wrapper benötigen, dann setzen Sie das Attribut auf die Wrapper-Funktion statt:

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 

Immerhin ersetzen Sie die verpackte Funktion mit dem Wrapper von dem Dekorateur zurückgegeben, so dass das sind Objekt suchen Sie nach dem Attribut auf.

Answer 2

Ihr Dekorateur sollte eine Funktion zurück, die do_x oder do_y ersetzen kann, nicht das Ausführungsergebnis von do_x zurückgeben oder do_y Sie können Sie wie unten modity dekorieren:

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

Natürlich haben Sie eine weitere kurze Lösung :

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