refactoring

2 files changed, 15 insertions, 14 deletions

diff --git a/kvmd/aioproc.py b/kvmd/aioproc.py
index 65502d35..ecf6230e 100644
--- a/kvmd/aioproc.py
+++ b/kvmd/aioproc.py
@@ -56,6 +56,19 @@ async def log_process(cmd: List[str], logger: logging.Logger) -> asyncio.subproc
     return proc
 
 
+async def log_stdout_infinite(proc: asyncio.subprocess.Process, logger: logging.Logger) -> None:  # pylint: disable=no-member
+    empty = 0
+    async for line_bytes in proc.stdout:  # type: ignore
+        line = line_bytes.decode(errors="ignore").strip()
+        if line:
+            logger.info("Console: %s", line)
+            empty = 0
+        else:
+            empty += 1
+            if empty == 100:  # asyncio bug
+                raise RuntimeError("asyncio process: too many empty lines")
+
+
 def ignore_sigint() -> None:
     signal.signal(signal.SIGINT, signal.SIG_IGN)
 
diff --git a/kvmd/apps/kvmd/streamer.py b/kvmd/apps/kvmd/streamer.py
index afddb1db..48336c98 100644
--- a/kvmd/apps/kvmd/streamer.py
+++ b/kvmd/apps/kvmd/streamer.py
@@ -375,23 +375,11 @@ class Streamer:  # pylint: disable=too-many-instance-attributes
         while True:  # pylint: disable=too-many-nested-blocks
             try:
                 await self.__start_streamer_proc()
-
-                empty = 0
-                async for line_bytes in self.__streamer_proc.stdout:  # type: ignore
-                    line = line_bytes.decode(errors="ignore").strip()
-                    if line:
-                        logger.info("Console: %s", line)
-                        empty = 0
-                    else:
-                        empty += 1
-                        if empty == 100:  # asyncio bug
-                            raise RuntimeError("Streamer/asyncio: too many empty lines")
-
+                assert self.__streamer_proc is not None
+                await aioproc.log_stdout_infinite(self.__streamer_proc, logger)
                 raise RuntimeError("Streamer unexpectedly died")
-
             except asyncio.CancelledError:
                 break
-
             except Exception:
                 if self.__streamer_proc:
                     logger.exception("Unexpected streamer error: pid=%d", self.__streamer_proc.pid)