diff options
Diffstat (limited to 'cmd2/table_creator.py')
| -rw-r--r-- | cmd2/table_creator.py | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/cmd2/table_creator.py b/cmd2/table_creator.py index 3df6ce37..38102a07 100644 --- a/cmd2/table_creator.py +++ b/cmd2/table_creator.py @@ -604,20 +604,20 @@ class SimpleTable(TableCreator): header = self.generate_row(inter_cell=self.inter_cell) header_buf.write(header) - # Create the divider if necessary - if self.divider_char is not None: - total_width = self.total_width() - divider_char_width = ansi.style_aware_wcswidth(self.divider_char) - - # Make divider as wide as table and use padding if width of - # divider_char does not divide evenly into table width. - divider = self.divider_char * (total_width // divider_char_width) - divider += SPACE * (total_width % divider_char_width) - - header_buf.write('\n') - header_buf.write(divider) + # Add the divider if necessary + divider = self.generate_divider() + if divider: + header_buf.write('\n' + divider) + return header_buf.getvalue() + def generate_divider(self) -> str: + """Generate divider row""" + if self.divider_char is None: + return '' + + return utils.align_left('', fill_char=self.divider_char, width=self.total_width()) + def generate_data_row(self, row_data: Sequence[Any]) -> str: """ Generate a data row |