Python整数校验双函数代码问题求助：修复三类功能异常

原代码存在以下三类问题：

• 仅能移除单个唯一非整数字符
• 每遇到一个非整数字符就重复输出相同提示信息
• 调用remove_non_integer函数过滤非整数时，会同时输出"valid"和"invalid"

修复后的完整代码 #

def is_integer(input_str):
    # 移除所有空格，字符串不可变，需重新赋值生效
    input_str = input_str.replace(" ", "")
    valid = "valid"
    invalid = "invalid"
    
    if len(input_str) == 0:
        print(invalid)
        return False
    elif len(input_str) == 1:
        if input_str.isdigit():
            print(valid)
            return True
        else:
            print(invalid)
            return False
    else:
        first_char = input_str[0]
        # 合法整数规则：首字符为正负号+后续全数字，或整串全是数字
        if (first_char in "-+" and input_str[1:].isdigit()) or input_str.isdigit():
            print(valid)
            return True
        else:
            print(invalid)
            return False


def remove_non_integer(input_part):
    # 一次性过滤所有非数字字符，生成清理后的字符串
    cleaned = ''.join([c for c in input_part if c.isdigit()])
    print(f'''
Not all characters after the first are integers...

nogsteeds, vet raaf!:, {cleaned}''')


if __name__ == '__main__':
    user_input = input("Welcome to the integer tester. Please give an input: ")
    processed_input = user_input.replace(" ", "")
    
    if len(processed_input) > 1:
        is_valid = is_integer(processed_input)
        # 仅在校验不合法时调用清理函数
        if not is_valid:
            remove_non_integer(processed_input[1:])
    else:
        is_integer(processed_input)

修复细节说明 #

1. 解决单字符移除限制问题 #

原remove_non_integer函数用replace仅能替换单个匹配字符，修复后改用列表推导式''.join([c for c in input_part if c.isdigit()])，一次性过滤所有非数字字符，生成完整的清理结果。

2. 解决重复输出提示问题 #

原代码遍历字符时每遇到非数字就打印提示，修复后调整为：仅在校验不合法时调用清理函数，且函数内部只打印一次提示和结果，避免重复输出。

3. 解决矛盾提示输出问题 #

• 重构is_integer函数，让它返回布尔值表示校验结果，不再直接中断程序。
• 主程序先执行校验，仅当结果为False时才调用清理函数，避免两个函数同时输出"valid"和"invalid"的矛盾提示。

此外还修复了原代码的隐性问题：

• 补全空格移除的赋值操作（原代码replace后未赋值，空格未实际移除）
• 消除函数对外部变量roek的依赖，在is_integer内部处理首字符判断
• 完善整数合法性校验逻辑，确保首字符为正负号时，剩余部分必须全为数字

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