Add type hints for mypy

This commit is contained in:
Isaiah Odhner 2023-07-15 21:37:16 -04:00
parent 3ff0b47650
commit 641a46e368
2 changed files with 6 additions and 5 deletions

View File

@ -4050,7 +4050,7 @@ Columns: {len(palette) // 2}
# DialogWindow has a special focus_next action that wraps within the dialog. # DialogWindow has a special focus_next action that wraps within the dialog.
# await self.run_action("focus_next", self.query_one(DialogWindow)) # await self.run_action("focus_next", self.query_one(DialogWindow))
# There may be multiple dialogs open, so we need to find the one that's focused. # There may be multiple dialogs open, so we need to find the one that's focused.
node = self.focused node: DOMNode | None = self.focused
while node is not None: while node is not None:
if isinstance(node, DialogWindow): if isinstance(node, DialogWindow):
# await self.run_action("focus_next", node) # await self.run_action("focus_next", node)

View File

@ -2,7 +2,7 @@ from typing import Any, Callable
from typing_extensions import Self from typing_extensions import Self
from textual import events, on from textual import events, on
from textual.dom import NoScreen from textual.dom import DOMNode, NoScreen
from textual.message import Message from textual.message import Message
from textual.app import ComposeResult from textual.app import ComposeResult
from textual.containers import Container from textual.containers import Container
@ -81,8 +81,8 @@ class Window(Container):
) -> None: ) -> None:
"""Initialize a window.""" """Initialize a window."""
super().__init__(*children, **kwargs) super().__init__(*children, **kwargs)
self.mouse_at_drag_start = None self.mouse_at_drag_start: Offset | None = None
self.offset_at_drag_start = None self.offset_at_drag_start: Offset | None = None
self.title_bar = WindowTitleBar(title=title, allow_maximize=allow_maximize, allow_minimize=allow_minimize) self.title_bar = WindowTitleBar(title=title, allow_maximize=allow_maximize, allow_minimize=allow_minimize)
self.content = Container(classes="window_content") self.content = Container(classes="window_content")
self.maximized = False self.maximized = False
@ -108,7 +108,7 @@ class Window(Container):
"""Returns True if widget exists and is within .buttons.""" """Returns True if widget exists and is within .buttons."""
if not widget: if not widget:
return False return False
node = widget node: DOMNode | None = widget
while node: while node:
if node.has_class("buttons"): if node.has_class("buttons"):
return True return True
@ -654,6 +654,7 @@ class MessageBox(DialogWindow):
) -> None: ) -> None:
"""Initialize the message box.""" """Initialize the message box."""
super().__init__(*children, handle_button=handle_button, **kwargs) super().__init__(*children, handle_button=handle_button, **kwargs)
self.message_widget: Widget
if isinstance(message, str): if isinstance(message, str):
self.message_widget = Static(message, markup=False) self.message_widget = Static(message, markup=False)
else: else: