43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# -*- encoding: utf-8 -*-
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
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 class="table"><code>'
|
|
for i, row in enumerate(rows):
|
|
if i == 1:
|
|
for j, cell in enumerate(row):
|
|
text += '|' + '-' * (cols_width[j] + 2)
|
|
text += '|\n'
|
|
|
|
for j, cell in enumerate(row):
|
|
text += '| ' + ''.join(str(node) for node in cell.contents) \
|
|
+ ' ' * (cols_width[j] - len(cell.text) + 1)
|
|
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))
|