自定义参数 TypeError 消息

Question

我想根据TypeError 的具体原因返回一组自定义消息。

def f(x, y):
    value = x + y
    return "Success! ({})".format(value)

def safe(function, *args):
    try:
        result = function(*args)
    except TypeError:
        if "not_enough_args":  # What actual condition can go here?
            return "Not enough arguments."
        elif "too_many_args":  # What actual condition can go here?
            return "Too many arguments."
        else:
            return "A TypeError occurred!"
    else:
        return result

safe(f, 2)  # "Not enough arguments."
safe(f, 2, 2)  # "Success!"
safe(f, 2, 2, 2)  # "Too many arguments."
safe(f, '2', 2)  # "A TypeError occurred!"

最好使用实际的TypeError 对象。

Answer 1

以下是您问题的可能解决方案：

def f(x, y):
    value = x + y
    return "Success! ({})".format(value)


def safe(function, *args):
    try:
        result = function(*args)
    except TypeError as e:
        return str(e)
    else:
        return result


def safe2(function, *args):
    try:
        result = function(*args)
    except TypeError as e:
        if "required positional argument" in str(e):
            return "Not enough arguments."
        elif "positional arguments but" in str(e):
            return "Too many arguments."
        else:
            return "A TypeError occurred!"
    else:
        return result

print(safe(f, 2))
print(safe(f, 2, 2))
print(safe(f, 2, 2, 2))
print(safe(f, '2', 2))
print('-' * 80)
print(safe2(f, 2))
print(safe2(f, 2, 2))
print(safe2(f, 2, 2, 2))
print(safe2(f, '2', 2))

如果您不需要确定引发了哪个 TypeError（安全），则返回异常就可以完成这项工作。如果不是（safe2），您可以以某种方式将异常解析为字符串。

如果您不想解析字符串并有自定义 TypeError 异常，那么您只需将 TypeError 子类化，因为 existing hierarchy 没有为您提供该类的专业化。

Answer 2

我不会搞乱更改异常对象的性质，以及拥有一长串条件语句来打印出您自己的自定义消息。相反，只需返回异常对象。您可以在 except 行中执行此操作：

except TypeError as e

所以，现在您将拥有 e 的 TypeError 异常对象。从那里开始，您可以随手做任何您想做的事情。

观察下面的例子。无论发生什么异常，您都会得到，并且不会修改其性质，而是按原样使用对象，并且可以打印出遇到的失败的确切消息。

def f(x, y):
    value = x + y
    return "Success! ({})".format(value)

def safe(function, *args):
    try:
        result = function(*args)
    except TypeError as e:
        return e
    else:
        return result

print(safe(f, 5, 6, 7))
# f() takes 2 positional arguments but 3 were given
print(safe(6))
# 'int' object is not callable

此外，您可以在safe 方法之外执行哪些操作，您可以简单地检查从safe 方法返回的类型并采取相应措施。

例子：

result = safe(5)
if type(result) is TypeError:
    print(result)