Print prettified tables with SQLite and SQLAlchemy

pretty_print_tables.py

import sqlite3
import sqlalchemy

def print_formatted_table(table: list[list]):
    column_lengths = [
        max(len(table[i][column_num]) for i in range(len(table)))
        for column_num in range(len(table[0]))
    ]
    for row in table:
        print("|", end=" ")
        for item, column_length in zip(row, column_lengths):
            print(item.rjust(column_length), "|", end=" ")
        print()


def print_sqlite_cursor_as_table(title: str, cursor: sqlite3.Cursor):
    headers = [d[0] for d in cursor.description]
    
    table: list[list[str]] = [headers]
    for row in cursor.fetchall():
        table.append(list(map(str, row)))
    
    print(title)
    print_formatted_table(table)
    print()


def print_sqlalchemy_cursor_as_table(
    connection: sqlalchemy.Connection,
    title: str,
    statement: sqlalchemy.Select,
):
    cursor = connection.execute(statement)

    headers = list(cursor.keys())

    table: list[list[str]] = [headers]
    for row in cursor.fetchall():
        table.append(list(map(str, row)))
    
    print(title)
    print_formatted_table(table)
    print()