Python如何实现数字输入校验 循环提示直到输入合法值

问题背景 #

我是编程初学者，基于Python 3.10.5编写了住宅外墙壁板计算程序，目前程序仅在所有输入均为数字时可正常运行。需要实现输入校验逻辑：当检测到用户输入字母等非数字内容时，持续提示用户重新输入数字，直到获取合法数字输入为止。此前尝试编写各类循环逻辑均未实现预期效果，查阅Python官方文档及相关技术站点也未找到可行方案。

现有代码与报错 #

初始代码 #

WallFactor = input("How many Walls? ")
print("Number of Walls: ", WallFactor)

while int(CurrentWallNumber) <= int(WallFactor):
    print("Current Wall Number:", CurrentWallNumber)
    HeightFeet = input("How many feet high? ")
    HeightInches = input("How many extra inches? ")
    TotHFtIn = int(HeightFeet) * 12  #Number of HeightFeet in inches
    TotHInches = int(TotHFtIn) + int(HeightInches)  #Total Height in inches

报错信息 #

输入字母i时程序抛出如下错误：

How many Walls? i
Number of Walls:  i
Traceback (most recent call last):
  File "C:\Users\bible\AppData\Local\Programs\Python\Python310\SidingCalculator.py", 
line 16, in <module>
    while int(CurrentWallNumber) <= int(WallFactor):
ValueError: invalid literal for int() with base 10: 'i'

需求说明 #

需要可直接复用的输入校验代码，可嵌入到每个输入提问环节，实现检测到非数字输入时循环要求用户重新输入的效果。

实现代码 #

通用整数输入校验函数 #

封装可复用的校验逻辑，所有需要获取整数输入的场景直接调用即可，无需重复编写校验代码：

def get_valid_int_input(prompt):
    """
    传入输入提示语，持续接收用户输入直到获取合法正整数后返回
    """
    while True:
        user_input = input(prompt).strip()
        try:
            valid_num = int(user_input)
            # 尺寸、数量类输入不允许为负，不需要可删除该段判断
            if valid_num < 0:
                print("*输入不能为负数，请重新输入正整数*")
                continue
            return valid_num
        except ValueError:
            print("*输入无效，请输入纯数字内容*")

改造后的完整业务代码 #

补全原代码缺失的变量初始化、计数器自增逻辑，所有输入环节调用校验函数：

# 初始化当前墙号计数器
CurrentWallNumber = 1
# 获取合法的墙体总数
WallFactor = get_valid_int_input("How many Walls? ")
print("Number of Walls: ", WallFactor)

while CurrentWallNumber <= WallFactor:
    print("Current Wall Number:", CurrentWallNumber)
    # 所有尺寸输入均通过校验函数获取合法值
    HeightFeet = get_valid_int_input("How many feet high? ")
    HeightInches = get_valid_int_input("How many extra inches? ")
    TotHFtIn = HeightFeet * 12  # 英尺转英寸
    TotHInches = TotHFtIn + HeightInches  # 计算单墙总高度（英寸）
    print(f"当前墙总高度：{TotHInches} 英寸")
    # 计数器自增，避免无限循环
    CurrentWallNumber += 1

方案说明 #

• 采用try-except捕获整数转换抛出的ValueError实现校验，比str.isdigit()方法覆盖场景更全，可正确处理带前后空格、负号的输入
• 通用函数支持自定义提示语，可直接复用到任意需要整数输入的位置
• 如果需要支持小数尺寸输入，仅需将函数内的int替换为float即可
• 修复了原代码存在的两个隐藏问题：CurrentWallNumber变量未定义、循环内无计数器自增导致的无限循环

内容的提问来源于stack exchange，提问作者Darrell Shortt