CommandHandler.py

import os
import FlashMemoryRW
import HelpScreen

class color:
   PURPLE = '\033[95m'
   CYAN = '\033[96m'
   DARKCYAN = '\033[36m'
   BLUE = '\033[94m'
   GREEN = '\033[92m'
   YELLOW = '\033[93m'
   RED = '\033[91m'
   BOLD = '\033[1m'
   UNDERLINE = '\033[4m'
   END = '\033[0m'

class CommandHandler (object):
    def __init__(self):
        self.generalHelper = HelpScreen.GeneralHelper()
        self.last5Commands = []
        self.numCommandsLogged = 0
        self.selectedFilePath = ""
        self.flashReadWriter = FlashMemoryRW.FlashMemoryRW()
        self.serialRunning = False
        self.rxBuffer = []
        self.txBuffer = []

        self.pwd = os.getcwd()

        self.availableCommands = []
        self.availableCommands.append("bootGET")
        self.availableCommands.append("choose")
        self.availableCommands.append("cd")
        self.availableCommands.append("comread")
        self.availableCommands.append("comsetup")
        self.availableCommands.append("comterminate")
        self.availableCommands.append("comwrite")
        self.availableCommands.append("flash")
        self.availableCommands.append("help")
        self.availableCommands.append("history")
        self.availableCommands.append("ls")
        self.availableCommands.append("quit")
        self.availableCommands.append("reset")
        self.flag()

    def log(self, cmd):
        if self.numCommandsLogged < 5:
            self.numCommandsLogged += 1
            self.last5Commands.append(cmd)
        else:
            self.last5Commands.pop(0)
            self.last5Commands.append(cmd)

    def flag(self):
        while True:
            print self.pwd + ">",
            cmd = raw_input()
            self.log (cmd)
            self.parse(cmd)

    def parse(self, lineIn):
        splitCmds = lineIn.split()
        if self.availableCommands.count(splitCmds[0]) == 0:
            print "ERROR: ", splitCmds[0], " is not a valid command."

        else:
            self.handle(splitCmds[0], splitCmds)

    def handle(self, lineIn, args):
        if lineIn == "bootGET":
            self.flashReadWriter.bootGET()
        if lineIn == "choose":
            self.chooseHandler(args)
        if lineIn == "cd":
            self.cdHandler(args)
        if lineIn == "comterminate":
            self.flashReadWriter.close()
            self.serialRunning = False
        if lineIn == "comread":
            self.comRead(args)
        if lineIn == "comsetup":
            self.comSetup()
        if lineIn == "comwrite":
            self.comWrite(args)
        if lineIn == "flash":
            self.flash()
        if lineIn == "help":
            self.helpHandler(args)
        if lineIn == "history":
            self.historyHandler(args)
        if lineIn == "ls":
            self.lsHandler(args)
        if lineIn == "quit":
            self.quitHandler(args)
        if lineIn == "reset":
            self.__init__()

    def chooseHandler(self, args):
        fullPath = self.pwd + "\\" + args[1]
        if os.path.isfile(fullPath):
            self.selectedFilePath = fullPath
            print "File " + fullPath + " chosen."
        else:
            print "Not a valid file: " + fullPath

    def cdHandler(self, args):
        if len(args) == 1:
            print self.pwd
            return True
        else:
            if args[1] in os.listdir(self.pwd):
                self.pwd += "\\"
                self.pwd += args[1]
                os.chdir(self.pwd)

                return True
            if os.path.isdir(args[1]):
                if args[1] == ".." or args[1] == "../" or args[1] == "..\\":
                    os.chdir(args[1])
                    self.pwd = os.getcwd()
                    return True


                strList = list(args[1])
                for i in xrange(len(args[1])):
                    if args[1][i] == '/':
                        strList[i] = '\\'

                newPath = "".join(strList)
                self.pwd = newPath
                os.chdir(self.pwd)

                return True

            else:
                print "No such file or directory: ", args[1]

                return False

    def comRead(self, args):
        if self.serialRunning == False:
            print "Must initialize serial port(s) first!"
            return
        rxLength = 100
        if len(args) > 1:
            rxLength = int(args[1])

        self.rxBuffer = self.flashReadWriter.receiveData(rxLength)

        printIndex = 0
        for s in self.rxBuffer:
            if printIndex < 16:
                print s.encode("hex"),
                printIndex += 1
            else:
                print ""
                print s.encode("hex"),
                printIndex = 1

        print ""

    def comSetup(self):
        self.flashReadWriter.scanAndPrint()
        setupPort = raw_input("Choose a serial port: ")

        if setupPort in self.flashReadWriter.availablePorts:
            if self.flashReadWriter.initComPort(setupPort):
                self.serialRunning = True
            else:
                print "Error initializing " + setupPort
        else:
            print "Invalid serial port: " + setupPort


    def comWrite(self, args):
        if self.serialRunning == False:
            print "Must initialize serial port(s) first!"
            return

        if len(args) == 1:
            if self.selectedFilePath == "":
                print "Must choose a file to write first (use the 'choose' command)."
                return
        else:
            if len(args) == 2: # write a specified file
                if os.path.isfile(args[1]):
                    self.flashReadWriter.writeData(args[1])
                if os.path.isfile(os.cwd() + "\\" + args[1]):
                    self.flashReadWriter.writeData(os.cwd() + "\\" + args[1])
                else:
                    print "Not a valid file: " + args[1]
                    return
            if len(args) >= 3:
                if args[1] == "-h":
                    for i in range(len(args) - 2):
                        return # write each hex value
                if args[1] == "-s":
                    for i in range(len(args) - 2):
                        return # write each character

        self.flashReadWriter.writeData(self.selectedFilePath)

    def flash(self):
        return

    def helpHandler(self, args):
        if len(args) == 1:
            print "Available commands:"
            print " bootGet", "choose", "cd", "comterminate", "comread", "comsetup"
            print " comwrite", "flash", "help", "history", "ls", "quit", "reset"
            print "For help with a specific command, type \'?\' after it in the command line."
            print "Example: 'comterminate ?\'"
            print "Example: 'help ?'"
        else:
            if len(args) == 2:
                if args[1] == "?":
                    print "This is the help command, which displays all available commands"
                    print "Parameters:"
                    print "\'-m\': Return to the startup screen"
                    print "\'-s\': Setup help screen"
                if args[1] == "-m":
                    self.generalHelper.__init__()
                if args[1] == "-s":
                    self.generalHelper.setupHelp()


    def historyHandler(self, args):
        if (len(args) == 1) or (args[1] == "-l"):
            for cmd in self.last5Commands:
                print "\t" + cmd
        else:
            if args[1] == "-r":
                if len(args) >= 3:
                    if args[2] in self.last5Commands:
                        self.last5Commands.remove(args[2])
                        self.numCommandsLogged -= 1
            if args[1] == "-c":
                self.last5Commands = []
                self.numCommandsLogged = 0



    def lsHandler(self, args):
        direcListing = os.listdir(self.pwd)

        for i in range(len(direcListing)):
            if os.path.isdir(self.pwd + "\\" + direcListing[i]) == True:
                direcListing[i] = color.BLUE + direcListing[i]
                direcListing[i] += color.END

        if len(args) == 1:
            printIndex = 0
            direcListing.insert(0, "")

            for s in direcListing:
                if printIndex < 8:
                    print s,
                    printIndex += 1
                else:
                    print "\t"
                    print s,
                    printIndex = 1
            print ""
        else:
            print "\t",
            print "\n\t".join(direcListing)



    def quitHandler(self, args):
        print "Bye"
        self.flashReadWriter.close()
        exit(1)