MDtables base OK. Use bs4. #12

lib/pre_tables.py

@@ -0,0 +1,72 @@
+# -*- 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))