L'analyse des arguments et kwargs dans les décorateurs

J'ai une fonction qui prend en arguments et kwargs, et j'ai besoin de faire quelque chose dans ma décorateur basé sur la valeur de la 2e arg dans la fonction, comme dans le code ci-dessous:

def workaround_func():
    def decorator(fn):
        def case_decorator(*args, **kwargs):
            if args[1] == 2:
                print('The second argument is a 2!')
            return fn(*args, **kwargs)
        return case_decorator
    return decorator

@workaround_func()
def my_func(arg1, arg2, kwarg1=None):
    print('arg1: {} arg2: {}, kwargs: {}'.format(arg1, arg2, kwarg1))

Le problème est que python permet aux utilisateurs d'appeler la fonction avec le deuxième argument comme un régulier d'un argument OU d'un mot-clé-argument, donc si l'utilisateur appelle my_func avec arg2 comme un kwarg, il soulève une IndexError, voir ci-dessous:

In [8]: d.my_func(1, 2, kwarg1=3)
The second argument is a 2!
arg1: 1 arg2: 2, kwargs: 3

In [9]: d.my_func(1, arg2=2, kwarg1=3)
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-9-87dc89222a9e> in <module>()
----> 1 d.my_func(1, arg2=2, kwarg1=3)

/home/camsparr/decoratorargs.py in case_decorator(*args, **kwargs)
      2     def decorator(fn):
      3         def case_decorator(*args, **kwargs):
----> 4             if args[1] == 2:
      5                 print('The second argument is a 2!')
      6             return fn(*args, **kwargs)

IndexError: tuple index out of range

Est-il un moyen de contourner cela sans faire juste un try/except et attraper le IndexError?

OriginalL'auteur Cameron Sparr | 2013-09-19

  1. 5

    C'est le plus robuste de la façon que je peux penser à le manipuler... Le truc, c'est d'inspecter le nom de la deuxième argument. Ensuite, dans le décorateur, vous vérifiez pour voir si le nom est présent dans kwargs. Si oui, alors vous pouvez l'utiliser. Si non, alors vous utilisez args.

    from inspect import getargspec    

def decorate(fn):
    argspec = getargspec(fn)
    second_argname = argspec[0][1]
    def inner(*args, **kwargs):
        special_value = (kwargs[second_argname] 
                         if second_argname in kwargs else args[1])
        if special_value == 2:
            print "foo"
        else:
            print "no foo for you"
        return fn(*args, **kwargs)
    return inner

@decorate
def foo(a, b, c=3):
    pass

foo(1,2,3)
foo(1,b=2,c=4)
foo(1,3,5)
foo(1,b=6,c=5)

    l'exécution de cette résultats dans:

    foo
foo
no foo for you
no foo for you

    comme prévu.

    OriginalL'auteur mgilson

  2. 4

    J'ai trouvé une réponse à l'aide du python decorator paquet. Une caractéristique de ce paquet est qu'il préserve de position/mot-clé args peu importe la façon dont l'utilisateur passe. Il a l'avantage supplémentaire de réduire la quantité de code, donc mon code d'origine:

    def workaround_func():
    def decorator(fn):
        def case_decorator(*args, **kwargs):
            if args[1] == 2:
                print('The second argument is a 2!')
            return fn(*args, **kwargs)
        return case_decorator
    return decorator

@workaround_func()
def my_func(arg1, arg2, kwarg1=None):
    print('arg1: {} arg2: {}, kwargs: {}'.format(arg1, arg2, kwarg1))

    devient:

    from decorator import decorator

@decorator
def workaround_decorator(f, *args, **kwargs):
    if args[1] == 2:
        print('The second argument is 2!')
    return f(*args, **kwargs)

@workaround_decorator
def my_func(arg1, arg2, kwarg1=None):
    print('arg1: {} arg2: {}, kwargs: {}'.format(arg1, arg2, kwarg1))

    OriginalL'auteur Cameron Sparr

  3. -1

    Il semble que arg1 est un argument obligatoire, mais arg2 peut être facultatif. Vous pouvez modifier la ligne 3:

        def case_decorator(arg1, arg2=None, *args, **kwargs):
                            # or =0 or ='' whichever best

    et ligne 6:

            return fn(arg1, arg2, *args, **kwargs)

    Si arg1 est aussi disponible en option, modifiez la ligne 3 avec arg1= "None" ou " =0 ou ="

    J'ai oublié de mentionner la source où j'ai trouvé un aperçu: tutorialspoint.com/python/python_functions.htm
    Vous pouvez passer de position args par mot-clé si vous le souhaitez. Cela ne signifie pas que l'argument est facultatif (ce qui signifie vraiment qu'il a une valeur par défaut).
    tcaswell a droite, il n'est pas que arg2 est facultatif, c'est que les gens peuvent passer de position args comme mot-clé args en python.

    OriginalL'auteur chapelo