teilchensammler-cli/src/teilchensammler_cli/main.py

"""
This is where the application is implemented.
"""

import logging

from textual import on
from textual.app import App, ComposeResult
from textual.containers import HorizontalGroup
from textual.logging import TextualHandler
from textual.screen import Screen
from textual.widget import Widget
from textual.widgets import Button, DataTable, Footer, Header, Input, Static

from .database import create_db_and_tables
from .models import add_to_database, load_initial_data, make_teilchen_input

logging.basicConfig(
    level="NOTSET",
    handlers=[TextualHandler()],
)
logger = logging.getLogger(__name__)

TEILCHEN_DATA_HEADER = "pk Name Description Number Tags".split()


class SammlerApp(App):
    async def on_mount(self) -> None:
        create_db_and_tables()
        self.push_screen(AddInventoryScreen())


class SearchBar(Static):
    DEFAULT_CSS = """
    #teilchen-input {
        width: 4fr;
    }
    #button-search, #button-add {
        width: 1fr;
    }
    """

    def compose(self) -> ComposeResult:
        with HorizontalGroup(id="search-bar-widget"):
            yield Input(
                placeholder='This is a name. "This is the description" #these #are #tags',
                tooltip=(
                    "Enter a name followed by a period, then a description "
                    'enclosed in double quotes ("). You should use #hashtags for any meta information. '
                    "Hashtags can be placed anywhere."
                ),
                id="teilchen-input",
                type="text",
            )
            yield Button("Add", variant="success", classes="search-bar-buttons", id="button-add")
            yield Button(
                "Search",
                variant="default",
                classes="search-bar-buttons",
                id="button-search",
            )

    @on(Input.Submitted)
    async def parse_input(self, event: Input.Submitted) -> None:

        if not (tc := await make_teilchen_input(event.value)):
            return

        event.input.value = ""

        teilchen = await add_to_database(tc)
        self.screen.query_one(DataTable).add_row(
            teilchen.id, teilchen.name, teilchen.description, teilchen.number, teilchen.tags
        )


class SearchResults(Widget):
    def compose(self) -> ComposeResult:
        yield DataTable(id="table-search-result", cursor_type="row", zebra_stripes=True)

    async def on_mount(self) -> None:
        table: DataTable = self.query_one(DataTable)
        table.add_columns(*TEILCHEN_DATA_HEADER)
        table.add_rows(await load_initial_data())


class AddInventoryScreen(Screen):
    def compose(self) -> ComposeResult:
        yield Header()
        yield SearchBar()
        yield SearchResults()
        yield Footer()


# so we can import it without running it
app = SammlerApp()


def main() -> None:
    app.run()


if __name__ == "__main__":
    main()