ynab-cli 0.5.2__tar.gz → 0.5.3__tar.gz

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## v0.5.3 (2025-06-09)
+
+### Fix
+
+- **tui**: moves the app settings into a separate modal dialog
+
 ## v0.5.2 (2025-06-07)
 
 ### Fix

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ynab-cli
-Version: 0.5.2
+Version: 0.5.3
 Summary: YNAB Command Line Interface
 Project-URL: Changelog, https://github.com/jeffsawatzky/ynab-cli/blob/main/CHANGELOG.md
 Project-URL: Documentation, https://jeffsawatzky.github.io/ynab-cli/

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ynab-cli"
-version = "0.5.2"
+version = "0.5.3"
 description = "YNAB Command Line Interface"
 readme = "README.md"
 requires-python = ">=3.11"

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/adapters/textual/io.py

@@ -1,13 +1,11 @@
 from typing import Any
 
-from textual import on
 from textual.app import App, ComposeResult
-from textual.containers import Vertical
-from textual.screen import ModalScreen
 from textual.widgets import Input, Label, Log, ProgressBar
 from typing_extensions import override
 
 from ynab_cli.domain.ports.io import IO, Progress
+from ynab_cli.host.textual.widgets.common.dialogs import CANCELLED, DialogForm, SaveCancelDialogScreen
 
 
 class TextualProgress(Progress):
@@ -28,9 +26,7 @@ class TextualProgress(Progress):
         self._progress_bar.update(**kwargs)
 
 
