Type curses windows

2 files changed, 20 insertions, 5 deletions

diff --git a/poezio/tabs/basetabs.py b/poezio/tabs/basetabs.py
index 0cae3c6a..2efc77a0 100644
--- a/poezio/tabs/basetabs.py
+++ b/poezio/tabs/basetabs.py
@@ -21,7 +21,15 @@ import time
 from math import ceil, log10
 from datetime import datetime
 from xml.etree import cElementTree as ET
-from typing import Any, Callable, Dict, List, Optional, Union
+from typing import (
+    Any,
+    Callable,
+    Dict,
+    List,
+    Optional,
+    Union,
+    TYPE_CHECKING,
+)
 
 from poezio import mam, poopt, timed_events, xhtml, windows
 from poezio.core.structs import Command, Completion, Status
@@ -35,6 +43,9 @@ from poezio.windows.funcs import truncate_nick
 
 from slixmpp import JID, InvalidJID, Message
 
+if TYPE_CHECKING:
+    from _curses import _CursesWindow
+
 log = logging.getLogger(__name__)
 
 # getters for tab colors (lambdas, so that they are dynamic)
@@ -176,7 +187,7 @@ class Tab:
             self._state = 'normal'
 
     @staticmethod
-    def resize(scr):
+    def resize(scr: '_CursesWindow'):
         Tab.height, Tab.width = scr.getmaxyx()
         windows.base_wins.TAB_WIN = scr
 
diff --git a/poezio/windows/base_wins.py b/poezio/windows/base_wins.py
index 6dabd7b8..361ae338 100644
--- a/poezio/windows/base_wins.py
+++ b/poezio/windows/base_wins.py
@@ -7,7 +7,7 @@ the text window, the roster window, etc.
 A Tab (see the poezio.tabs module) is composed of multiple Windows
 """
 
-TAB_WIN = None
+TAB_WIN = None  # type: _CursesWindow
 
 import logging
 log = logging.getLogger(__name__)
@@ -15,7 +15,8 @@ log = logging.getLogger(__name__)
 import curses
 import string
 
-from typing import Optional, Tuple
+from contextlib import contextmanager
+from typing import Optional, Tuple, TYPE_CHECKING
 
 from poezio.theming import to_curses_attr, read_tuple
 
@@ -24,6 +25,9 @@ FORMAT_CHAR = '\x19'
 # I guess. But maybe we can find better chars that are even less risky.
 format_chars = '\x0E\x0F\x10\x11\x12\x13\x14\x15\x16\x17\x18\x1A'
 
+if TYPE_CHECKING:
+    from _curses import _CursesWindow
+
 
 class DummyWin:
     def __getattribute__(self, name: str):
@@ -40,7 +44,7 @@ class Win:
     __slots__ = ('_win', 'height', 'width', 'y', 'x')
 
     def __init__(self) -> None:
-        self._win = None
+        self._win = None  # type: _CursesWindow
         self.height, self.width = 0, 0
 
     def _resize(self, height: int, width: int, y: int, x: int) -> None: