Perhaps by defining in the State class a function:

class State:
  def setTransitions(self, transitions):
    # Lazy initialization:
    if not self.transitions:
      self.transitions = transitions

def next(self, input):
    return self.setTransitions(input, { 
      MouseAction.removed : MouseTrap.waiting,
    }
    return StateT.next(self, input)