hook: allow periodic loop hook to return awaitable coroutine (in simi…

…lar style as call)
pazz · May 6, 2020 · b11aa98 · b11aa98
1 parent a1a2436
commit b11aa98
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/alot/ui.py b/alot/ui.py
@@ -28,7 +28,9 @@
 async def periodic(callable_, period, *args, **kwargs):
     while True:
         try:
-            callable_(*args, **kwargs)
+            t = callable_(*args, **kwargs)
+            if asyncio.iscoroutine(t):
+                await t
         except Exception as e:
             logging.error('error in loop hook %s', str(e))
         await asyncio.sleep(period)