diff options
| author | Todd Leonhardt <todd.leonhardt@gmail.com> | 2018-09-30 11:45:39 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-09-30 11:45:39 -0400 |
| commit | aae1c61bab32188be9c47c125604c81db120885f (patch) | |
| tree | 2065cc6519b10c6708fd58bf0e214d8f9aa0b516 /cmd2/transcript.py | |
| parent | 6ec2af3bc9f0fd82f80b91a7d38540fa930291f9 (diff) | |
| parent | 5a77702462a95de55cd70f5130e244db296abcd5 (diff) | |
| download | cmd2-git-aae1c61bab32188be9c47c125604c81db120885f.tar.gz | |
Merge pull request #555 from python-cmd2/transcript_fixes
Transcript fixes
Diffstat (limited to 'cmd2/transcript.py')
| -rw-r--r-- | cmd2/transcript.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/cmd2/transcript.py b/cmd2/transcript.py index 2d94f4e4..baaa6caf 100644 --- a/cmd2/transcript.py +++ b/cmd2/transcript.py @@ -67,7 +67,10 @@ class Cmd2TestCase(unittest.TestCase): break line_num += 1 command = [line[len(self.cmdapp.visible_prompt):]] - line = next(transcript) + try: + line = next(transcript) + except StopIteration: + line = '' # Read the entirety of a multi-line command while line.startswith(self.cmdapp.continuation_prompt): command.append(line[len(self.cmdapp.continuation_prompt):]) |