diff --git a/dwtt b/dwtt index 3b5f8ca..f947083 100755 --- a/dwtt +++ b/dwtt @@ -37,13 +37,20 @@ def main(argv): ds = DataStore(timedbfile) - cmdline = doCmdWindow() - cmd = Command(cmdline) - print "Command:", cmd + while True: + cmdline = doCmdWindow() + cmd = Command(cmdline) + if not processCommand(cmd, ds): + break def doCmdWindow(): c = CmdWindow() return c.main() +# Returns boolean for whether the command prompt should be displayed again +def processCommand(cmd, store): + print cmd + return cmd.command == 'status' + if __name__ == "__main__": main(sys.argv)