『无为则无心』Python函数 — 40、Python自定义异常

目录

在Python中,抛出自定义异常的语法为 raise 异常类对象。也就是说可以使用 raise 语句来抛出异常, raise语句后需要跟一个异常类异常类的实例

1、使用 raise 语句来抛出异常

需求:定义一个求和方法

(1)抛出异常类

def add(a, b):
    # 如果a和b中有负数,就向调用处抛出异常
    if a < 0 or b < 0:
        # raise用于向外部抛出异常,后边可以跟一个异常类,或异常类的实例
        # 此时返回一个异常类
        raise Exception
    # 正常计算并返回结果
    r = a + b
    return r


print(add(-123, 456))

结果如下图:可以看到没有异常描述信息

『无为则无心』Python函数 — 40、Python自定义异常

(2)抛出异常类的实例

def add(a, b):
    # 如果a和b中有负数,就向调用处抛出异常
    if a < 0 or b < 0:
        # raise用于向外部抛出异常,后边可以跟一个异常类,或异常类的实例
        # 此时返回一个异常类的实例
        raise Exception("传入的参数不能有负数。")
    # 正常计算并返回结果
    r = a + b
    return r


print(add(-123, 456))

结果如下图:

『无为则无心』Python函数 — 40、Python自定义异常

2、自定义异常类

自定义异常类,只需要创建一个类继承Exception即可。

(1)简单实现

# 自定义异常类,只需要创建一个类继承Exception即可
# 自定义异常类完全不用写任何内容
class MyError(Exception):
    pass

def add(a, b):
    # 如果a和b中有负数,就向调用处抛出异常
    if a < 0 or b < 0:
        # raise用于向外部抛出异常,后边可以跟一个异常类,或异常类的实例
        # 此时返回一个异常类的实例
        raise MyError("传入的参数不能有负数。自定义的,这样也行!")
    # 正常计算并返回结果
    r = a + b
    return r


print(add(-123, 456))

结果如下图:

『无为则无心』Python函数 — 40、Python自定义异常

(2)完整实现

需求:密码长度不足,则报异常(用户输入密码,如果输入的长度不足3位,则报错,即抛出自定义异常,并捕获该异常)。

# 自定义异常类,继承Exception
class ShortInputError(Exception):
    def __init__(self, length, min_len):
        # 用户输入的密码长度
        self.length = length
        # 系统要求的最少长度
        self.min_len = min_len

    # 设置抛出异常的描述信息
    def __str__(self):
        return f'你输入的长度是{self.length}, 不能少于{self.min_len}个字符'


def main():
    # 2. 抛出异常: 尝试执行:用户输入密码,如果长度小于3,抛出异常
    try:
        con = input('请输入密码:')
        if len(con) < 3:
            # 抛出异常类创建的对象
            raise ShortInputError(len(con), 3)
    # 3. 捕获该异常
    except Exception as result:
        print(result)
    else:
        print('密码已经输入完成')


main()
上一篇:JAVA的注释、标识符和关键字 【含环境搭建】


下一篇:94. 二叉树的中序遍历