zrb 0.17.2__py3-none-any.whl → 0.21.0__py3-none-any.whl

zrb/__init__.py

@@ -45,6 +45,7 @@ from zrb.task_input.bool_input import BoolInput
 from zrb.task_input.choice_input import ChoiceInput
 from zrb.task_input.float_input import FloatInput
 from zrb.task_input.int_input import IntInput
+from zrb.task_input.multiline_input import MultilineInput
 from zrb.task_input.password_input import PasswordInput
 from zrb.task_input.str_input import StrInput
 from zrb.task_input.task_input import Input
@@ -93,6 +94,7 @@ assert BoolInput
 assert ChoiceInput
 assert FloatInput
 assert IntInput
+assert MultilineInput
 assert PasswordInput
 assert StrInput
 assert Env

zrb/builtin/project/add/app/generator/generator.py

@@ -26,6 +26,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
 @python_task(
     name="validate",
     inputs=[project_dir_input, package_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/app/generator/template/src/kebab-zrb-package-name/src/snake_zrb_package_name/snake_zrb_generator_name/_input.py

@@ -1,5 +1,6 @@
 import os
 
+from zrb.helper.util import coalesce, to_kebab_case, to_snake_case
 from zrb.task_input.int_input import IntInput
 from zrb.task_input.str_input import StrInput
 
@@ -41,9 +42,9 @@ app_image_name_input = StrInput(
     name="app-image-name",
     description="App image name",
     prompt="App image name",
-    default=app_image_default_namespace
-    + "/"
-    + "{{util.to_kebab_case(input.app_name)}}",  # noqa
+    default=lambda m: "/".join(
+        [app_image_default_namespace, to_kebab_case(m.get("app_name"))]
+    ),
 )
 
 http_port_input = IntInput(
@@ -58,5 +59,7 @@ env_prefix_input = StrInput(
     name="env-prefix",
     description="OS environment prefix",
     prompt="OS environment prefix",
-    default='{{util.to_snake_case(util.coalesce(input.app_name, input.task_name, "MY")).upper()}}',  # noqa
+    default=lambda m: to_snake_case(
+        coalesce(m.get("app_name"), m.get("task_name"), "MY")
+    ).upper(),
 )

zrb/builtin/project/add/app/generator/template/src/kebab-zrb-package-name/src/snake_zrb_package_name/snake_zrb_generator_name/snake_zrb_generator_name.py

@@ -32,6 +32,7 @@ _snake_app_name_tpl = "{{util.to_snake_case(input.app_name)}}"
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/app/python/python.py

@@ -31,6 +31,7 @@ _snake_app_name_tpl = "{{util.to_snake_case(input.app_name)}}"
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/fastapp/app/_input.py

@@ -1,5 +1,6 @@
 import os
 
+from zrb.helper.util import coalesce, to_kebab_case, to_snake_case
 from zrb.task_input.int_input import IntInput
 from zrb.task_input.str_input import StrInput
 
@@ -37,13 +38,14 @@ app_description_input = StrInput(
 app_image_default_namespace = os.getenv(
     "PROJECT_IMAGE_DEFAULT_NAMESPACE", "docker.io/" + SYSTEM_USER
 )
+
 app_image_name_input = StrInput(
     name="app-image-name",
     description="App image name",
     prompt="App image name",
-    default=app_image_default_namespace
-    + "/"
-    + "{{util.to_kebab_case(input.app_name)}}",  # noqa
+    default=lambda m: "/".join(
+        [app_image_default_namespace, to_kebab_case(m.get("app_name"))]
+    ),
 )
 
 http_port_input = IntInput(
@@ -58,5 +60,7 @@ env_prefix_input = StrInput(
     name="env-prefix",
     description="OS environment prefix",
     prompt="OS environment prefix",
-    default='{{util.to_snake_case(util.coalesce(input.app_name, input.task_name, "MY")).upper()}}',  # noqa
+    default=lambda m: to_snake_case(
+        coalesce(m.get("app_name"), m.get("task_name"), "MY")
+    ).upper(),
 )

zrb/builtin/project/add/fastapp/app/app.py

@@ -35,6 +35,7 @@ _snake_app_name_tpl = "{{util.to_snake_case(input.app_name)}}"
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/fastapp/app/template/src/kebab-zrb-app-name/src/component/rpc/messagebus/caller.py

@@ -15,7 +15,7 @@ class MessagebusCaller(Caller):
         publisher: Publisher,
         consumer_factory: Callable[[], Consumer],
         timeout: float = 30,
-        check_reply_interval: float = 0.1,
+        check_reply_interval: float = 0.05,
     ):
         self.logger = logger
         self.admin = admin

zrb/builtin/project/add/fastapp/crud/_input.py

@@ -10,9 +10,10 @@ entity_name_input = StrInput(
 
 plural_entity_name_input = StrInput(
     name="plural-entity-name",
+    shortcut="p",
     description="Plural entity name",
     prompt="Plural entity name",
-    default="books",
+    default=lambda m: m.get("entity_name") + "s",
 )
 
 main_column_name_input = StrInput(

zrb/builtin/project/add/fastapp/crud/crud.py

@@ -35,6 +35,7 @@ _CODEMOD_DIR = os.path.join(_CURRENT_DIR, "nodejs", "codemod")
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input, module_name_input, entity_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/fastapp/crud/nodejs/codemod/package-lock.json

@@ -60,10 +60,13 @@
       }
     },
     "node_modules/@types/node": {
-      "version": "20.6.0",
-      "resolved": "https://registry.npmjs.org/@types/node/-/node-20.6.0.tgz",
-      "integrity": "sha512-najjVq5KN2vsH2U/xyh2opaSEz6cZMR2SetLIlxlj08nOcmPOemJmUK2o4kUzfLqfrWE0PIrNeE16XhYDd3nqg==",
-      "dev": true
+      "version": "20.14.2",
+      "resolved": "https://registry.npmjs.org/@types/node/-/node-20.14.2.tgz",
+      "integrity": "sha512-xyu6WAMVwv6AKFLB+e/7ySZVr/0zLCzOa7rSpq6jNwpqOrUbcACDWC+53d4n2QHOnDou0fbIsg8wZu/sxrnI4Q==",
+      "dev": true,
+      "dependencies": {
+        "undici-types": "~5.26.4"
+      }
     },
     "node_modules/balanced-match": {
       "version": "1.0.2",
@@ -79,11 +82,11 @@
       }
     },
     "node_modules/braces": {
-      "version": "3.0.2",
-      "resolved": "https://registry.npmjs.org/braces/-/braces-3.0.2.tgz",
-      "integrity": "sha512-b8um+L1RzM3WDSzvhm6gIz1yfTbBt6YTlcEKAvsmqCZZFw46z626lVj9j1yEPW33H5H+lBQpZMP1k8l+78Ha0A==",
+      "version": "3.0.3",
+      "resolved": "https://registry.npmjs.org/braces/-/braces-3.0.3.tgz",
+      "integrity": "sha512-yQbXgO/OSZVD2IsiLlro+7Hf6Q18EJrKSEsdoMzKePKXct3gvD8oLcOQdIzGupr5Fj+EDe8gO/lxc1BzfMpxvA==",
       "dependencies": {
-        "fill-range": "^7.0.1"
+        "fill-range": "^7.1.1"
       },
       "engines": {
         "node": ">=8"
@@ -95,9 +98,9 @@
       "integrity": "sha512-q4dMFMlXtKR3XNBHyMHt/3pwYNA69EDk00lloMOaaUMKPUXBw6lpXtbu3MMVG6/uOihGnRDOlkyqsONEUj60+w=="
     },
     "node_modules/fast-glob": {
-      "version": "3.3.1",
-      "resolved": "https://registry.npmjs.org/fast-glob/-/fast-glob-3.3.1.tgz",
-      "integrity": "sha512-kNFPyjhh5cKjrUltxs+wFx+ZkbRaxxmZ+X0ZU31SOsxCEtP9VPgtq2teZw1DebupL5GmDaNQ6yKMMVcM41iqDg==",
+      "version": "3.3.2",
+      "resolved": "https://registry.npmjs.org/fast-glob/-/fast-glob-3.3.2.tgz",
+      "integrity": "sha512-oX2ruAFQwf/Orj8m737Y5adxDQO0LAB7/S5MnxCdTNDd4p6BsyIVsv9JQsATbTSq8KHRpLwIHbVlUNatxd+1Ow==",
       "dependencies": {
         "@nodelib/fs.stat": "^2.0.2",
         "@nodelib/fs.walk": "^1.2.3",
@@ -110,17 +113,17 @@
       }
     },
     "node_modules/fastq": {
-      "version": "1.15.0",
-      "resolved": "https://registry.npmjs.org/fastq/-/fastq-1.15.0.tgz",
-      "integrity": "sha512-wBrocU2LCXXa+lWBt8RoIRD89Fi8OdABODa/kEnyeyjS5aZO5/GNvI5sEINADqP/h8M29UHTHUb53sUu5Ihqdw==",
+      "version": "1.17.1",
+      "resolved": "https://registry.npmjs.org/fastq/-/fastq-1.17.1.tgz",
+      "integrity": "sha512-sRVD3lWVIXWg6By68ZN7vho9a1pQcN/WBFaAAsDDFzlJjvoGx0P8z7V1t72grFJfJhu3YPZBuu25f7Kaw2jN1w==",
       "dependencies": {
         "reusify": "^1.0.4"
       }
     },
     "node_modules/fill-range": {
-      "version": "7.0.1",
-      "resolved": "https://registry.npmjs.org/fill-range/-/fill-range-7.0.1.tgz",
-      "integrity": "sha512-qOo9F+dMUmC2Lcb4BbVvnKJxTPjCm+RRpe4gDuGrzkL7mEVl/djYSu2OdQ2Pa302N4oqkSg9ir6jaLWJ2USVpQ==",
+      "version": "7.1.1",
+      "resolved": "https://registry.npmjs.org/fill-range/-/fill-range-7.1.1.tgz",
+      "integrity": "sha512-YsGpe3WHLK8ZYi4tWDg2Jy3ebRz2rXowDxnld4bkQB00cc/1Zw9AWnC0i9ztDJitivtQvaI9KaLyKrc+hBW0yg==",
       "dependencies": {
         "to-regex-range": "^5.0.1"
       },
@@ -175,11 +178,11 @@
       }
     },
     "node_modules/micromatch": {
-      "version": "4.0.5",
-      "resolved": "https://registry.npmjs.org/micromatch/-/micromatch-4.0.5.tgz",
-      "integrity": "sha512-DMy+ERcEW2q8Z2Po+WNXuw3c5YaUSFjAO5GsJqfEl7UjvtIuFKO6ZrKvcItdy98dwFI2N1tg3zNIdKaQT+aNdA==",
+      "version": "4.0.7",
+      "resolved": "https://registry.npmjs.org/micromatch/-/micromatch-4.0.7.tgz",
+      "integrity": "sha512-LPP/3KorzCwBxfeUuZmaR6bG2kdeHSbe0P2tY3FLRU4vYrjYz5hI4QZwV0njUx3jeuKe67YukQ1LSPZBKDqO/Q==",
       "dependencies": {
-        "braces": "^3.0.2",
+        "braces": "^3.0.3",
         "picomatch": "^2.3.1"
       },
       "engines": {
@@ -301,9 +304,9 @@
       }
     },
     "node_modules/typescript": {
-      "version": "5.2.2",
-      "resolved": "https://registry.npmjs.org/typescript/-/typescript-5.2.2.tgz",
-      "integrity": "sha512-mI4WrpHsbCIcwT9cF4FZvr80QUeKvsUsUvKDoR+X/7XHQH98xYD8YHZg7ANtz2GtZt/CBq2QJ0thkGJMHfqc1w==",
+      "version": "5.4.5",
+      "resolved": "https://registry.npmjs.org/typescript/-/typescript-5.4.5.tgz",
+      "integrity": "sha512-vcI4UpRgg81oIRUFwR0WSIHKt11nJ7SAVlYNIu+QpqeyXP+gpQJy/Z4+F0aGxSE4MqwjyXvW/TzgkLAx2AGHwQ==",
       "dev": true,
       "bin": {
         "tsc": "bin/tsc",
@@ -312,6 +315,12 @@
       "engines": {
         "node": ">=14.17"
       }
+    },
+    "node_modules/undici-types": {
+      "version": "5.26.5",
+      "resolved": "https://registry.npmjs.org/undici-types/-/undici-types-5.26.5.tgz",
+      "integrity": "sha512-JlCMO+ehdEIKqlFxk6IfVoAUVmgz7cU7zD/h9XZ0qzeosSHmUJVOzSQvvYSYWXkFXC+IfLKSIffhv0sVZup6pA==",
+      "dev": true
     }
   }
 }

zrb/builtin/project/add/fastapp/field/field.py

@@ -32,6 +32,7 @@ from zrb.task.task import Task
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input, module_name_input, entity_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/plugin/plugin.py

@@ -28,7 +28,11 @@ from zrb.task.task import Task
 _CURRENT_DIR = os.path.dirname(__file__)
 
 
-@python_task(name="validate", inputs=[project_dir_input, package_name_input])
+@python_task(
+    name="validate",
+    inputs=[project_dir_input, package_name_input],
+    retry=0,
+)
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")
     validate_existing_project_dir(project_dir)

zrb/builtin/project/add/task/cmd/add.py

@@ -23,6 +23,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
         project_dir_input,
         task_name_input,
     ],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/task/docker_compose/add.py

@@ -27,6 +27,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
         project_dir_input,
         task_name_input,
     ],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/task/python/add.py

