73 lines
1.6 KiB
Python
73 lines
1.6 KiB
Python
# -*- 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><table></code></td>
|
|
<td><code><e></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))
|