stop the reaper task too on EVENT_HOMEASSISTANT_STOP

Author: craigbarratt

custom_components/pyscript/__init__.py

@@ -179,6 +179,8 @@ async def stop_triggers(event):
             if not global_ctx_name.startswith("file."):
                 continue
             global_ctx.stop()
+        # tell reaper task to exit (after other tasks are cancelled)
+        Function.task_cancel(None)
 
     hass.bus.async_listen(EVENT_HOMEASSISTANT_STARTED, start_triggers)
     hass.bus.async_listen(EVENT_HOMEASSISTANT_STOP, stop_triggers)

custom_components/pyscript/function.py

@@ -78,6 +78,8 @@ async def task_cancel_reaper(reaper_q):
                 try:
                     try:
                         task = await reaper_q.get()
+                        if task is None:
+                            return
                         task.cancel()
                         await task
                     except asyncio.CancelledError: