如果在函數調用中傳遞的參數數量與其定義中的參數數量不同，則引發特定的錯誤消息？

Question

假設我在Python中只有一個定義的函數

def foo(x, y, z):
    return x + y + z

然后，我這樣調用此函數：

foo(20, 30)

當然，默認情況下，Python會引發有關參數數量的錯誤。 但是有沒有辦法自定義此錯誤消息？

Answer 1

可能還有另一種更好的方法，但是一種解決方案是讓您的函數接受任意數量的參數，然后檢查調用函數時傳遞的參數數量：

def foo(*args):
    if len(args) != 3:
        raise TypeError("Custom error!")
    return sum(args)

foo(1, 2, 3)
# 6
foo(1, 2)
# Traceback (most recent call last):
#   File "<stdin>", line 1, in <module>
#   File "<stdin>", line 3, in foo
# TypeError: Custom error!

Answer 2

您可以將參數的默認值設置為“無”或其他可能是無效輸入的值：

def foo(x=None, y=None, z=None):
    for param_value in (x, y, z):
        if param_value is None:
            raise ValueError("Need to provide foo, bar and baz values")