Более элегантный способ для этого декоратора python

Question

В принципе, я хочу, чтобы декоратор имел список аргументов, который содержит больше, чем просто функцию, которая может быть вызвана как из @ -, так и из обычных форм. Я «разработал» быстрое обходное решение, но он уродлив и немедленно выполняет функцию в @ -form, что является нежелательным побочным эффектом (это связано с возвратом body_two(), конечно).

def status_display_with_comment(comment, closure = None): 
    def body_one(function = None): 
     def body_two(): 
      print(comment) 
      #an ugly workaround to be able to run both the @- and regular forms 
      if function != None: 
       print("Entering", function.__name__) 
       function() 
       print("Exited", function.__name__) 
      elif closure != None: 
       print("Entering", closure.__name__) 
       closure() 
       print("Exited", closure.__name__) 
     return body_two() 
    return body_one 

def a_function(): 
    print('a_function executes') 

@status_display_with_comment(comment = 'some comment') 
def a_function_with_comment(): 
    print('a_function_with_comment executes') 

a_function_status_display_with_comment = status_display_with_comment(closure = a_function, comment = 'a comment') 

a_function_status_display_with_comment() 

Заранее спасибо.

P.S .: Я должен обернуть голову вокруг этой цели. Это интересно, учитывая, что это можно сделать рекурсивно, как в Scheme (давно для меня).

Answer 1

Вы хотите функцию, которая возвращает декоратора:

def status_display_with_comment(comment): 
    def decorator(function): 
     def wrapper(): 
      print(comment) 
      print("Entering", function.__name__) 
      result = function() 
      print("Exited", function.__name__) 
      return result 
     return wrapper 

    return decorator 


def a_function(): 
    print('a_function executes') 



a_function_SD_WC = status_display_with_comment('a comment')(a_function) 
a_function_SD_WC() 

работает также:

@status_display_with_comment('a comment') 
def a_function(): 
    print('a_function executes') 


a_function() 

---

Регулярное прямое декоратор уже возвращает замыкание:

def a_normal_decorator(function): 
    def wrapper(): 
     return function() 
    return wrapper 

wrapper здесь закрытие, так как оно должно содержать function вокруг даже после завершения a_normal_decorator.

---

Для справки, это как идиоматическое декоратор обычно пишется:

import functools 

def decorator(function): 
    @functools.wraps(function) 
    def wrapper(*a, **kw): 
     return function(*a, **kw) 

    return wrapper 

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

functools.wraps копии из функции завернутые в функцию обертки __name__, __module__, __annotations__ и __doc__ строка документации.