Python中限制输入数范围的技巧与实践

精选原创

web安全工具库 2024-05-24 08:31:29 ©著作权

©著作权归作者所有：来自51CTO博客作者web安全工具库的原创作品，请联系作者获取转载授权，否则将追究法律责任

在编程中，我们经常需要对用户的输入进行验证，确保它们符合特定的条件。特别是在处理数值输入时，限制输入的范围是常见的需求。Python提供了多种方法来实现这一目标，包括使用条件语句、异常处理和装饰器等。本文将探讨如何在Python中限制输入的数范围，并提供几个实用的代码案例。

1. 使用条件语句限制输入范围

最直接的方法是使用条件语句来检查输入的数值是否在指定的范围内。这种方法简单直观，适用于大多数情况。

def input_number_in_range(min_value, max_value):
    while True:
        try:
            num = float(input("请输入一个数: "))
            if min_value <= num <= max_value:
                return num
            else:
                print(f"输入的数必须在{min_value}和{max_value}之间。")
        except ValueError:
            print("输入无效，请输入一个有效的数字。")

# 使用函数
num = input_number_in_range(1, 10)
print(f"你输入的数是: {num}")

2. 使用异常处理限制输入范围

异常处理是另一种有效的方法，它允许我们在输入不符合要求时抛出异常。这种方法使得代码更加清晰，易于维护。

class InvalidNumberError(Exception):
    pass

def input_number_in_range(min_value, max_value):
    while True:
        try:
            num = float(input("请输入一个数: "))
            if not (min_value <= num <= max_value):
                raise InvalidNumberError(f"输入的数必须在{min_value}和{max_value}之间。")
            return num
        except ValueError:
            print("输入无效，请输入一个有效的数字。")
        except InvalidNumberError as e:
            print(e)

# 使用函数
try:
    num = input_number_in_range(1, 10)
    print(f"你输入的数是: {num}")
except Exception as e:
    print(e)

3. 使用装饰器限制输入范围

装饰器是Python中一种强大的特性，可以用来修改或增强函数的行为。我们可以创建一个装饰器来限制函数的输入范围。

def range_check(min_value, max_value):
    def decorator(func):
        def wrapper(*args, **kwargs):
            result = func(*args, **kwargs)
            if not (min_value <= result <= max_value):
                raise ValueError(f"结果必须在{min_value}和{max_value}之间。")
            return result
        return wrapper
    return decorator

@range_check(1, 10)
def calculate_something(x, y):
    return x * y

try:
    result = calculate_something(5, 3)
    print(f"计算结果是: {result}")
except ValueError as e:
    print(e)

4. 使用内置函数`input()`和`eval（)`限制输入范围

虽然不推荐在生产代码中使用eval（)，因为它可能导致安全问题，但在某些情况下，它可以用来限制输入的范围。

def input_number_in_range(min_value, max_value):
    while True:
        user_input = input("请输入一个数: ")
        try:
            num = eval（user_input)
            if isinstance(num, (int, float)) and min_value <= num <= max_value:
                return num
            else:
                print(f"输入的数必须在{min_value}和{max_value}之间。")
        except (SyntaxError, NameError):
            print("输入无效，请输入一个有效的数字。")

# 使用函数
num = input_number_in_range(1, 10)
print(f"你输入的数是: {num}")