textual/sandbox/will/basic.py

from rich.console import RenderableType
from rich.style import Style
from rich.syntax import Syntax
from rich.text import Text

from textual.app import App
from textual.reactive import Reactive
from textual.widget import Widget
from textual.widgets import Static, DataTable

CODE = '''
class Offset(NamedTuple):
    """A point defined by x and y coordinates."""

    x: int = 0
    y: int = 0

    @property
    def is_origin(self) -> bool:
        """Check if the point is at the origin (0, 0)"""
        return self == (0, 0)

    def __bool__(self) -> bool:
        return self != (0, 0)

    def __add__(self, other: object) -> Offset:
        if isinstance(other, tuple):
            _x, _y = self
            x, y = other
            return Offset(_x + x, _y + y)
        return NotImplemented

    def __sub__(self, other: object) -> Offset:
        if isinstance(other, tuple):
            _x, _y = self
            x, y = other
            return Offset(_x - x, _y - y)
        return NotImplemented

    def __mul__(self, other: object) -> Offset:
        if isinstance(other, (float, int)):
            x, y = self
            return Offset(int(x * other), int(y * other))
        return NotImplemented
'''


lorem_short = """Lorem ipsum dolor sit amet, consectetur adipiscing elit. In velit liber a a a, volutpat nec hendrerit at, faucibus in odio. Aliquam hendrerit nibh sed quam volutpat maximus. Nullam suscipit convallis lorem quis sodales. In tristique lobortis ante et dictum. Ut at finibus ipsum."""
lorem = (
    lorem_short
    + """ In urna dolor, placerat et mi facilisis, congue sollicitudin massa. Phasellus felis turpis, cursus eu lectus et, porttitor malesuada augue. Sed feugiat volutpat velit, sollicitudin fringilla velit bibendum faucibus. Lorem ipsum dolor sit amet, consectetur adipiscing elit. In velit libero, volutpat nec hendrerit at, faucibus in odio. Aliquam hendrerit nibh sed quam volutpat maximus. Nullam suscipit convallis lorem quis sodales. In tristique lobortis ante et dictum. Ut at finibus ipsum. In urna dolor, placerat et mi facilisis, congue sollicitudin massa. Phasellus felis turpis, cursus eu lectus et, porttitor malesuada augue. Sed feugiat volutpat velit, sollicitudin fringilla velit bibendum faucibus. """
)

lorem_short_text = Text.from_markup(lorem_short)
lorem_long_text = Text.from_markup(lorem * 2)


class TweetHeader(Widget):
    def render(self) -> RenderableType:
        return Text("Lorem Impsum", justify="center")


class TweetBody(Widget):
    short_lorem = Reactive(False)

    def render(self) -> Text:
        return lorem_short_text if self.short_lorem else lorem_long_text


class Tweet(Widget):
    pass


class OptionItem(Widget):
    def render(self) -> Text:
        return Text("Option")


class Error(Widget):
    def render(self) -> Text:
        return Text("This is an error message", justify="center")


class Warning(Widget):
    def render(self) -> Text:
        return Text("This is a warning message", justify="center")


class Success(Widget):
    def render(self) -> Text:
        return Text("This is a success message", justify="center")


class BasicApp(App, css_path="basic.css"):
    """A basic app demonstrating CSS"""

    def on_load(self):
        """Bind keys here."""
        self.bind("s", "toggle_class('#sidebar', '-active')")

    def on_mount(self):
        """Build layout here."""

        table = DataTable()
        self.scroll_to_target = Tweet(TweetBody())
        self.mount(
            header=Static(
                Text.from_markup(
                    "[b]This is a [u]Textual[/u] app, running in the terminal"
                ),
            ),
            content=Widget(
                Tweet(TweetBody()),
                Widget(
                    Static(Syntax(CODE, "python"), classes="code"),
                    classes="scrollable",
                ),
                table,
                Error(),
                Tweet(TweetBody(), classes="scrollbar-size-custom"),
                Warning(),
                Tweet(TweetBody(), classes="scroll-horizontal"),
                Success(),
                Tweet(TweetBody(), classes="scroll-horizontal"),
                Tweet(TweetBody(), classes="scroll-horizontal"),
                Tweet(TweetBody(), classes="scroll-horizontal"),
                Tweet(TweetBody(), classes="scroll-horizontal"),
                Tweet(TweetBody(), classes="scroll-horizontal"),
            ),
            footer=Widget(),
            sidebar=Widget(
                Widget(classes="title"),
                Widget(classes="user"),
                OptionItem(),
                OptionItem(),
                OptionItem(),
                Widget(classes="content"),
            ),
        )
        table.add_column("Foo", width=80)
        table.add_column("Bar", width=50)
        table.add_column("Baz", width=40)
        table.zebra_stripes = True
        for n in range(100):
            table.add_row(
                f"{n} This is an example of a [b]DataTable widget[/b] within a larger [bold magenta]Textual UI",
                "Cells may contain just about any kind of data",
                "Where there is a Will there is a Way",
                height=1,
            )

    async def on_key(self, event) -> None:
        await self.dispatch_key(event)

    def key_d(self):
        self.dark = not self.dark

    async def key_q(self):
        await self.shutdown()

    def key_x(self):
        self.panic(self.tree)

    def key_escape(self):
        self.app.bell()

    def key_t(self):
        # Pressing "t" toggles the content of the TweetBody widget, from a long "Lorem ipsum..." to a shorter one.
        tweet_body = self.query("TweetBody").first()
        tweet_body.short_lorem = not tweet_body.short_lorem

    def key_v(self):
        self.get_child(id="content").scroll_to_widget(self.scroll_to_target)

    def key_space(self):
        self.bell()


app = BasicApp()

if __name__ == "__main__":
    app.run()

    from textual.geometry import Region
    from textual.color import Color

    print(Region.intersection.cache_info())
    print(Region.overlaps.cache_info())
    print(Region.union.cache_info())
    print(Region.split_vertical.cache_info())
    print(Region.__contains__.cache_info())
    from textual.css.scalar import Scalar

    print(Scalar.resolve_dimension.cache_info())

    from rich.style import Style

    print(Style._add.cache_info())