diff options
Diffstat (limited to 'cmd2/table_creator.py')
| -rw-r--r-- | cmd2/table_creator.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/cmd2/table_creator.py b/cmd2/table_creator.py index d159ddbf..3b0db7b3 100644 --- a/cmd2/table_creator.py +++ b/cmd2/table_creator.py @@ -309,6 +309,11 @@ class TableCreator: if data_line_index > 0: wrapped_buf.write('\n') + # If the last line is empty, then add a newline and stop + if data_line_index == len(data_str_lines) - 1 and not data_line: + wrapped_buf.write('\n') + break + # Locate the styles in this line styles = utils.get_styles_in_text(data_line) @@ -351,10 +356,6 @@ class TableCreator: last_word = data_line_index == len(data_str_lines) - 1 and char_index == len(data_line) add_word(cur_word_buf.getvalue(), last_word) - # If the last line is empty, then add a newline - elif data_line_index == len(data_str_lines) - 1: - wrapped_buf.write('\n') - # Stop line loop if we've written to max_lines if total_lines == max_lines: # If this isn't the last data line and there is space |