使用Python

时间:2018-02-03 03:14:45

标签: python arguments args kwargs

我一直在寻找这个问题的答案,但我总是最终看到涵盖基础知识的教程。我的问题是 -

如果我有一个功能:

def foo(a, b, *args, **kwargs):

    if args[0]:
        if args[0] in range(1, 4):
            x=args[0]
        else:
            raise ValueError("Eww that is one ugly x!")
    else:
        x = kwargs.get('x', 3)
    if args[1]:
        if args[1] in ['some','list','of','strings']:
             y = args[1]
        else:
           raise ValueError("Invalid y")
    else:        
        y = kwargs.get('y', "some")

    if x == 1:
       print("Good")
    elif x == 2:
       print("Bad")
    elif x == 3:
       print("Clint")
    else:
       raise ValueError("Eww that is one ugly x!")

    if y == 'some':
       print(y + str(x))
    elif y == 'list':
       print("happy")
    elif y == 'of':
       print("healthy")
    elif y == 'strings':
       print(y + 'me')
    else:
       raise ValueError("Invalid y")

我正在寻找一种更简单的方法来将args [0]和kwargs.get('x')视为等效的 - 只要我想执行相同的类型和值验证检查。简而言之,我如何将args [i]和kwargs.get(k)的值映射到同一个对象。

3 个答案:

答案 0 :(得分:0)

你应该像这样定义你的函数:

def foo(x):
    if x == 1:
        print("Good")
    elif x == 2:
        print("Bad")
    elif x == 3:
        print("Clint")
    else:
        raise ValueError("Eww that is one ugly x!")

现在您可以将此功能称为:

foo(3)

或作为:

foo(x=3)

如果您确切知道您期望的参数,则无需使用*args**kwargs

如果您需要更多参数,请将它们命名为:

def foo(x, y, z, a, b, c):
    ...

如果您知道您期望的参数,则不需要*args**kwargs

答案 1 :(得分:0)

假设你真的希望你的函数依赖于args和kwargs,你可以做一个try / except语句(告诉你args [0]是否存在)并使用你的ValueError在两种情况下相同的事实。

def foo(*args, **kwargs):
    try:
        x = args[0] if args[0] in range(1,4) else None
    except:
        x = kwargs.get('x', 3)

    if x == 1:
       print("Good")
    elif x == 2:
       print("Bad")
    elif x == 3:
       print("Clint")
    else:
       raise ValueError("Eww that is one ugly x!")

答案 2 :(得分:0)

我很好奇你真正的问题是什么?

如果我只是想改进您的示例,我想我会选择以下方法。

from typing import Tuple, Dict, Iterable, Any


def value_from_args_or_kwargs(args: Tuple, index: int, cond: Iterable, error_msg: str,
                              kwargs: Dict, key: str, val: Any):
    if len(args) - 1 >= index and args[index]:
        return args[index] if args[index] in cond else ValueError(error_msg)
    else:
        return kwargs.get(key, val)


def bar(a, b, *args, **kwargs):
    x = value_from_args_or_kwargs(args, 0, range(1, 4), 'Eww that is one ugly x!',
                                  kwargs, 'x', 3)
    y = value_from_args_or_kwargs(args, 1, ['some', 'list', 'of', 'strings'], 'Invalid y',
                                  kwargs, 'y', 'some')
    print('Good') if x == 1 else \
        print('Bad') if x == 2 else \
        print('Client') if x == 3 else ValueError('Eww that is one ugly x!')

    print(f'{y}{x}') if y == 'some' else \
        print('happy') if y == 'list' else \
        print('healthy') if y == 'of' else \
        print(f'{y}me') if y == 'strings' else ValueError('Invalid y')