From e8e7ad3b7547f9598e611c266f86a915c8162fb2 Mon Sep 17 00:00:00 2001 From: Kovid Goyal Date: Thu, 16 Sep 2021 18:34:51 +0530 Subject: [PATCH] DRYer --- kittens/tui/loop.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/kittens/tui/loop.py b/kittens/tui/loop.py index 01ab5028c..3d72099fc 100644 --- a/kittens/tui/loop.py +++ b/kittens/tui/loop.py @@ -329,12 +329,15 @@ class Loop: self.handler.image_manager.handle_response(apc) # }}} + @property + def total_pending_bytes_to_write(self) -> int: + return sum(map(len, self.write_buf)) + def _write_ready(self, handler: Handler, fd: int) -> None: if len(self.write_buf) > self.iov_limit: self.write_buf[self.iov_limit - 1] = b''.join(self.write_buf[self.iov_limit - 1:]) del self.write_buf[self.iov_limit:] - sizes = tuple(map(len, self.write_buf)) - total_size = sum(sizes) + total_size = self.total_pending_bytes_to_write if total_size: try: written = os.writev(fd, self.write_buf)