Skip to content

Commit c892897

Browse files
committed
Correctly shutdown receiver worker task
1 parent 54f9b1d commit c892897

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

plugwise_usb/connection/receiver.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -159,10 +159,12 @@ async def _stop_running_tasks(self) -> None:
159159
"""Cancel and stop any running task."""
160160
for task in self._delayed_processing_tasks.values():
161161
task.cancel()
162-
cancel_response = StickResponse()
163-
cancel_response.priority = Priority.CANCEL
164-
await self._receive_queue.put(cancel_response)
165-
await self._receive_worker_task
162+
if self._receive_worker_task is not None and not self._receive_worker_task.done():
163+
cancel_response = StickResponse()
164+
cancel_response.priority = Priority.CANCEL
165+
await self._receive_queue.put(cancel_response)
166+
await self._receive_worker_task
167+
self._receive_worker_task = None
166168

167169
def data_received(self, data: bytes) -> None:
168170
"""Receive data from USB-Stick connection.

0 commit comments

Comments
 (0)