SpawnProcess: make _main cancel pipe_logger when appropriate

Signed-off-by: Zac Medico <zmedico@gentoo.org>
zmedico · Feb 29, 2020 · 6eb6500 · 6eb6500
1 parent 10cc6fe
commit 6eb6500
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/_emerge/SpawnProcess.py b/lib/_emerge/SpawnProcess.py
@@ -155,8 +155,13 @@ def _async_start(self):
 
 	@coroutine
 	def _main(self, pipe_logger):
-		if pipe_logger.poll() is None:
-			yield pipe_logger.async_wait()
+		try:
+			if pipe_logger.poll() is None:
+				yield pipe_logger.async_wait()
+		except asyncio.CancelledError:
+			if pipe_logger.poll() is None:
+				pipe_logger.cancel()
+			raise
 
 	def _main_exit(self, main_task):
 		try: