capture: Fix close_async in case the stderr task is still running

KimiNewt · Nov 2, 2023 · 7142c5b · 7142c5b
1 parent e7da566
commit 7142c5b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/pyshark/capture/capture.py b/src/pyshark/capture/capture.py
@@ -1,4 +1,5 @@
 import asyncio
+import contextlib
 import os
 import threading
 import subprocess
@@ -398,7 +399,10 @@ async def close_async(self):
         self._running_processes.clear()
 
         # Wait for all stderr handling to finish
-        await asyncio.gather(*self._stderr_handling_tasks)
+        for task in self._stderr_handling_tasks:
+            task.cancel()
+            with contextlib.suppress(asyncio.CancelledError):
+                await task
 
     def __del__(self):
         if self._running_processes: