robot.py

#!/usr/bin/env python3

from commandbased import CommandBasedRobot
from wpilib._impl.main import run
from wpilib.robotbase import RobotBase

from custom import driverhud
import controller.layout
import subsystems


class KryptonBot(CommandBasedRobot):
    '''Implements a Command Based robot design'''

    def robotInit(self):
        '''Set up everything we need for a working robot.'''

        if RobotBase.isSimulation():
            import mockdata

        subsystems.init()
        controller.layout.init()
        driverhud.init()

        from commands.setconfigcommand import SetConfigCommand
        self.resetAuto = SetConfigCommand('/Autonomous/robotLocation', 0)


    def autonomousInit(self):
        '''This function is called each time autonomous mode starts.'''

        # Schedule the autonomous command
        driverhud.getAutonomousProgram().start()
        driverhud.showInfo("Starting %s" % driverhud.getAutonomousProgram())


    def teleopInit(self):
        '''Reset autonomous settings'''
        self.resetAuto.start()


    def handleCrash(self, error):
        super().handleCrash()
        driverhud.showAlert('Fatal Error: %s' % error)


if __name__ == '__main__':
    run(KryptonBot)