Décorateur qui ajoute un paramètre de mot-clé

Question

J'essaie de définir un décorateur qui ajoute simplement un argument mot-clé à la fonction décorée f. f peut avoir n'importe quelle combinaison de paramètres (positionnel, mot-clé, etc.). J'ai essayé ceci:

def capture_wrap(f): 
    def captured(name=False, *args, **kwargs): 
     """name can be False, True or str. if str, then use it as a name.""" 
     print (name, f, args, kwargs) 
     inner = f(*args, **kwargs) 
     if name is False: 
      return inner 
     elif name is True: 
      return capture(inner) 
     else: 
      return capture(inner, name=name) 
    return captured 

Si je tente de l'utiliser avec une fonction qui accepte un seul argument:

@capture_wrap 
def any_of(s): 
    """s must be in the right format. 
    See https://docs.python.org/3/library/re.html#regular-expression-syntax .""" 
    return wrap('[', Dinant(s, escape=False), ']') 

0-9A-Fa-f <function any_of at 0x7fb370548a60>() {} 
Traceback (most recent call last): 
File "./dinant.py", line 222, in <module> 
    hex = one_or_more(any_of('0-9A-Fa-f')) 
File "./dinant.py", line 116, in captured 
    inner = f(*args, **kwargs) 
TypeError: any_of() missing 1 required positional argument: 's' 

je reçois une erreur parce que le tout argument devient le paramètre name dans captured() . Comment dois-je le faire correctement?

Answer 1

Il suffit de passer le nom où vous appelez la fonction f = f intérieure (nom, * args, ** kwargs)

Answer 2

Vous pouvez faire un namekeyword-only argument pour y parvenir. Actuellement le problème est que c'est un argument de position et quand vous faites any_of('foo') le 'foo' est passé à l'argument name.

def capture_wrap(f): 
    def captured(*args, name=False, **kwargs): 
     """name can be False, True or str. if str, then use it as a name.""" 
     print(name, f, args, kwargs) 
     inner = f(*args, **kwargs) 
     return inner 
    return captured 


@capture_wrap 
def any_of(s1, s2, s3, **kwargs): 
    return "I do something" 

Démo:

>>> any_of('0-9A-Fa-f', 1, 2) 
False <function any_of at 0x1041daea0> ('0-9A-Fa-f', 1, 2) {} 
'I do something' 

>>> any_of('0-9A-Fa-f', 2, 3, a=1, b=2, name='some-name') 
some-name <function any_of at 0x1041daea0> ('0-9A-Fa-f', 2, 3) {'a': 1, 'b': 2} 
'I do something'