detect and handle task cancellations semi-properly (again)

2021-08-13 02:28:41 -07:00 · 2021-08-13 02:28:41 -07:00 · 12d0b4b738
parent 3ca869ccad
commit 12d0b4b738
1 changed files with 3 additions and 1 deletions
--- a/respodns/dns.py
+++ b/respodns/dns.py
@ -173,8 +173,10 @@ async def try_ip(db, server_ip, checks, context, opts: Options, callback=None):
        task = create_task(process(check))
        tasks.append(task)
    for task in tasks:
-        if not task.cancelled():
+        try:
            await task
+        except CancelledError:
+            pass

    if not opts.dry:
        for entry in entries: