fix(debounce): only await coroutines

fixes Kitware#130
PaulHax · Oct 10, 2024 · a875af3 · a875af3
1 parent 1c9359e
commit a875af3
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/nrtk_explorer/library/debounce.py b/src/nrtk_explorer/library/debounce.py
@@ -1,5 +1,6 @@
 import asyncio
 from functools import wraps
+from trame.app import asynchronous
 
 
 def debounce(wait, state=None):
@@ -19,13 +20,16 @@ async def debounced():
                     await asyncio.sleep(wait)
                     if state:
                         with state:
-                            await func(*args, **kwargs)
+                            result = func(*args, **kwargs)
                     else:
-                        await func(*args, **kwargs)
+                        result = func(*args, **kwargs)
+
+                    if asyncio.iscoroutine(result):
+                        await result
                 except asyncio.CancelledError:
                     pass
 
-            task = asyncio.create_task(debounced())
+            task = asynchronous.create_task(debounced())
 
         return wrapper