@@ -23,6 +23,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
         project_dir_input,
         task_name_input,
     ],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/config/config.py

@@ -32,6 +32,7 @@ def _get_default_tmp_dir() -> str:
 
 tmp_dir = os.getenv("ZRB_TMP_DIR", _get_default_tmp_dir())
 default_shell = os.getenv("ZRB_SHELL", _get_current_shell())
+default_editor = os.getenv("ZRB_EDITOR", "nano")
 init_script_str = os.getenv("ZRB_INIT_SCRIPTS", "")
 init_scripts = init_script_str.split(":") if init_script_str != "" else []
 logging_level = untyped_to_logging_level(os.getenv("ZRB_LOGGING_LEVEL", "WARNING"))

zrb/helper/multiline.py

@@ -0,0 +1,13 @@
+import click
+
+from zrb.helper.typecheck import typechecked
+
+
+@typechecked
+def edit(editor: str, mark_comment: str, text: str, extension: str = "txt") -> str:
+    result = click.edit(
+        text="\n".join([mark_comment, text]), editor=editor, extension=f".{extension}"
+    )
+    if result is None:
+        result = text
+    return "\n".join(result.split(mark_comment)).strip()

zrb/task/any_task.py

@@ -25,11 +25,11 @@ TAnyTask = TypeVar("TAnyTask", bound="AnyTask")
 
 class AnyTask(ABC):
     """
-    Contract/interface for all Zrb Task.
+    Abstraction for Zrb Task.
 
-    This class acts as a template for creating new Zrb Task types.
+    This class acts as a template for creating new Task type.
 
-    To define a new Zrb Task, you should extend this class and implement all its methods.
+    To define a new Task type, you should extend this class and implement all its methods.
     The easiest way to do so is by extending `Task`
     """
 
