codekingpro/portable-devtools

1import colorsys2import io3from time import process_time4 5from rich import box6from rich.color import Color7from rich.console import Console, ConsoleOptions, Group, RenderableType, RenderResult8from rich.markdown import Markdown9from rich.measure import Measurement10from rich.pretty import Pretty11from rich.segment import Segment12from rich.style import Style13from rich.syntax import Syntax14from rich.table import Table15from rich.text import Text16 17 18class ColorBox:19    def __rich_console__(20        self, console: Console, options: ConsoleOptions21    ) -> RenderResult:22        for y in range(0, 5):23            for x in range(options.max_width):24                h = x / options.max_width25                l = 0.1 + ((y / 5) * 0.7)26                r1, g1, b1 = colorsys.hls_to_rgb(h, l, 1.0)27                r2, g2, b2 = colorsys.hls_to_rgb(h, l + 0.7 / 10, 1.0)28                bgcolor = Color.from_rgb(r1 * 255, g1 * 255, b1 * 255)29                color = Color.from_rgb(r2 * 255, g2 * 255, b2 * 255)30                yield Segment("▄", Style(color=color, bgcolor=bgcolor))31            yield Segment.line()32 33    def __rich_measure__(34        self, console: "Console", options: ConsoleOptions35    ) -> Measurement:36        return Measurement(1, options.max_width)37 38 39def make_test_card() -> Table:40    """Get a renderable that demonstrates a number of features."""41    table = Table.grid(padding=1, pad_edge=True)42    table.title = "Rich features"43    table.add_column("Feature", no_wrap=True, justify="center", style="bold red")44    table.add_column("Demonstration")45 46    color_table = Table(47        box=None,48        expand=False,49        show_header=False,50        show_edge=False,51        pad_edge=False,52    )53    color_table.add_row(54        (55            "✓ [bold green]4-bit color[/]\n"56            "✓ [bold blue]8-bit color[/]\n"57            "✓ [bold magenta]Truecolor (16.7 million)[/]\n"58            "✓ [bold yellow]Dumb terminals[/]\n"59            "✓ [bold cyan]Automatic color conversion"60        ),61        ColorBox(),62    )63 64    table.add_row("Colors", color_table)65 66    table.add_row(67        "Styles",68        "All ansi styles: [bold]bold[/], [dim]dim[/], [italic]italic[/italic], [underline]underline[/], [strike]strikethrough[/], [reverse]reverse[/], and even [blink]blink[/].",69    )70 71    lorem = "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Quisque in metus sed sapien ultricies pretium a at justo. Maecenas luctus velit et auctor maximus."72    lorem_table = Table.grid(padding=1, collapse_padding=True)73    lorem_table.pad_edge = False74    lorem_table.add_row(75        Text(lorem, justify="left", style="green"),76        Text(lorem, justify="center", style="yellow"),77        Text(lorem, justify="right", style="blue"),78        Text(lorem, justify="full", style="red"),79    )80    table.add_row(81        "Text",82        Group(83            Text.from_markup(84                """Word wrap text. Justify [green]left[/], [yellow]center[/], [blue]right[/] or [red]full[/].\n"""85            ),86            lorem_table,87        ),88    )89 90    def comparison(renderable1: RenderableType, renderable2: RenderableType) -> Table:91        table = Table(show_header=False, pad_edge=False, box=None, expand=True)92        table.add_column("1", ratio=1)93        table.add_column("2", ratio=1)94        table.add_row(renderable1, renderable2)95        return table96 97    table.add_row(98        "Asian\nlanguage\nsupport",99        ":flag_for_china:  该库支持中文，日文和韩文文本！\n:flag_for_japan:  ライブラリは中国語、日本語、韓国語のテキストをサポートしています\n:flag_for_south_korea:  이 라이브러리는 중국어, 일본어 및 한국어 텍스트를 지원합니다",100    )101 102    markup_example = (103        "[bold magenta]Rich[/] supports a simple [i]bbcode[/i]-like [b]markup[/b] for [yellow]color[/], [underline]style[/], and emoji! "104        ":+1: :apple: :ant: :bear: :baguette_bread: :bus: "105    )106    table.add_row("Markup", markup_example)107 108    example_table = Table(109        show_edge=False,110        show_header=True,111        expand=False,112        row_styles=["none", "dim"],113        box=box.SIMPLE,114    )115    example_table.add_column("[green]Date", style="green", no_wrap=True)116    example_table.add_column("[blue]Title", style="blue")117    example_table.add_column(118        "[cyan]Production Budget",119        style="cyan",120        justify="right",121        no_wrap=True,122    )123    example_table.add_column(124        "[magenta]Box Office",125        style="magenta",126        justify="right",127        no_wrap=True,128    )129    example_table.add_row(130        "Dec 20, 2019",131        "Star Wars: The Rise of Skywalker",132        "$275,000,000",133        "$375,126,118",134    )135    example_table.add_row(136        "May 25, 2018",137        "[b]Solo[/]: A Star Wars Story",138        "$275,000,000",139        "$393,151,347",140    )141    example_table.add_row(142        "Dec 15, 2017",143        "Star Wars Ep. VIII: The Last Jedi",144        "$262,000,000",145        "[bold]$1,332,539,889[/bold]",146    )147    example_table.add_row(148        "May 19, 1999",149        "Star Wars Ep. [b]I[/b]: [i]The phantom Menace",150        "$115,000,000",151        "$1,027,044,677",152    )153 154    table.add_row("Tables", example_table)155 156    code = '''\157def iter_last(values: Iterable[T]) -> Iterable[Tuple[bool, T]]:158    """Iterate and generate a tuple with a flag for last value."""159    iter_values = iter(values)160    try:161        previous_value = next(iter_values)162    except StopIteration:163        return164    for value in iter_values:165        yield False, previous_value166        previous_value = value167    yield True, previous_value'''168 169    pretty_data = {170        "foo": [171            3.1427,172            (173                "Paul Atreides",174                "Vladimir Harkonnen",175                "Thufir Hawat",176            ),177        ],178        "atomic": (False, True, None),179    }180    table.add_row(181        "Syntax\nhighlighting\n&\npretty\nprinting",182        comparison(183            Syntax(code, "python3", line_numbers=True, indent_guides=True),184            Pretty(pretty_data, indent_guides=True),185        ),186    )187 188    markdown_example = """\189# Markdown190 191Supports much of the *markdown* __syntax__!192 193- Headers194- Basic formatting: **bold**, *italic*, `code`195- Block quotes196- Lists, and more...197    """198    table.add_row(199        "Markdown", comparison("[cyan]" + markdown_example, Markdown(markdown_example))200    )201 202    table.add_row(203        "+more!",204        """Progress bars, columns, styled logging handler, tracebacks, etc...""",205    )206    return table207 208 209if __name__ == "__main__":  # pragma: no cover210    console = Console(211        file=io.StringIO(),212        force_terminal=True,213    )214    test_card = make_test_card()215 216    # Print once to warm cache217    start = process_time()218    console.print(test_card)219    pre_cache_taken = round((process_time() - start) * 1000.0, 1)220 221    console.file = io.StringIO()222 223    start = process_time()224    console.print(test_card)225    taken = round((process_time() - start) * 1000.0, 1)226 227    c = Console(record=True)228    c.print(test_card)229 230    print(f"rendered in {pre_cache_taken}ms (cold cache)")231    print(f"rendered in {taken}ms (warm cache)")232 233    from rich.panel import Panel234 235    console = Console()236 237    sponsor_message = Table.grid(padding=1)238    sponsor_message.add_column(style="green", justify="right")239    sponsor_message.add_column(no_wrap=True)240 241    sponsor_message.add_row(242        "Textualize",243        "[u blue link=https://github.com/textualize]https://github.com/textualize",244    )245    sponsor_message.add_row(246        "Twitter",247        "[u blue link=https://twitter.com/willmcgugan]https://twitter.com/willmcgugan",248    )249 250    intro_message = Text.from_markup(251        """\252We hope you enjoy using Rich!253 254Rich is maintained with [red]:heart:[/] by [link=https://www.textualize.io]Textualize.io[/]255 256- Will McGugan"""257    )258 259    message = Table.grid(padding=2)260    message.add_column()261    message.add_column(no_wrap=True)262    message.add_row(intro_message, sponsor_message)263 264    console.print(265        Panel.fit(266            message,267            box=box.ROUNDED,268            padding=(1, 2),269            title="[b red]Thanks for trying out Rich!",270            border_style="bright_blue",271        ),272        justify="center",273    )274