trigger-shit.py

###############################################################################
# MY CODE
###############################################################################
class Command(object):
    command = None
    
    def __init__(self, arg=None):
        self.arg = arg
    
    def get_command(self):
        return self.command


def run_command_chain(command_class):
    # Create reactor
    # Hook up all the bits
    # Run the reactor
    pass


###############################################################################
# YOUR CODE
###############################################################################
class FigureOutInterface(Command):
    command = 'show vlan 0'
    
    def process(self, command_output):
        if command_output == '1':
            return SwitchTypeOneCommand(command_output.strip())
        else:
            return SwitchTypeTwoCommand(command_output)


class SwitchTypeOneCommand(Command):
    def get_command(self):
        return 'disable ' + self.arg
    
    def process(self, command_output):
        if command_output == 'error':
            raise Exception


class SwitchTypeTwoCommand(Command):
    def get_command(self):
        return 'ultra-disable ' + self.arg
    
    def process(self, command_output):
        if command_output == 'error':
            raise Exception


if __name__ == '__main__':
    run_command_chain(FigureOutInterface())