zrb 0.0.120__py3-none-any.whl → 0.0.121__py3-none-any.whl

zrb/__init__.py

@@ -1,6 +1,7 @@
 from zrb.runner import runner
 from zrb.task.decorator import python_task
 from zrb.task.any_task import AnyTask
+from zrb.task.parallel import AnyParallel, Parallel
 from zrb.task.task import Task
 from zrb.task.cmd_task import CmdTask
 from zrb.task.docker_compose_task import DockerComposeTask, ServiceConfig
@@ -33,6 +34,8 @@ from zrb.helper.default_env import inject_default_env
 
 assert runner
 assert AnyTask
+assert AnyParallel
+assert Parallel
 assert python_task
 assert Task
 assert CmdTask

zrb/task/base_task/base_task.py

@@ -12,6 +12,7 @@ from zrb.task.base_task.component.trackers import (
 )
 from zrb.task.base_task.component.renderer import Renderer
 from zrb.task.base_task.component.base_task_model import BaseTaskModel
+from zrb.task.parallel import AnyParallel
 from zrb.advertisement import advertisements
 from zrb.task_group.group import Group
 from zrb.task_env.env import Env
@@ -102,9 +103,15 @@ class BaseTask(
         self.__is_execution_triggered: bool = False
         self.__is_execution_started: bool = False
 
-    def __rshift__(self, other_task: AnyTask):
-        other_task.add_upstream(self)
-        return other_task
+    def __rshift__(self, operand: Union[AnyParallel, AnyTask]):
+        if isinstance(operand, AnyTask):
+            operand.add_upstream(self)
+            return operand
+        if isinstance(operand, AnyParallel):
+            other_tasks: List[AnyTask] = operand.get_tasks()
+            for other_task in other_tasks:
+                other_task.add_upstream(self)
+            return operand
 
     def copy(self) -> AnyTask:
         return copy.deepcopy(self)

zrb/task/parallel.py

@@ -0,0 +1,36 @@
+from zrb.helper.typing import TypeVar, List, Union
+from abc import ABC, abstractmethod
+from zrb.helper.typecheck import typechecked
+from zrb.task.any_task import AnyTask
+
+
+TParallel = TypeVar('TParallel', bound='Parallel')
+
+
+class AnyParallel(ABC):
+    @abstractmethod
+    def get_tasks(self) -> List[AnyTask]:
+        pass
+
+
+@typechecked
+class Parallel(AnyParallel):
+    def __init__(self, *tasks: AnyTask):
+        self.__tasks = list(tasks)
+
+    def get_tasks(self) -> List[AnyTask]:
+        return self.__tasks
+
+    def __rshift__(
+        self, operand: Union[AnyTask, AnyParallel]
+    ) -> Union[AnyTask, AnyParallel]:
+        if isinstance(operand, AnyTask):
+            for task in self.__tasks:
+                operand.add_upstream(task)
+            return operand
+        if isinstance(operand, AnyParallel):
+            other_tasks: List[AnyTask] = operand.get_tasks()
+            for task in self.__tasks:
+                for other_task in other_tasks:
+                    other_task.add_upstream(task)
+            return operand

{zrb-0.0.120.dist-info → zrb-0.0.121.dist-info}/METADATA

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

{zrb-0.0.120.dist-info → zrb-0.0.121.dist-info}/RECORD

@@ -1,4 +1,4 @@
-zrb/__init__.py,sha256=uX1pE1kUwxy-RobzktNyzjP69cnZcM206JiSk7fJnKM,2040
+zrb/__init__.py,sha256=1SSyh2yX2Zk3BnraVUTNiXDVsQOstiSC8cPR2ZukEtQ,2127
 zrb/__main__.py,sha256=CdfuYSxqlJhnsJPOBOL2_uzEaTZHC3MtpyTuz8QUfUI,314
 zrb/advertisement.py,sha256=e-1tFPlmEuz8IqaIJ_9-2p9x5cuGsxssJGu5F0wHthI,505
 zrb/runner.py,sha256=MPCNPMCyiYNZeubSxd1eM2Zr6PCIKF-9pkG385AXElw,118
@@ -1156,6 +1156,7 @@ zrb/task/docker_compose_task.py,sha256=clbq1OOKC_zAz7IL3iBpNnCGR96JCF056r4o0v0E7
 zrb/task/flow_task.py,sha256=l0xCb6mlaaNK5NVZxhKWE1qKXUI2akzC6CZfciGh7P8,3939
 zrb/task/http_checker.py,sha256=PoneoC9S7bTw41IP7hB9ewaWO-lF3ZWT75MXEPyThYA,5244
 zrb/task/notifier.py,sha256=UVhx1fLORhqyXprzqQqdoVYIFGrirwnTk-Ak18ZyucQ,5332
+zrb/task/parallel.py,sha256=OnCdh4aQwL-N2ICNf3gshhqx5-Hq_QPZD3ycMHqrFRg,1042
 zrb/task/path_checker.py,sha256=BSBf8ZXEcTwVI3UskSgdXXWrCLWggnwk6sHHbEI0z5g,3382
 zrb/task/path_watcher.py,sha256=9r36cMc7EaLehQVw0149TN-qOZitucR9mRW3S-w7SQw,4695
 zrb/task/port_checker.py,sha256=IvOUhefkpYcKZJ9zGr0RNUZju5yLvXoNJlLlODpLZZ8,4111
@@ -1166,7 +1167,7 @@ zrb/task/rsync_task.py,sha256=3vxENH-unD1VLgZI09Ts_lAQRMi2i1UiJpsaI1dltzI,3966
 zrb/task/task.py,sha256=l1jzJil7AqnsFn9e8WyO-rgCIG1JSvB6DaRBQZQo00I,246
 zrb/task/time_watcher.py,sha256=cbmmKtbeMrkRwpRZi_-jyJ_hC4KnsOPHUIpzvns2hPQ,3728
 zrb/task/base_task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/task/base_task/base_task.py,sha256=PBToO1WgljuxyQYAsoIvHOIb1Kffh5Ta1yG_FupufQ0,16216
+zrb/task/base_task/base_task.py,sha256=w593NE0-sfuV7WKjw4QxcdtT1lngnCIeXPg60ueJJcs,16540
 zrb/task/base_task/component/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/base_task/component/base_task_model.py,sha256=mi7v0ScvVlqkXSM6N0GUEWomVd3wwzL492LU60u2jZQ,9845
 zrb/task/base_task/component/common_task_model.py,sha256=9qwrnVARWafGm8PNpZtPmxlgVSD_3ZNt3hZxiOJO2m4,10478
@@ -1189,8 +1190,8 @@ zrb/task_input/int_input.py,sha256=VYduGcAE0m7eDItRQboTFDCW5whMur_uH1Y9qtBIwV4,1
 zrb/task_input/password_input.py,sha256=i6U-smP6zFo8Ts9x14YXVgx7PKkRWr4U_EzPFp9nucw,1687
 zrb/task_input/str_input.py,sha256=w9TTUMlMQtlOXh2ahHPbRnFyTiV94lXizucCc8SuWoo,1668
 zrb/task_input/task_input.py,sha256=HotqM1iYSzwE4PIj8grnEUsvJqjx1dS6Ek7i6ZJLq2Y,83
-zrb-0.0.120.dist-info/entry_points.txt,sha256=xTgXc1kBKYhJHEujdaSPHUcJT3-hbyP1mLgwkv-5sSk,40
-zrb-0.0.120.dist-info/LICENSE,sha256=WfnGCl8G60EYOPAEkuc8C9m9pdXWDe08NsKj3TBbxsM,728
-zrb-0.0.120.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-zrb-0.0.120.dist-info/METADATA,sha256=kZ2WFfdx-ZLc-oZ4wZrZIMUzzkPHqbnOsG_KuqPQKWU,15079
-zrb-0.0.120.dist-info/RECORD,,
+zrb-0.0.121.dist-info/entry_points.txt,sha256=xTgXc1kBKYhJHEujdaSPHUcJT3-hbyP1mLgwkv-5sSk,40
+zrb-0.0.121.dist-info/LICENSE,sha256=WfnGCl8G60EYOPAEkuc8C9m9pdXWDe08NsKj3TBbxsM,728
+zrb-0.0.121.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+zrb-0.0.121.dist-info/METADATA,sha256=A07FBVLViYyRae2p0Efcs_hmBmnFojeeZeFk4hv-_8M,15079
+zrb-0.0.121.dist-info/RECORD,,