-class IOScreen(ModalScreen[str]):
-    """A simple modal screen which asks for user input."""
-
+class IODialogForm(DialogForm[str]):
     def __init__(self, prompt: str, password: bool, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
         self._prompt = prompt
@@ -38,16 +34,15 @@ class IOScreen(ModalScreen[str]):
 
     @override
     def compose(self) -> ComposeResult:
-        with Vertical():
-            yield Label(self._prompt)
-            yield Input(password=self._password)
+        yield Label(self._prompt)
+        yield Input(password=self._password)
 
-    @on(Input.Submitted)
-    def return_response(self, event: Input.Submitted) -> None:
-        self.dismiss(event.value)
+    @override
+    async def get_result(self) -> str:
+        return self.query_one(Input).value.strip()
 
 
-class TextualWorkerIO(IO):
+class TextualIO(IO):
     def __init__(self, app: App[Any], log: Log, progress_bar: ProgressBar) -> None:
         self._app = app
         self._log = log
@@ -55,7 +50,10 @@ class TextualWorkerIO(IO):
 
     @override
     async def prompt(self, prompt: str, password: bool = False) -> str:
-        return await self._app.push_screen_wait(IOScreen(prompt, password))
+        result = await self._app.push_screen_wait(SaveCancelDialogScreen(IODialogForm(prompt, password), title="Input"))
+        if result is CANCELLED:
+            return ""
+        return result
 
     @override
     async def print(self, message: str) -> None:

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/adapters/ynab/util.py

@@ -55,6 +55,9 @@ async def run_asyncio_detailed(
             new_access_token = await io.prompt(
                 prompt="API rate limit exceeded. Enter a new access token", password=True
             )
+            if not new_access_token:
+                raise e
+
             client.token = new_access_token
             response = await asyncio_detailed(*args, **kwargs)
             parsed_data = get_parsed_response_data(response)

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/domain/use_cases/budgets.py

@@ -15,6 +15,7 @@ class ListAllParams(TypedDict):
 
 async def list_all(settings: Settings, io: ports.IO, params: ListAllParams) -> AsyncIterator[models.BudgetSummary]:
     async with ynab.AuthenticatedClient(base_url=YNAB_API_URL, token=settings.ynab.access_token) as client:
+        progress_total = 0
         try:
             budgets = (
                 await util.get_asyncio_detailed(io, get_budgets.asyncio_detailed, include_accounts=True, client=client)
@@ -22,8 +23,9 @@ async def list_all(settings: Settings, io: ports.IO, params: ListAllParams) -> A
             budgets.sort(key=lambda b: b.name)
 
             progress_total = len(budgets)
+            await io.progress.update(total=progress_total)
             for budget in budgets:
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
                 yield budget
 
         except util.ApiError as e:
@@ -31,3 +33,5 @@ async def list_all(settings: Settings, io: ports.IO, params: ListAllParams) -> A
                 await io.print("API rate limit exceeded. Try again later, or get a new access token.")
             else:
                 await io.print(f"Exception when calling YNAB: {e}\n")
+        finally:
+            await io.progress.update(total=progress_total, completed=progress_total)

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/domain/use_cases/categories.py

@@ -22,6 +22,7 @@ class ListUnusedParams(TypedDict):
 
 async def list_unused(settings: Settings, io: ports.IO, params: ListUnusedParams) -> AsyncIterator[models.Category]:
     async with ynab.AuthenticatedClient(base_url=YNAB_API_URL, token=settings.ynab.access_token) as client:
+        progress_total = 0
         try:
             category_groups = (
                 await util.get_asyncio_detailed(
@@ -31,16 +32,18 @@ async def list_unused(settings: Settings, io: ports.IO, params: ListUnusedParams
             category_groups.sort(key=lambda cg: cg.name)
 
             progress_total = len(category_groups)
+            await io.progress.update(total=progress_total)
             for category_group in category_groups:
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
 
                 if _should_skip_category_or_group(category_or_group=category_group):
                     continue
 
                 progress_total += len(category_group.categories)
+                await io.progress.update(total=progress_total)
                 category_group.categories.sort(key=lambda c: c.name)
                 for category in category_group.categories:
-                    await io.progress.update(total=progress_total, advance=1)
+                    await io.progress.update(advance=1)
 
                     if _should_skip_category_or_group(category_or_group=category):
                         continue
@@ -65,6 +68,8 @@ async def list_unused(settings: Settings, io: ports.IO, params: ListUnusedParams
                 await io.print("API rate limit exceeded. Try again later, or get a new access token.")
             else:
                 await io.print(f"Exception when calling YNAB: {e}\n")
+        finally:
+            await io.progress.update(total=progress_total, completed=progress_total)
 
 
 class ListAllParams(TypedDict):
@@ -73,6 +78,7 @@ class ListAllParams(TypedDict):
 
 async def list_all(settings: Settings, io: ports.IO, params: ListAllParams) -> AsyncIterator[models.Category]:
     async with ynab.AuthenticatedClient(base_url=YNAB_API_URL, token=settings.ynab.access_token) as client:
+        progress_total = 0
         try:
             category_groups = (
                 await util.get_asyncio_detailed(
@@ -82,16 +88,18 @@ async def list_all(settings: Settings, io: ports.IO, params: ListAllParams) -> A
             category_groups.sort(key=lambda cg: cg.name)
 
             progress_total = len(category_groups)
+            await io.progress.update(total=progress_total)
             for category_group in category_groups:
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
 
                 if _should_skip_category_or_group(category_or_group=category_group):
                     continue
 
                 progress_total += len(category_group.categories)
+                await io.progress.update(total=progress_total)
                 category_group.categories.sort(key=lambda c: c.name)
                 for category in category_group.categories:
-                    await io.progress.update(total=progress_total, advance=1)
+                    await io.progress.update(advance=1)
 
                     if _should_skip_category_or_group(category_or_group=category):
                         continue
@@ -103,3 +111,5 @@ async def list_all(settings: Settings, io: ports.IO, params: ListAllParams) -> A
                 await io.print("API rate limit exceeded. Try again later, or get a new access token.")
             else:
                 await io.print(f"Exception when calling YNAB: {e}\n")
+        finally:
+            await io.progress.update(total=progress_total, completed=progress_total)

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/domain/use_cases/payees.py

@@ -51,6 +51,8 @@ async def normalize_names(
         client = ynab.AuthenticatedClient(base_url=YNAB_API_URL, token=settings.ynab.access_token)
 
     async with client:
+        progress_total = 0
+
         try:
             payees = (
                 await util.get_asyncio_detailed(io, get_payees.asyncio_detailed, settings.ynab.budget_id, client=client)
@@ -58,8 +60,9 @@ async def normalize_names(
             payees.sort(key=lambda p: p.name)
 
             progress_total = len(payees)
+            await io.progress.update(total=progress_total)
             for payee in payees:
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
 
                 if _should_skip_payee(payee=payee):
                     continue
@@ -83,6 +86,8 @@ async def normalize_names(
                 await io.print("API rate limit exceeded. Try again later, or get a new access token.")
             else:
                 await io.print(f"Exception when calling YNAB: {e}\n")
+        finally:
+            await io.progress.update(total=progress_total, completed=progress_total)
 
 
 class ListDuplicatesParams(TypedDict):
@@ -96,6 +101,8 @@ async def list_duplicates(
         client = ynab.AuthenticatedClient(base_url=YNAB_API_URL, token=settings.ynab.access_token)
 
     async with client:
+        progress_total = 0
+
         try:
             possible_duplicates: dict[tuple[UUID, str], list[tuple[UUID, str]]] = {}
 
@@ -105,8 +112,9 @@ async def list_duplicates(
             payees.sort(key=lambda p: p.name)
 
             progress_total = len(payees)
+            await io.progress.update(total=progress_total)
             for idx, payee in enumerate(payees):
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
 
                 if _should_skip_payee(payee=payee):
                     continue
@@ -139,6 +147,8 @@ async def list_duplicates(
                 await io.print("API rate limit exceeded. Try again later, or get a new access token.")
             else:
                 await io.print(f"Exception when calling YNAB: {e}\n")
+        finally:
+            await io.progress.update(total=progress_total, completed=progress_total)
 
 
 class ListUnusedParams(TypedDict):
@@ -152,6 +162,8 @@ async def list_unused(
         client = ynab.AuthenticatedClient(base_url=YNAB_API_URL, token=settings.ynab.access_token)
 
     async with client:
+        progress_total = 0
+
         try:
             payees = (
                 await util.get_asyncio_detailed(io, get_payees.asyncio_detailed, settings.ynab.budget_id, client=client)
@@ -159,8 +171,9 @@ async def list_unused(
             payees.sort(key=lambda p: p.name)
 
             progress_total = len(payees)
+            await io.progress.update(total=progress_total)
             for payee in payees:
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
 
                 if _should_skip_payee(payee=payee):
                     continue
@@ -217,8 +230,9 @@ async def list_all(
             payees.sort(key=lambda p: p.name)
 
             progress_total = len(payees)
+            await io.progress.update(total=progress_total)
             for payee in payees:
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
 
                 if _should_skip_payee(payee=payee):
                     continue
@@ -230,3 +244,5 @@ async def list_all(
                 await io.print("API rate limit exceeded. Try again later, or get a new access token.")
             else:
                 await io.print(f"Exception when calling YNAB: {e}\n")
+        finally:
+            await io.progress.update(total=progress_total, completed=progress_total)

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/domain/use_cases/transactions.py

@@ -47,6 +47,7 @@ async def apply_rules(
     settings: Settings, io: ports.IO, params: ApplyRulesParams
 ) -> AsyncIterator[tuple[models.TransactionDetail, models.SaveTransactionWithIdOrImportId]]:
     async with ynab.AuthenticatedClient(base_url=YNAB_API_URL, token=settings.ynab.access_token) as client:
+        progress_total = 0
         try:
             transactions = (
                 await util.get_asyncio_detailed(
@@ -62,8 +63,9 @@ async def apply_rules(
             save_transactions = []
 
             progress_total = len(transactions)
+            await io.progress.update(total=progress_total)
             for transaction in transactions:
-                await io.progress.update(total=progress_total, advance=1)
+                await io.progress.update(advance=1)
 
                 if _should_skip_transaction(transaction=transaction):
                     continue
@@ -91,3 +93,5 @@ async def apply_rules(
                 await io.print("API rate limit exceeded. Try again later, or get a new access token.")
             else:
                 await io.print(f"Exception when calling YNAB: {e}\n")
+        finally:
+            await io.progress.update(total=progress_total, completed=progress_total)

ynab_cli-0.5.3/src/ynab_cli/host/textual/app.py

@@ -0,0 +1,126 @@
+from typing import Any, ClassVar
+
+from textual import on, work
+from textual.app import App, ComposeResult
+from textual.binding import Binding, BindingType
+from textual.containers import Horizontal, Vertical
+from textual.reactive import reactive
+from textual.widgets import Button, Checkbox, Footer, Header, Input, Label, TabbedContent, TabPane
+from typing_extensions import override
+
+from ynab_cli.domain.settings import Settings, YnabSettings
+from ynab_cli.host.textual.widgets.categories import CategoriesTabs
+from ynab_cli.host.textual.widgets.common.dialogs import CANCELLED, DialogForm, SaveCancelDialogScreen
+from ynab_cli.host.textual.widgets.common.runnable_widget import RunnableWidget
+from ynab_cli.host.textual.widgets.payees import PayeesTabs
+from ynab_cli.host.textual.widgets.transactions import TransactionsTabs
+
+
+class CommandTabs(RunnableWidget):
+    @override
+    def compose(self) -> ComposeResult:
+        with TabbedContent():
+            with TabPane("Transactions", id="transactions"):
+                yield TransactionsTabs().data_bind(settings=CommandTabs.settings)
+            with TabPane("Categories", id="categories"):
+                yield CategoriesTabs().data_bind(settings=CommandTabs.settings)
+            with TabPane("Payees", id="payees"):
+                yield PayeesTabs().data_bind(settings=CommandTabs.settings)
+
+    @override
+    async def run_command(self) -> None:
+        tabbed_content = self.query_one(TabbedContent)
+        if tabbed_content.active_pane:
+            if tabbed_content.active_pane.id == "transactions":
+                await self.query_one(TransactionsTabs).run_command()
+            elif tabbed_content.active_pane.id == "categories":
+                await self.query_one(CategoriesTabs).run_command()
+            elif tabbed_content.active_pane.id == "payees":
+                await self.query_one(PayeesTabs).run_command()
+
+
+class SettingsDialogForm(DialogForm[Settings]):
+    def __init__(self, settings: Settings, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+        self._settings = settings
+
+    @override
+    def compose(self) -> ComposeResult:
+        with Horizontal():
+            with Vertical():
+                yield Label("Access Token")
+                yield Input(self._settings.ynab.access_token, placeholder="Access Token", id="access_token")
+            with Vertical():
+                yield Label("Budget Id")
+                yield Input(self._settings.ynab.budget_id, placeholder="Budget Id", id="budget_id")
+        with Horizontal():
+            yield Checkbox("Debug", self._settings.debug, id="debug")
+            yield Checkbox("Dry Run", self._settings.dry_run, id="dry_run")
+
+    @override
+    async def get_result(self) -> Settings:
+        """Get the result from the dialog form."""
+        access_token = self.query_one("#access_token", Input).value.strip()
+        budget_id = self.query_one("#budget_id", Input).value.strip()
+        debug = self.query_one("#debug", Checkbox).value
+        dry_run = self.query_one("#dry_run", Checkbox).value
+
+        return Settings(
+            ynab=YnabSettings(
+                access_token=access_token,
+                budget_id=budget_id,
+            ),
+            debug=debug,
+            dry_run=dry_run,
+        )
+
+
+class YnabCliApp(App[None]):
+    TITLE = "YNAB CLI"
+    CSS_PATH: ClassVar[str] = "app.tcss"
+
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("ctrl+c", "quit", "Quit", show=True, priority=True),
+        Binding("r", "run", "Run", show=True, priority=False),
+    ]
+
+    settings: reactive[Settings] = reactive(Settings())
+
+    def __init__(
+        self,
+        settings: Settings,
+        **kwargs: Any,
+    ) -> None:
+        super().__init__(**kwargs)
+        self.settings = settings
+
+    @override
+    def compose(self) -> ComposeResult:
+        yield Header(id="header")
+
+        with Vertical(id="toolbar"):
+            yield Button("Settings", id="settings_button")
+
+        with Vertical(id="main"):
+            yield CommandTabs().data_bind(settings=YnabCliApp.settings)
+
+        yield Footer(id="footer")
+
+    def on_mount(self) -> None:
+        self.query_one(CommandTabs).query_one(TabbedContent).focus()
+
+    async def action_run(self) -> None:
+        command_tabs = self.query_one(CommandTabs)
+        await command_tabs.run_command()
+
+    @on(Button.Pressed, "#settings_button")
+    async def _settings_button_pressed(self, event: Button.Pressed) -> None:
+        self._get_settings()
+
+    @work(exclusive=True)
+    async def _get_settings(self) -> None:
+        result = await self.push_screen_wait(
+            SaveCancelDialogScreen(SettingsDialogForm(self.settings), title="Settings")
+        )
+        if result is not CANCELLED:
+            self.settings = result

ynab_cli-0.5.3/src/ynab_cli/host/textual/app.tcss

@@ -0,0 +1,96 @@
+Screen {
+    
+
+    & > Vertical#toolbar {
+        border: solid $primary;
+        height: auto;
+        margin-top: 1;
+    }
+
+    & > Vertical#main {
+        height: 1fr;
+        margin-bottom: 1;
+        margin-top: 1;
+    }
+}
+
+CommandTabs {
+    height: auto;
+
+    & CategoriesTabs,
+    & PayeesTabs,
+    & TransactionsTabs {
+        border: solid $primary;
+        height: auto;
+        padding-top: 1;
+
+        & DataTable {
+            border: solid $secondary;
+            height: 1fr;
+        }
+
+        & Log {
+            border: solid $secondary;
+            height: 5;
+        }
+    }
+}
+
+RunnableWidget {
+    height: auto;
+}
+
+DialogScreen {
+    align: center middle;
+
+    & > Dialog {
+        background: $panel;
+        border-title-background: $panel;
+        border-title-color: $text;
+        border: solid $primary;
+        height: auto;
+        max-width: 90%;
+        width: auto;
+
+        & > Static {
+            margin:1;
+        }
+    }
+}
+
+Dialog {
+    border: $border;
+    height: auto;
+}
+
+SaveCancelDialog {
+    & > Horizontal {
+        align: right middle;
+        height: auto;
+
+        & Widget {
+            margin: 1;
+        }
+    }
+}
+
+DialogForm {
+    border: solid $secondary;
+    height: auto;
+
+    & Static {
+        margin: 1;
+    }
+}
+
+SettingsDialogForm {
+    height: auto;
+
+    & > Horizontal {
+        height: auto;
+
+        & > Vertical {
+            height: auto;
+        }
+    }
+}

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/host/textual/cli.py

@@ -6,11 +6,11 @@ from textual.logging import TextualHandler
 
 from ynab_cli.domain.settings import Settings, YnabSettings
 from ynab_cli.host.constants import CONTEXT_KEY_SETTINGS, ENV_PREFIX
-from ynab_cli.host.textual.app import App
+from ynab_cli.host.textual.app import YnabCliApp
 
 
 async def _tui(settings: Settings) -> None:
-    app = App(settings)
+    app = YnabCliApp(settings)
     await app.run_async()
 
 

{ynab_cli-0.5.2 → ynab_cli-0.5.3}/src/ynab_cli/host/textual/widgets/categories.py

@@ -2,9 +2,10 @@ from textual.app import ComposeResult
 from textual.widgets import DataTable, Log, ProgressBar, TabbedContent, TabPane
 from typing_extensions import override
 
-from ynab_cli.adapters.textual.io import TextualWorkerIO
+from ynab_cli.adapters.textual.io import TextualIO
 from ynab_cli.domain.use_cases import categories as use_cases
-from ynab_cli.host.textual.widgets import BaseCommand, RunnableWidget
+from ynab_cli.host.textual.widgets.common.base_command import BaseCommand
+from ynab_cli.host.textual.widgets.common.runnable_widget import RunnableWidget
 
 
 class CategoriesTabs(RunnableWidget):
@@ -48,7 +49,7 @@ class CategoriesListUnusedCommand(BaseCommand[use_cases.ListUnusedParams]):
         log = self.query_one(Log)
 
         async for category in use_cases.list_unused(
-            self.settings, TextualWorkerIO(self.app, log, progress_bar), use_case_params
+            self.settings, TextualIO(self.app, log, progress_bar), use_case_params
         ):
             table.add_row(
                 category.category_group_id,
@@ -80,7 +81,7 @@ class CategoriesListAllCommand(BaseCommand[use_cases.ListAllParams]):
         log = self.query_one(Log)
 
         async for category in use_cases.list_all(
-            self.settings, TextualWorkerIO(self.app, log, progress_bar), use_case_params
+            self.settings, TextualIO(self.app, log, progress_bar), use_case_params
         ):
             table.add_row(
                 category.category_group_id,

ynab_cli-0.5.2/src/ynab_cli/host/textual/widgets/__init__.py → ynab_cli-0.5.3/src/ynab_cli/host/textual/widgets/common/base_command.py

@@ -1,30 +1,18 @@
-import logging
 from typing import Generic, TypeVar
 
-from textual import work
+from textual import on, work
 from textual.app import ComposeResult
 from textual.containers import Vertical
 from textual.message import Message
-from textual.reactive import var
-from textual.widgets import DataTable, Log, ProgressBar, Static
+from textual.widgets import DataTable, Log, ProgressBar
 from textual.worker import Worker, WorkerState
 from typing_extensions import override
 
-from ynab_cli.domain.settings import Settings
-
-log = logging.getLogger(__name__)
-
+from ynab_cli.host.textual.widgets.common.runnable_widget import RunnableWidget
 
 PARAMS = TypeVar("PARAMS")
 
 
-class RunnableWidget(Static):
-    settings: var[Settings] = var(Settings())
-
-    async def run_command(self) -> None:
-        raise NotImplementedError
-
-
 class BaseCommand(RunnableWidget, Generic[PARAMS]):
     class CommandCompleted(Message):
         def __init__(self, worker_state: WorkerState) -> None:
@@ -50,7 +38,8 @@ class BaseCommand(RunnableWidget, Generic[PARAMS]):
     async def _run_command(self, use_case_params: PARAMS) -> None:
         raise NotImplementedError
 
-    def on_worker_state_changed(self, event: Worker.StateChanged) -> None:
+    @on(Worker.StateChanged)
+    def _worker_state_changed(self, event: Worker.StateChanged) -> None:
         log = self.query_one(Log)
         log.write_line(f"Worker state changed: {event.state}")
 

ynab_cli-0.5.3/src/ynab_cli/host/textual/widgets/common/dialogs/__init__.py

@@ -0,0 +1,10 @@
+from .base import CANCELLED, Dialog, DialogForm, DialogScreen, SaveCancelDialog, SaveCancelDialogScreen
+
+__all__ = [
+    "CANCELLED",
+    "Dialog",
+    "DialogForm",
+    "DialogScreen",
+    "SaveCancelDialog",
+    "SaveCancelDialogScreen",
+]

ynab_cli-0.5.3/src/ynab_cli/host/textual/widgets/common/dialogs/base/__init__.py

@@ -0,0 +1,12 @@
+from .dialog import Dialog, SaveCancelDialog
+from .form import DialogForm
+from .screen import CANCELLED, DialogScreen, SaveCancelDialogScreen
+
+__all__ = [
+    "CANCELLED",
+    "Dialog",
+    "DialogForm",
+    "DialogScreen",
+    "SaveCancelDialog",
+    "SaveCancelDialogScreen",
+]