From 432d7ffdaf701df556a6d4e45b8e0b75bd887597 Mon Sep 17 00:00:00 2001
From: Trygve Aaberge <trygveaa@gmail.com>
Date: Sun, 15 Jan 2023 20:53:49 +0100
Subject: Fix async functions returning None and without await never finishing

---
 slack/task.py | 26 ++++++++++++++++++++------
 1 file changed, 20 insertions(+), 6 deletions(-)

(limited to 'slack/task.py')

diff --git a/slack/task.py b/slack/task.py
index 8f4e58d..9ebcad5 100644
--- a/slack/task.py
+++ b/slack/task.py
@@ -27,11 +27,25 @@ class Future(Awaitable[T]):
             self.id = str(uuid4())
         else:
             self.id = future_id
-        self.result: Optional[T] = None
+        self._finished = False
+        self._result: Optional[T] = None
 
     def __await__(self) -> Generator[Future[T], T, T]:
-        self.result = yield self
-        return self.result
+        result = yield self
+        self.set_result(result)
+        return result
+
+    @property
+    def finished(self):
+        return self._finished
+
+    @property
+    def result(self):
+        return self._result
+
+    def set_result(self, result: T):
+        self._result = result
+        self._finished = True
 
 
 class FutureProcess(Future[Tuple[str, int, str, str]]):
@@ -50,7 +64,7 @@ class Task(Future[T]):
 
 def weechat_task_cb(data: str, *args: Any) -> int:
     future = shared.active_futures.pop(data)
-    future.result = args
+    future.set_result(args)
     tasks = shared.active_tasks.pop(data)
     for task in tasks:
         task_runner(task, args)
@@ -61,14 +75,14 @@ def task_runner(task: Task[Any], response: Any):
     while True:
         try:
             future = task.coroutine.send(response)
-            if future.result is not None:
+            if future.finished:
                 response = future.result
             else:
                 shared.active_tasks[future.id].append(task)
                 shared.active_futures[future.id] = future
                 break
         except StopIteration as e:
-            task.result = e.value
+            task.set_result(e.value)
             if task.id in shared.active_tasks:
                 tasks = shared.active_tasks.pop(task.id)
                 for active_task in tasks:
-- 
cgit