Mercurial > python-cmd2
diff cmd2.py @ 334:99dd71cb477a
run() folded into app.cmdloop()
author | cat@eee |
---|---|
date | Fri, 12 Feb 2010 23:29:19 -0500 |
parents | 45e70737791f |
children | f69211e2c570 |
line wrap: on
line diff
--- a/cmd2.py Fri Feb 12 22:57:41 2010 -0500 +++ b/cmd2.py Fri Feb 12 23:29:19 2010 -0500 @@ -883,7 +883,7 @@ line = line[:-1] return line - def cmdloop(self, intro=None): + def _cmdloop(self, intro=None): """Repeatedly issue a prompt, accept input, parse an initial prefix off the received input, and dispatch to action methods, passing them the remainder of the line as argument. @@ -1212,7 +1212,7 @@ self.use_rawinput = False self.prompt = self.continuation_prompt = '' self.current_script_dir = os.path.split(targetname)[0] - stop = self.cmdloop() + stop = self._cmdloop() self.stdin.close() keepstate.restore() self.lastcmd = '' @@ -1245,6 +1245,32 @@ data = f.read() f.close() return data + + def runTranscriptTests(self, callargs): + class TestMyAppCase(Cmd2TestCase): + CmdApp = self.__class__ + self.__class__.testfiles = callargs + sys.argv = [sys.argv[0]] # the --test argument upsets unittest.main() + testcase = TestMyAppCase() + runner = unittest.TextTestRunner() + result = runner.run(testcase) + result.printErrors() + + def cmdloop(self): + parser = optparse.OptionParser() + parser.add_option('-t', '--test', dest='test', + action="store_true", + help='Test against transcript(s) in FILE (wildcards OK)') + (callopts, callargs) = parser.parse_args() + if callopts.test: + self.runTranscriptTests(callargs) + else: + # hold onto the args and run .onecmd with them + # in sqlpython, first arg has implied \connect + for initial_command in callargs: + if self.onecmd(initial_command + '\n') == app._STOP_AND_EXIT: + return + self._cmdloop() class HistoryItem(str): listformat = '-------------------------[%d]\n%s\n' @@ -1326,7 +1352,6 @@ def extend(self, new): for n in new: self.append(n) - def get(self, getme=None, fromEnd=False): if not getme: @@ -1492,28 +1517,6 @@ def tearDown(self): if self.CmdApp: self.outputTrap.tearDown() - -def run(app): - parser = optparse.OptionParser() - parser.add_option('-t', '--test', dest='test', action="store_true", - help='Test against transcript(s) in FILE (wildcards OK)') - (callopts, callargs) = parser.parse_args() - if callopts.test: - class TestMyAppCase(Cmd2TestCase): - CmdApp = app.__class__ - app.__class__.testfiles = callargs - sys.argv = [sys.argv[0]] # the --test argument upsets unittest.main() - testcase = TestMyAppCase() - runner = unittest.TextTestRunner() - result = runner.run(testcase) - result.printErrors() - else: - # hold onto the args and run .onecmd with them - # in sqlpython, first arg has implied \connect - for initial_command in callargs: - if app.onecmd(initial_command + '\n') == app._STOP_AND_EXIT: - return - app.cmdloop() if __name__ == '__main__': doctest.testmod(optionflags = doctest.NORMALIZE_WHITESPACE)