@@ -568,6 +568,13 @@ class AnyTask(ABC):
         """
         pass
 
+    @abstractmethod
+    def _lock_checkers(self):
+        """
+        Lock checkers so that it cannot be altered anymore
+        """
+        pass
+
     @abstractmethod
     def _lock_upstreams(self):
         """

zrb/task/base_remote_cmd_task.py

@@ -1,4 +1,3 @@
-import copy
 import os
 import pathlib
 
@@ -50,8 +49,10 @@ class RemoteConfig:
         password: JinjaTemplate = "",
         ssh_key: JinjaTemplate = "",
         port: Union[int, JinjaTemplate] = 22,
+        name: Optional[str] = None,
         config_map: Optional[Mapping[str, JinjaTemplate]] = None,
     ):
+        self.name = name if name is not None else host
         self.host = host
         self.user = user
         self.password = password
@@ -91,7 +92,7 @@ class SingleBaseRemoteCmdTask(CmdTask):
         on_retry: Optional[OnRetry] = None,
         on_failed: Optional[OnFailed] = None,
         checkers: Iterable[AnyTask] = [],
-        checking_interval: Union[float, int] = 0,
+        checking_interval: Union[float, int] = 0.05,
         retry: int = 2,
         retry_interval: Union[float, int] = 1,
         max_output_line: int = 1000,
