dekoratör işlevinin niteliklerini ayarlamak için

Question

Farklı işlevlerin, yalnızca oturum açmış kullanıcının gerekli izin düzeyine sahip olması durumunda yürütülebilir olmasını istiyorum.

Hayatımı daha karmaşık hale getirmek için dekoratörler kullanmak istiyorum. Aşağıda, aşağıda gösterildiği gibi 'dekore edilmiş' işlevler üzerinde permission özniteliğini ayarlamak için dikkatle çalışıyorum.

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

Ama ne zaman:

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

bir hata 'function' object has no attribute 'permission_required'

ben eksik olsun?

Answer 1

Özniteliği iç (sarmalayıcı) işlevinde ayarlıyorsunuz. Hepiniz bir sarıcı işlev gerekmez:

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

Kişisel dekoratör orijinal fonksiyonunu değiştiririzşey dönmek gerekiyor. Özgün işlevin kendisi (özellik eklendiğinde) bunun için iyi olacaktır, çünkü tüm yapmak istediğiniz bir özellik eklemektir.

hala sarmalayıcı gerekiyorsa, o zaman bunun yerine sarıcı işlevi üzerine niteliğini ayarlayın: o yüzden Sonuçta

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 

, sen dekoratör tarafından döndürülen sargı ile sarılmış işlevi değiştiriyorsanız nesne üzerinde özellik arıyor olacaksınız.

Answer 2

do_x veya do_y, do_x veya do_y yürütülmesi sonuç döndürmez yerini alabilir bir işlev dönmelidir Kişisel dekoratör Sen olarak aşağıda süslemek modity edebilirsiniz: Elbette

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

, başka kısa çözümü var : bir yan not olarak

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