Formatted Table String

2022-08-22.py

question = """
Given a string that represents a markdown table, 
return a formatted markdown table. A formatted markdown 
table means that the width of each column is the width of 
the longest cell in the column.

Example:

Input:
| Syntax | Description |
| --- | ----------- |
| Header | Title |
| Paragraph | Text |

Output:
| Syntax    | Description |
| --------- | ----------- |
| Header    | Title       |
| Paragraph | Text        |

"""

import re

def formattedTable(table):
    p = re.compile(r'^-+$')
    table_data = []
    row_widths = []
    for row in table.split("\n"):
        table_row = []
        i = 0
        for col in row.split("|"):
            col = re.sub(r'^ ', '', col)
            col = re.sub(r' $', '', col)
            if col != "":
                if len(row_widths) <= i:
                    row_widths.append(len(col))
                elif row_widths[i] < len(col):
                    row_widths[i] = len(col)
                m = p.search(col)
                if m:
                    col = '-'
                table_row.append(col)
                i += 1
        if table_row != []:
            table_data.append(table_row)

    for row in table_data:
        print("| ", end='')
        for i in range(0, len(row)):
            if row[i] == '-':
                print(row[i] * row_widths[i], end='')
            else:
                print(row[i] + " " * (row_widths[i] - len(row[i])), end='')
            print(" | ", end='')
        print("")
    return

table = """
| Syntax | Description |
| --- | ----------- |
| Header | Title |
| Paragraph | Text |
"""

formattedTable(table)