# -*- encoding: utf-8 -*-

from bs4 import BeautifulSoup

html = """

<table>
<thead>
<tr>
  <th>ID</th>
  <th>Name</th>
</tr>
</thead>
<tbody>
<tr>
  <td>56</td>
  <td>Matt</td>
</tr>
<tr>
  <td>42</td>
  <td>Colin</td>
</tr>
<tr>
  <td>23</td>
  <td>Lisa</td>
</tr>
<tr>
  <td>45</td>
  <td>John</td>
</tr>
<tr>
  <td><code>&lt;table&gt;</code></td>
  <td><code>&lt;e&gt;</code></td>
</tr>
</tbody>
</table>

<p><a href="https://upload.wikimedia.org/wikipedia/en/4/4c/Sublime_Text_Logo.png">Sublime Text Logo</a></p>
"""

def python_table(s_table):
    """Transform BeautifulSoup table into list of list"""
    rows = []
    for row in s_table.find_all('tr'):
        # rows.append(list(map( lambda td: td.text, row.find_all(['th', 'td']) )))
        rows.append(row.find_all(['th', 'td']))
    return rows

def pre_table(s_table):
    rows = python_table(s_table)
    cols_width = [len(cell) for cell in rows[0]]
    for j, row in enumerate(rows):
        for i, cell in enumerate(row):
            if cols_width[i] < len(cell.text):
                cols_width[i] = len(cell.text)
    text = '<pre><code>'
    for i, row in enumerate(rows):
        for j, cell in enumerate(row):
            text += '| ' + ''.join(str(node) for node in cell.contents) + ' ' * (cols_width[j] - len(cell.text))
        text += '|\n'
    text += '</pre></code>'
    return text

def pre_tables(html):
    soup = BeautifulSoup(html, 'html.parser')
    for table in soup.find_all('table'):
        table.replace_with(BeautifulSoup(pre_table(table), 'html.parser'))
    return str(soup)

if __name__ == "__main__":
    # CSW: ignore
    print(pre_tables(html))