@@ -146,7 +147,7 @@ class SingleBaseRemoteCmdTask(CmdTask):
         self._remote_config = remote_config
 
     def copy(self) -> TSingleBaseRemoteCmdTask:
-        return copy.deepcopy(self)
+        return super().copy()
 
     def inject_envs(self):
         super().inject_envs()
@@ -227,6 +228,7 @@ class BaseRemoteCmdTask(BaseTask):
         post_cmd_path: CmdVal = "",
         cwd: Optional[Union[str, pathlib.Path]] = None,
         upstreams: Iterable[AnyTask] = [],
+        fallbacks: Iterable[AnyTask] = [],
         on_triggered: Optional[OnTriggered] = None,
         on_waiting: Optional[OnWaiting] = None,
         on_skipped: Optional[OnSkipped] = None,
@@ -235,7 +237,7 @@ class BaseRemoteCmdTask(BaseTask):
         on_retry: Optional[OnRetry] = None,
         on_failed: Optional[OnFailed] = None,
         checkers: Iterable[AnyTask] = [],
-        checking_interval: Union[float, int] = 0,
+        checking_interval: Union[float, int] = 0.05,
         retry: int = 2,
         retry_interval: Union[float, int] = 1,
         max_output_line: int = 1000,
@@ -247,9 +249,10 @@ class BaseRemoteCmdTask(BaseTask):
         should_show_cmd: bool = True,
         should_show_working_directory: bool = True,
     ):
