aboutsummaryrefslogtreecommitdiffstats
path: root/slack/task.py
diff options
context:
space:
mode:
authorTrygve Aaberge <trygveaa@gmail.com>2023-01-28 16:38:25 +0100
committerTrygve Aaberge <trygveaa@gmail.com>2024-02-18 11:32:53 +0100
commitc449329680b971d049d53f67dd25dd10a6be184f (patch)
treef571f9e00a970d5028815403eae25209df784715 /slack/task.py
parent9ba4a562541ff031e95b695d4939c83d165d6f70 (diff)
downloadwee-slack-c449329680b971d049d53f67dd25dd10a6be184f.tar.gz
Only print/raise errors for finished tasks
Diffstat (limited to 'slack/task.py')
-rw-r--r--slack/task.py33
1 files changed, 21 insertions, 12 deletions
diff --git a/slack/task.py b/slack/task.py
index 149b180..0e73a16 100644
--- a/slack/task.py
+++ b/slack/task.py
@@ -1,5 +1,6 @@
from __future__ import annotations
+import traceback
from typing import (
Any,
Awaitable,
@@ -95,19 +96,27 @@ def task_runner(task: Task[Any], response: Any):
future = task.coroutine.send(response)
except BaseException as e:
result = e.value if isinstance(e, StopIteration) else e
+ in_active_tasks = task.id in shared.active_tasks
process_ended_task(task, result)
- if isinstance(e, HttpError):
- print_error(
- f"Error calling URL {e.url}: return code: {e.return_code}, "
- f"http status code: {e.http_status_code}, error: {e.error}"
- )
- elif isinstance(e, SlackApiError):
- print_error(
- f"Error from Slack API method {e.method} for workspace "
- f"{e.workspace.name}: {e.response}"
- )
- elif not isinstance(e, StopIteration):
- raise e
+
+ if isinstance(result, BaseException):
+ create_task_in_stack = "create_task" in [
+ stack.name for stack in traceback.extract_stack()
+ ]
+ if not in_active_tasks and not create_task_in_stack:
+ if isinstance(e, HttpError):
+ print_error(
+ f"Error calling URL {e.url}: return code: {e.return_code}, "
+ f"http status code: {e.http_status_code}, error: {e.error}, task: {task}"
+ )
+ elif isinstance(e, SlackApiError):
+ print_error(
+ f"Error from Slack API method {e.method} with params {e.params} for workspace "
+ f"{e.workspace.name}: {e.response}, task: {task}"
+ )
+ else:
+ raise e
+
return
if future.finished: