Add a context manager in Win for colored text

1 files changed, 13 insertions, 0 deletions

diff --git a/poezio/windows/base_wins.py b/poezio/windows/base_wins.py
index 6a689067..d6c72912 100644
--- a/poezio/windows/base_wins.py
+++ b/poezio/windows/base_wins.py
@@ -77,6 +77,19 @@ class Win:
             # of the screen.
             pass
 
+    @contextmanager
+    def colored_text(self, color: Optional[Tuple]=None, attr: Optional[int]=None):
+        """Context manager which sets up an attr/color when inside"""
+        if attr is None:
+            if color is not None:
+                attr = to_curses_attr(color)
+            else:
+                yield None
+                return
+        self._win.attron(attr)
+        yield None
+        self._win.attroff(attr)
+
     def addstr(self, *args) -> None:
         """
         Safe call to addstr