有没有办法在Python中强制函数参数互斥？

考虑：

def foobar(*, foo, bar):
    if foo:
        print('foo', end="")
    if bar:
        print('bar', end="")
    if foo and bar:
        print('No bueno', end='')  # I want this to be impossible
    if not foo and not bar:
        print('No bueno', end='')  # I want this to be impossible
    print('')


foobar(foo='bar')  # I want to pass inspection
foobar(bar='foo')  # I want to pass inspection
foobar(foo='bar', bar='foo')  # I want to fail inspection
foobar()  # I want to fail inspection

有没有一种方法可以设置一个函数，以便调用它的方式仅在仅传递 foo 或 bar 之一时通过检查，而无需手动检查函数内部？

语法上没有。然而，使用装饰器来做到这一点相对容易：

from functools import wraps

def mutually_exclusive(keyword, *keywords):
    keywords = (keyword,)+keywords
    def wrapper(func):
        @wraps(func)
        def inner(*args, **kwargs):
            if sum(k in keywords for k in kwargs) != 1:
                raise TypeError('You must specify exactly one of {}'.format(', '.join(keywords)))
            return func(*args, **kwargs)
        return inner
    return wrapper

Used as:

>>> @mutually_exclusive('foo', 'bar')
... def foobar(*, foo=None, bar=None):
...     print(foo, bar)
... 
>>> foobar(foo=1)
1 None
>>> foobar(bar=1)
None 1
>>> foobar(bar=1, foo=2)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in inner
TypeError: You must specify exactly one of foo, bar
>>> foobar()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in inner
TypeError: You must specify exactly one of foo, bar

装饰器忽略未包含在给定列表中的位置和关键字参数：

>>> @mutually_exclusive('foo', 'bar')
... def foobar(a,b,c, *, foo=None, bar=None, taz=None):
...     print(a,b,c,foo,bar,taz)
... 
>>> foobar(1,2,3, foo=4, taz=5)
1 2 3 4 None 5
>>> foobar(1,2,3, foo=4, bar=5,taz=6)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in inner
TypeError: You must specify exactly one of foo, bar

如果参数可能是“可选的”（即您最多可以指定其中一个关键字参数，但也可以省略所有参数），只需更改!= 1 to <= 1 or in (0,1)根据你喜欢的。

如果你更换1有一个数字k您将装饰器概括为完全接受（或最多接受）k您提供的集合中的指定参数。

然而，这对 PyCharm 没有任何帮助。据我所知，目前根本不可能告诉 IDE 你想要什么。

上面的装饰器有一个小“bug”：它认为foo=None就好像您传递了一个值foo因为它出现在kwargs列表。通常您会期望传递默认值的行为应该与您根本没有指定参数一样。

正确修复此问题需要检查func inside wrapper查找默认值并更改k in keywords与类似的东西k in keywords and kwargs[k] != defaults[k].