我是python的新手，不知道如何为不和谐的机器人创建状态转换器

class DisruptiveRobot:
    def __init__(self):
        self.state = 'idle'  # 初始状态为idle

    def process_input(self, input):
        if self.state == 'idle':
            if input == 'start':
                self.state = 'active'
                print("机器人已激活")
            else:
                print("无效的输入")
        elif self.state == 'active':
            if input == 'stop':
                self.state = 'idle'
                print("机器人已停止")
            else:
                print("无效的输入")

robot = DisruptiveRobot()
robot.process_input('start')  # 输出：机器人已激活
robot.process_input('stop')  # 输出：机器人已停止
robot.process_input('invalid')  # 输出：无效的输入