time-slug-grid.py

#!/usr/bin/env python

from bs4 import BeautifulSoup
from click import command
from utz import *


def pretty_print_html(soup, indent=0):
    if isinstance(soup, str):
        print(" " * indent + soup.strip())
        return

    if soup.name == '[document]':
        for child in soup.children:
            if not isinstance(child, str) or child.strip():
                pretty_print_html(child, indent)
        return

    tag = soup.name
    attrs = " ".join([f'{k}="{v}"' for k, v in soup.attrs.items()])
    prefix = f"<{tag}{' ' + attrs if attrs else ''}>"

    if tag in ['td', 'th']:
        content = "".join(str(child) for child in soup.children)
        print(" " * indent + prefix + content + f"</{tag}>")
        return

    print(" " * indent + prefix)
    for child in soup.children:
        if not isinstance(child, str) or child.strip():
            pretty_print_html(child, indent + 2)
    print(" " * indent + f"</{tag}>")


@command()
def main():
    n = now()
    df = (
        pd.DataFrame([
            dict(
                unit=u,
                **{
                    b: c(getattr(n, u))
                    for b, c in { 'b62': b62, 'b64': b64, 'b90': b90 }.items()
                }
            )
            for u in ['s', 'ds', 'cs', 'ms', 'us']
        ])
        .set_index('unit')
    )
    html = df.to_html()
    soup = BeautifulSoup(html, 'html.parser')

    # Find the table and its header rows
    table = soup.find('table')
    table.attrs['style'] = "text-align: right"
    del table.attrs['class']
    header_rows = table.find_all('tr')[:2]

    # Combine the content of the two header rows
    combined_headers = []
    for i in range(len(header_rows[0].find_all('th'))):
        header1 = header_rows[0].find_all('th')[i].text.strip()
        header2 = header_rows[1].find_all('th')[i].text.strip()
        combined_header = f"{header1} {header2}".strip()
        combined_headers.append(combined_header)

    # Remove the second header row
    header_rows[1].decompose()

    # Update the first header row with the combined content
    for i, th in enumerate(header_rows[0].find_all('th')):
        th.string = combined_headers[i]

    # Wrap each <td>'s content in a `<code>` tag
    for td in table.find_all('td'):
        code_tag = soup.new_tag('code')
        code_tag.string = td.text.strip()
        td.string = ''
        td.append(code_tag)

    pretty_print_html(soup)


if __name__ == '__main__':
    main()