-        sub_tasks = [
+        self._remote_configs = list(remote_configs)
+        self._sub_tasks = [
             SingleBaseRemoteCmdTask(
-                name=f"{name}-{remote_config.host}",
+                name=f"{name}-{remote_config.name}",
                 remote_config=remote_config,
                 inputs=inputs,
                 envs=envs,
@@ -264,6 +267,7 @@ class BaseRemoteCmdTask(BaseTask):
                 post_cmd_path=post_cmd_path,
                 cwd=cwd,
                 upstreams=upstreams,
+                fallbacks=fallbacks,
                 on_triggered=on_triggered,
                 on_waiting=on_waiting,
                 on_skipped=on_skipped,
@@ -284,7 +288,7 @@ class BaseRemoteCmdTask(BaseTask):
                 should_show_cmd=should_show_cmd,
                 should_show_working_directory=should_show_working_directory,
             )
-            for remote_config in list(remote_configs)
+            for remote_config in self._remote_configs
         ]
         BaseTask.__init__(
             self,
@@ -293,7 +297,61 @@ class BaseRemoteCmdTask(BaseTask):
             color=color,
             group=group,
             description=description,
-            upstreams=sub_tasks,
+            upstreams=self._sub_tasks,
             retry=0,
             return_upstream_result=True,
         )
+
+    def insert_input(self, *inputs: AnyInput):
+        super().insert_input(*inputs)
+        for subtask in self._sub_tasks:
+            subtask.insert_input(*inputs)
+
+    def add_input(self, *inputs: AnyInput):
+        super().add_input(*inputs)
+        for subtask in self._sub_tasks:
+            subtask.add_input(*inputs)
+
+    def insert_env(self, *envs: Env):
+        super().insert_env(*envs)
+        for subtask in self._sub_tasks:
+            subtask.insert_env(*envs)
+
+    def add_env(self, *envs: Env):
+        super().add_env(*envs)
+        for subtask in self._sub_tasks:
+            subtask.add_env(*envs)
+
+    def insert_env_file(self, *env_files: EnvFile):
+        super().insert_env_file(*env_files)
+        for subtask in self._sub_tasks:
+            subtask.insert_env_file(*env_files)
+
+    def add_env_file(self, *env_files: Env):
+        super().add_env_file(*env_files)
+        for subtask in self._sub_tasks:
+            subtask.add_env_file(*env_files)
+
+    def insert_upstream(self, *upstreams: AnyTask):
+        for subtask in self._sub_tasks:
+            subtask.insert_upstream(*upstreams)
+
+    def add_upstream(self, *upstreams: AnyTask):
+        for subtask in self._sub_tasks:
+            subtask.add_upstream(*upstreams)
+
+    def insert_fallback(self, *fallbacks: AnyTask):
+        for subtask in self._sub_tasks:
+            subtask.insert_fallbacks(*fallbacks)
+
+    def add_fallback(self, *fallbacks: AnyTask):
+        for subtask in self._sub_tasks:
+            subtask.add_fallback(*fallbacks)
+
+    def insert_checker(self, *checkers: AnyTask):
+        for subtask in self._sub_tasks:
+            subtask.insert_checkers(*checkers)
+
+    def add_checker(self, *checkers: AnyTask):
+        for subtask in self._sub_tasks:
+            subtask.add_checker(*checkers)

zrb/task/base_task/base_task.py

@@ -37,8 +37,8 @@ from zrb.task_input.any_input import AnyInput
 @typechecked
 class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
     """
-    Base class for all tasks.
-    Every task definition should be extended from this class.
+    Base class for all Tasks.
+    Every Task definition should be extended from this class.
     """
 
     __running_tasks: List[AnyTask] = []
@@ -58,7 +58,7 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
         upstreams: Iterable[AnyTask] = [],
         fallbacks: Iterable[AnyTask] = [],
         checkers: Iterable[AnyTask] = [],
-        checking_interval: Union[float, int] = 0,
+        checking_interval: Union[float, int] = 0.05,
         run: Optional[Callable[..., Any]] = None,
         on_triggered: Optional[OnTriggered] = None,
         on_waiting: Optional[OnWaiting] = None,
@@ -72,7 +72,7 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
     ):
         # init properties
         retry_interval = retry_interval if retry_interval >= 0 else 0
