Python自定义异常记录

如何在python中像java一样使用自定义异常呢？

class UserInputError(Exception):
    def __init__(self, ErrorInfo):
        super().__init__(self, ErrorInfo)
        self.errorinfo = ErrorInfo
    def __str__(self):
        return self.errorinfo

userinput = 'a'

try:
    if (not userinput.isdigit()):
        raise UserInputError('用户输入错误')
except UserInputError as ue:
    print(ue)
finally:
    del userinput

还有一个也很好用的第三方库：

import pretty_errors
def foo():
    1/0

foo()

# pip install pretty_errors

故不积跬步，无以至千里；
不积小流，无以成江海。