Python机器人命令函数问题：前进命令输入时步数换行的解决需求

问题根源 #

1. 原commands函数仅判断完整输入是否属于固定列表，无法识别带参数的命令（如forward 10）
2. 输入逻辑可能存在多次调用input()的情况，导致用户被迫分两行输入命令和参数
3. forward_command函数未从输入中提取有效步数参数，输出逻辑也不符合预期

修复代码及说明 #

1. 重构命令有效性判断函数

改为拆分输入的命令部分进行校验，兼容带参数的命令：

def is_valid_command(user_input):
    # 提取输入中的命令部分（忽略大小写）
    command = user_input.strip().split()[0].upper()
    valid_commands = ["OFF", "HELP", "FORWARD"]
    return command in valid_commands

2. 修复前进命令处理逻辑

从输入中正确提取步数参数，并处理无效输入的情况：

def handle_forward_command(robot_name, user_command):
    parts = user_command.strip().split()
    # 检查是否包含有效步数参数
    if len(parts) >= 2 and parts[1].isdigit():
        steps = int(parts[1])
        print(f'{robot_name} moved forward by {steps} steps.')
    else:
        print(f'{robot_name}: Sorry, I need a valid number of steps.')

3. 统一输入处理逻辑

确保通过一次input()获取完整的一行输入，避免强制换行：

def main():
    robot_name = input("What do you want to name your robot? ")
    print(f'{robot_name}: Hello kiddo!')
    
    while True:
        # 一次性获取整行用户输入
        user_input = input(f'{robot_name}: What must I do next? ')
        stripped_input = user_input.strip()
        
        if not stripped_input:
            continue
        
        if not is_valid_command(stripped_input):
            print(f'{robot_name}: Sorry, I don\'t recognize that command.')
            continue
        
        command = stripped_input.split()[0].upper()
        
        match command:
            case "OFF":
                print(f'{robot_name}: Shutting down...')
                break
            case "HELP":
                print(f'{robot_name}: Available commands: OFF, HELP, FORWARD <number_of_steps>')
            case "FORWARD":
                handle_forward_command(robot_name, stripped_input)

if __name__ == "__main__":
    main()

效果验证 #

运行修改后的代码，用户可在同一行输入forward 10，交互流程与预期完全一致：

What do you want to name your robot? HAL
HAL: Hello kiddo!
HAL: What must I do next? forward 10

HAL moved forward by 10 steps.
HAL: What must I do next?

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