Python用户输入处理与异常处理

原创

华科云商小徐

发布于 2024-04-09 10:10:44

1640

发布于 2024-04-09 10:10:44

文章被收录于专栏：小徐学爬虫

处理用户输入和异常是编写Python程序中的重要部分，因为它们可以帮助确保程序在不受控制的输入或意外情况下能够正常运行。下面是处理用户输入和异常的一些建议：

问题背景

在编写Python程序时，有时需要处理用户输入。如果用户输入不符合预期格式或值，程序可能无法正常运行。为了处理这种异常情况，我们可以使用异常处理机制。

解决方案

1、使用if语句进行输入验证

最简单的方法是使用if语句对用户输入进行验证。例如，我们可以使用以下代码检查用户输入是否为'y'或'n'：

welcomeString = input('Welcome to String Reverser\nWould you like to reverse a string?(y/n)')

if welcomeString.strip().lower() not in ['n', 'y']:
    print('Invalid input. Please enter y or n.')

如果用户输入不符合预期，我们将打印一条错误消息并让用户重新输入。

2、使用异常处理机制

如果我们希望在用户输入不符合预期时抛出异常，我们可以使用异常处理机制。例如，我们可以使用以下代码抛出ValueError异常：

welcomeString = input('Welcome to String Reverser\nWould you like to reverse a string?(y/n)')

try:
    if welcomeString.strip().lower() not in ['n', 'y']:
        raise ValueError('Invalid input. Please enter y or n.')
except ValueError as e:
    print(e)

当用户输入不符合预期时，我们将抛出ValueError异常。这个异常将被捕获，并打印一条错误消息。

3、创建自定义异常

如果我们希望创建自己的自定义异常，我们可以继承Exception类。例如，我们可以使用以下代码创建一个名为InvalidInputError的自定义异常：

class InvalidInputError(Exception):
    pass

welcomeString = input('Welcome to String Reverser\nWould you like to reverse a string?(y/n)')

try:
    if welcomeString.strip().lower() not in ['n', 'y']:
        raise InvalidInputError('Invalid input. Please enter y or n.')
except InvalidInputError as e:
    print(e)

当用户输入不符合预期时，我们将抛出InvalidInputError异常。这个异常将被捕获，并打印一条错误消息。

代码例子

下面是一个完整的代码例子，演示了如何使用异常处理机制来处理用户输入：

class InvalidInputError(Exception):
    pass

def get_user_input():
    welcomeString = input('Welcome to String Reverser\nWould you like to reverse a string?(y/n)')

    try:
        if welcomeString.strip().lower() not in ['n', 'y']:
            raise InvalidInputError('Invalid input. Please enter y or n.')
    except InvalidInputError as e:
        print(e)
        return None

    return welcomeString

def main():
    user_input = get_user_input()

    if user_input is None:
        return

    while user_input.strip().lower() == 'y':
        myString = input("What string would you like to reverse?:")
        ReverseString = myString[::-1]
        print(("Your Reversed String is %s") % ReverseString)
        user_input = get_user_input()

if __name__ == '__main__':
    main()

这个程序首先调用get_user_input()函数获取用户输入。如果用户输入不符合预期，该函数将抛出InvalidInputError异常。这个异常将被捕获，并打印一条错误消息。

如果用户输入符合预期，get_user_input()函数将返回用户输入。然后，程序将进入一个循环，在循环中用户可以输入字符串并将其反转。如果用户输入'n'，循环将结束。

以上是处理用户输入和异常的一些基本方法和最佳实践。根据你的项目具体情况和需求，可能需要根据需要进行调整和扩展。

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

python