-        checking_interval = checking_interval if checking_interval > 0 else 0.05
+        checking_interval = checking_interval if checking_interval > 0 else 0
         retry = retry if retry >= 0 else 0
         # init parent classes
         FinishTracker.__init__(self, checking_interval=checking_interval)
@@ -270,18 +270,17 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
             new_kwargs = self.get_input_map()
             # make sure args and kwargs['_args'] are the same
             self.log_info("Set run args")
-            new_args = copy.deepcopy(args)
+            new_args = list(args)
             if len(args) == 0 and "_args" in kwargs:
                 new_args = kwargs["_args"]
-            new_kwargs["_args"] = new_args
-            # inject self as input_map['_task']
-            new_kwargs["_task"] = self
             self._set_args(new_args)
             self._set_kwargs(new_kwargs)
             # run the task
             coroutines = [
                 asyncio.create_task(self._loop_check(show_done_info=show_done_info)),
-                asyncio.create_task(self._run_all(*new_args, **new_kwargs)),
+                asyncio.create_task(
+                    self._run_all(*new_args, _args=new_args, _task=self, **new_kwargs)
+                ),
             ]
             results = await asyncio.gather(*coroutines)
             result = results[-1]
@@ -456,7 +455,7 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
         # set current task local keyval
         await self._set_local_keyval(kwargs=kwargs, env_prefix=env_prefix)
         # get new_kwargs for upstream and checkers
-        new_kwargs = copy.deepcopy(kwargs)
+        new_kwargs = {key: kwargs[key] for key in kwargs}
         new_kwargs.update(self.get_input_map())
         upstream_coroutines = []
         # set upstreams keyval
@@ -468,10 +467,9 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
                 )
             )
         # set checker keyval
+        self._lock_checkers()
         checker_coroutines = []
         for checker_task in self._get_checkers():
-            checker_task.add_input(*self._get_inputs())
-            checker_task.add_env(*self._get_envs())
             checker_coroutines.append(
                 asyncio.create_task(
                     checker_task._set_keyval(kwargs=new_kwargs, env_prefix=env_prefix)

zrb/task/base_task/component/base_task_model.py

@@ -58,7 +58,7 @@ class BaseTaskModel(CommonTaskModel, PidModel, TimeTracker):
         upstreams: Iterable[AnyTask] = [],
         fallbacks: Iterable[AnyTask] = [],
         checkers: Iterable[AnyTask] = [],
-        checking_interval: Union[int, float] = 0,
+        checking_interval: Union[int, float] = 0.05,
         run: Optional[Callable[..., Any]] = None,
         on_triggered: Optional[OnTriggered] = None,
         on_waiting: Optional[OnWaiting] = None,