_4_rich_example.py

from rich.console import Console
from helpers import create_source_output_table

console = Console()
print = console.print

def example_simple_console():
    # Normally you would just use this line 
    #from rich import print
    print("Hello, world!")
    print("Hello, [bold magenta]World[/bold magenta]!")
    print("Hello, 99 worlds!")
    print({"Hello": "world!", "Number": 99})
    print(["Hello", "world!", 99])
    print(["Lots of items"]*20)

def example_rich_advanced_table():
    from rich.table import Table
    user_data = [
        ["Alice", 24],
        ["Bob", 19],
        ["Charlie", 30],
        ["David", 40],
    ]
    table = Table(title="User Info")
    table.add_column("Name")
    table.add_column("Age")
    for user in user_data:
        table.add_row(user[0], str(user[1]))
    
    print(table)


def example_rich_advanced_tree():
    from rich.tree import Tree

    tree = Tree("example")
    tree.add("hello").add("world").add("!")

    foo = tree.add("foo")
    foo.add("bar")
    foo.add("baz")
    
    print(tree)


def example_inspect_basic():
    from rich import inspect
    my_list = [1, 2, 3, 4]
    print('inspect(my_list)')
    inspect(my_list, console=console)


def example_inspect_methods():
    from rich import inspect
    my_list = [1, 2, 3, 4]
    print('inspect(my_list, methods=True)')
    inspect(my_list, console=console, methods=True)


def example_inspect_all():
    from rich import inspect
    my_list = [1, 2, 3, 4]
    print('inspect(my_list, all=True)')
    inspect(my_list, console=console, all=True)


def example_traceback():
    my_number = 2
    my_string = "world"
    
    # This line will print the output directly if the program crashes
    #from rich.traceback import install
    #install(show_locals=True)
    #print(my_number + my_string)

    try:
        print(my_number + my_string)
    except Exception as e:
        console.print_exception(show_locals=False)
    
    try:
        print(my_number + my_string)
    except Exception as e:
        console.print_exception(show_locals=True)

examples = [
    example_simple_console, 
    # example_rich_advanced_table, 
    example_inspect_basic, 
    example_inspect_methods, 
    example_traceback
]
def main():
    for example in examples:
        console.clear()
        print(create_source_output_table(example, console))

        is_not_last_item = example != examples[-1]
        if is_not_last_item:
            input("Press Enter to continue to the next example...")

if __name__ == '__main__':
    main()