Fix graceful shutdown of daemon jobs (#315)

Author: droserasprout

CHANGELOG.md

@@ -6,6 +6,7 @@
 
 * context: Fixed reporting incorrect reindexing reason.
 * exceptions: Fixed crash with `FrozenInstanceError` when exception is raised from callback.
+* jobs: Fixed graceful shutdown of daemon jobs.
 
 ### Improved
 

src/dipdup/scheduler.py

@@ -1,3 +1,4 @@
+import asyncio
 import json
 from datetime import datetime
 from functools import partial
@@ -48,11 +49,13 @@ async def _job_wrapper(ctx: DipDupContext, *args, **kwargs) -> None:
             logger=logger,
             hook_config=hook_config,
         )
-
-        await job_ctx.fire_hook(hook_config.callback, *args, **kwargs)
-
-        if job_config.daemon:
-            raise ConfigurationError('Daemon jobs are intended to run forever')
+        try:
+            await job_ctx.fire_hook(hook_config.callback, *args, **kwargs)
+        except asyncio.CancelledError:
+            pass
+        else:
+            if job_config.daemon:
+                raise ConfigurationError('Daemon jobs are intended to run forever')
 
     logger = FormattedLogger(
         name=f'dipdup.hooks.{hook_config.callback}',