zrb 0.3.0__tar.gz → 0.3.1__tar.gz

{zrb-0.3.0 → zrb-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 0.3.0
+Version: 0.3.1
 Summary: A Framework to Enhance Your Workflow
 Author-email: Go Frendi Gunawan <gofrendiasgard@gmail.com>
 Requires-Python: >=3.10.0

{zrb-0.3.0 → zrb-0.3.1}/docs/getting-started.md

@@ -907,11 +907,37 @@ Parallel(set_xcom_cmd, set_xcom_py) >> Parallel(get_xcom_cmd, get_xcom_py) >> te
 runner.register(test_xcom)
 ```
 
-The example shows that `set-xcom-cmd` and `set-xcom-py` set XCom values `one` and `two` respectively.
+The example shows that `set-xcom-cmd` and `set-xcom-py` set XCom values `one` and `two`, respectively.
 
 On the other hand, `get-xcom-cmd` and `get-xcom-py` fetch the values and print them.
 
-Furthermore, every Zrb Task has their return values saved as `__xcom['execution-id']['task-name']`.
+Furthermore, every Zrb Task has its return values saved as `__xcom['execution-id']['task-name']`. Let's see the following example:
+
+```python
+from zrb import runner, Parallel, CmdTask, python_task
+
+hello_cmd = CmdTask(
+    name='hello-cmd',
+    cmd='echo hello-cmd',
+)
+
+@python_task(
+    name='hello-py'
+)
+def hello_py(*args, **kwargs):
+    return 'hello-py'
+
+hello = CmdTask(
+    name='hello',
+    cmd=[
+        'echo {{task.get_xcom("hello-cmd")}}',
+        'echo {{task.get_xcom("hello-py")}}',
+    ],
+)
+
+Parallel(hello_cmd, hello_py) >> hello
+runner.register(hello)
+```
 
 
 ## Basic Example

{zrb-0.3.0 → zrb-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "zrb"
-version = "0.3.0"
+version = "0.3.1"
 authors = [
   { name="Go Frendi Gunawan", email="gofrendiasgard@gmail.com" },
 ]

{zrb-0.3.0 → zrb-0.3.1}/src/zrb/task/base_task/base_task.py

@@ -255,7 +255,6 @@ class BaseTask(
             ]
             results = await asyncio.gather(*coroutines)
             result = results[-1]
-            self.set_xcom(self.get_name(), f'{result}')
             self._print_result(result)
             return result
         except Exception as e:
@@ -379,6 +378,8 @@ class BaseTask(
                 self._increase_attempt()
                 await asyncio.sleep(self._retry_interval)
                 await self.on_retry()
+        self.set_xcom(self.get_name(), f'{result}')
+        self.log_debug(f'XCom: {self.__xcom}')
         await self._mark_done()
         return result
 

{zrb-0.3.0 → zrb-0.3.1}/test/task/test_task_xcom.py

@@ -3,7 +3,7 @@ from zrb.task.cmd_task import CmdTask
 from zrb.task.task import Task
 
 
-def test_task_xcom():
+def test_set_xcom():
     set_xcom_cmd = CmdTask(
         name='set-xcom-cmd',
         cmd='echo "hi{{task.set_xcom("one", "ichi")}}"'
@@ -45,3 +45,27 @@ def test_task_xcom():
     assert result[0].output == 'ichi\nni'
     assert result[1] == 'ichi\nni'
 
+
+def test_get_return_value_as_xcom():
+    hello_cmd = CmdTask(
+        name='hello-cmd',
+        cmd='echo hello-cmd',
+    )
+
+    @python_task(
+        name='hello-py'
+    )
+    def hello_py(*args, **kwargs):
+        return 'hello-py'
+
+    hello = CmdTask(
+        name='hello',
+        upstreams=[hello_cmd, hello_py],
+        cmd=[
+            'echo {{task.get_xcom("hello-cmd")}}',
+            'echo {{task.get_xcom("hello-py")}}',
+        ],
+    )
+    fn = hello.to_function()
+    result = fn()
+    assert result.output == 'hello-cmd\nhello-py'