Pass task result to task after-callback

So that the callback can for example log the result

pyzeebe/job/job.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Optional, Any
 
 from pyzeebe.job.job_status import JobStatus
 from pyzeebe.types import Headers, Variables
@@ -25,6 +25,10 @@ class Job:
     variables: Variables
     tenant_id: Optional[str] = None
     status: JobStatus = JobStatus.Running
+    task_result = None
+
+    def set_task_result(self, task_result: Any) -> None:
+        object.__setattr__(self, "task_result", task_result)
 
     def _set_status(self, value: JobStatus) -> None:
         object.__setattr__(self, "status", value)

pyzeebe/task/task_builder.py

@@ -41,6 +41,7 @@ async def job_handler(job: Job, job_controller: JobController) -> Job:
         return_variables, succeeded = await run_original_task_function(
             prepared_task_function, task_config, job, job_controller
         )
+        job.set_task_result(return_variables)
         await job_controller.set_running_after_decorators_status()
         job = await after_decorator_runner(job)
         if succeeded:

tests/unit/task/task_builder_test.py

@@ -207,6 +207,29 @@ async def test_after_decorator_called(
 
         task_config.after.pop().assert_called_once()
 
+    @pytest.mark.asyncio
+    async def test_after_decorator_can_access_task_result(
+        self,
+        task_config: TaskConfig,
+        job: Job,
+        mocked_job_controller: JobController,
+    ):
+        async def task_function():
+            return {"result": 1}
+
+        self.task_result = dict()
+
+        async def after_decorator(job: Job):
+            self.task_result = job.task_result
+            return job
+
+        task_config.after.append(after_decorator)
+        job_handler = task_builder.build_job_handler(task_function, task_config)
+
+        await job_handler(job, mocked_job_controller)
+
+        assert self.task_result == {"result": 1}
+
     @pytest.mark.asyncio
     async def test_failing_decorator_continues(
         self,