zrb 0.15.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

zrb/helper/map/conversion.py

@@ -1,6 +1,10 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Mapping
 
+logger.debug(colored("Loading zrb.helper.map.conversion", attrs=["dark"]))
+
 
 @typechecked
 def to_str(

zrb/helper/python_task.py

@@ -1,10 +1,13 @@
 import sys
 
 from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.task import show_lines as task_show_lines
 from zrb.helper.typecheck import typechecked
 from zrb.task.task import Task
 
+logger.debug(colored("Loading zrb.helper.python_task", attrs=["dark"]))
+
 _DEPRECATION_WARNING = """
 DEPRECATED: zrb.helper.python_task
 Use zrb.helper.task instead

zrb/helper/render_data.py

@@ -3,6 +3,8 @@ import os
 import platform
 import time
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.util import (
     coalesce,
     coalesce_str,
@@ -14,6 +16,8 @@ from zrb.helper.util import (
     to_snake_case,
 )
 
+logger.debug(colored("Loading zrb.helper.render_data", attrs=["dark"]))
+
 DEFAULT_RENDER_DATA = {
     "datetime": datetime,
     "os": os,

zrb/helper/string/conversion.py

@@ -1,54 +1,31 @@
-import keyword
-import logging
-import re
-
-from zrb.helper.string.constant import FALSE_STRS, TRUE_STRS
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
+from zrb.helper.string.untyped_conversion import (
+    untyped_to_boolean,
+    untyped_to_cli_name,
+    untyped_to_logging_level,
+    untyped_to_variable_name,
+)
 from zrb.helper.typecheck import typechecked
 
-NON_WORD = re.compile(r"[\W]+")
-LEADING_NUM = re.compile(r"^\d+")
-LOGGING_LEVEL_MAP = {
-    "critical": logging.CRITICAL,
-    "fatal": logging.FATAL,
-    "error": logging.ERROR,
-    "warning": logging.WARNING,
-    "warn": logging.WARN,
-    "info": logging.INFO,
-    "debug": logging.DEBUG,
-    "notset": logging.NOTSET,
-}
+logger.debug(colored("Loading zrb.helper.string.conversion", attrs=["dark"]))
 
 
 @typechecked
 def to_cli_name(name: str) -> str:
-    return NON_WORD.sub("-", name).strip("-").lower()
+    return untyped_to_cli_name((name))
 
 
 @typechecked
 def to_variable_name(string: str) -> str:
-    # Replace any non-word characters with underscore
-    string = NON_WORD.sub("_", string).strip()
-    # Remove leading digits
-    string = LEADING_NUM.sub("", string)
-    # Convert to lowercase
-    string = string.lower()
-    if keyword.iskeyword(string):
-        return string + "_"
-    return string
+    return untyped_to_variable_name(string)
 
 
 @typechecked
 def to_boolean(string: str) -> bool:
-    if string.lower() in TRUE_STRS:
-        return True
-    if string.lower() in FALSE_STRS:
-        return False
-    raise Exception(f'Cannot infer boolean value from "{string}"')
+    return untyped_to_boolean(string)
 
 
 @typechecked
 def to_logging_level(logging_level_str: str) -> int:
-    lower_logging_level_str = logging_level_str.lower()
-    if lower_logging_level_str in LOGGING_LEVEL_MAP:
-        return LOGGING_LEVEL_MAP[lower_logging_level_str]
-    return logging.WARNING
+    return untyped_to_logging_level(logging_level_str)

zrb/helper/string/jinja.py

@@ -1,6 +1,10 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any
 
+logger.debug(colored("Loading zrb.helper.string.jinja", attrs=["dark"]))
+
 
 @typechecked
 def is_probably_jinja(value: Any) -> bool:

zrb/helper/string/modification.py

@@ -1,5 +1,9 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 
+logger.debug(colored("Loading zrb.helper.string.modification", attrs=["dark"]))
+
 
 @typechecked
 def double_quote(text: str) -> str:

zrb/helper/string/parse_replacement.py

@@ -1,6 +1,10 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Mapping
 
+logger.debug(colored("Loading zrb.helper.string.parse_replacment", attrs=["dark"]))
+
 
 @typechecked
 def parse_replacement(text: str, replacement: Mapping[str, str]):

zrb/helper/string/untyped_conversion.py

@@ -0,0 +1,49 @@
+import keyword
+import logging
+import re
+
+from zrb.helper.string.constant import FALSE_STRS, TRUE_STRS
+
+NON_WORD = re.compile(r"[\W]+")
+LEADING_NUM = re.compile(r"^\d+")
+LOGGING_LEVEL_MAP = {
+    "critical": logging.CRITICAL,
+    "fatal": logging.FATAL,
+    "error": logging.ERROR,
+    "warning": logging.WARNING,
+    "warn": logging.WARN,
+    "info": logging.INFO,
+    "debug": logging.DEBUG,
+    "notset": logging.NOTSET,
+}
+
+
+def untyped_to_cli_name(name: str) -> str:
+    return NON_WORD.sub("-", name).strip("-").lower()
+
+
+def untyped_to_variable_name(string: str) -> str:
+    # Replace any non-word characters with underscore
+    string = NON_WORD.sub("_", string).strip()
+    # Remove leading digits
+    string = LEADING_NUM.sub("", string)
+    # Convert to lowercase
+    string = string.lower()
+    if keyword.iskeyword(string):
+        return string + "_"
+    return string
+
+
+def untyped_to_boolean(string: str) -> bool:
+    if string.lower() in TRUE_STRS:
+        return True
+    if string.lower() in FALSE_STRS:
+        return False
+    raise Exception(f'Cannot infer boolean value from "{string}"')
+
+
+def untyped_to_logging_level(logging_level_str: str) -> int:
+    lower_logging_level_str = logging_level_str.lower()
+    if lower_logging_level_str in LOGGING_LEVEL_MAP:
+        return LOGGING_LEVEL_MAP[lower_logging_level_str]
+    return logging.WARNING

zrb/helper/task.py

@@ -1,6 +1,10 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.task.task import Task
 
+logger.debug(colored("Loading zrb.helper.task", attrs=["dark"]))
+
 
 @typechecked
 def show_lines(task: Task, *lines: str):

zrb/helper/typecheck.py

@@ -1,17 +1,16 @@
-import os
 from typing import TypeVar
 
-from beartype import beartype
-
-from zrb.helper.string.constant import FALSE_STRS
-
-enable_type_checking_str = os.getenv("ZRB_ENABLE_TYPE_CHECKING", "1").lower()
-enable_type_checking = enable_type_checking_str not in FALSE_STRS
+from zrb.config.config import enable_type_checking
+from zrb.helper.accessories.untyped_color import untyped_colored as colored
+from zrb.helper.log import logger
 
+logger.debug(colored("Loading zrb.helper.typecheck", attrs=["dark"]))
 T = TypeVar("T")
 
 
 def typechecked(anything: T) -> T:
     if enable_type_checking:
+        from beartype import beartype
+
         return beartype(anything)
     return anything

zrb/helper/typing.py

@@ -1,15 +1,35 @@
-from beartype.typing import (
-    Any,
-    Callable,
-    Iterable,
-    List,
-    Mapping,
-    Optional,
-    Tuple,
-    Type,
-    TypeVar,
-    Union,
-)
+from zrb.config.config import enable_type_checking
+from zrb.helper.accessories.untyped_color import untyped_colored as colored
+from zrb.helper.log import logger
+
+logger.debug(colored("Loading zrb.helper.typing", attrs=["dark"]))
+
+if enable_type_checking:
+    from beartype.typing import (
+        Any,
+        Callable,
+        Iterable,
+        List,
+        Mapping,
+        Optional,
+        Tuple,
+        Type,
+        TypeVar,
+        Union,
+    )
+else:
+    from typing import (
+        Any,
+        Callable,
+        Iterable,
+        List,
+        Mapping,
+        Optional,
+        Tuple,
+        Type,
+        TypeVar,
+        Union,
+    )
 
 JinjaTemplate = str
 

zrb/helper/util.py

@@ -2,10 +2,14 @@ import re
 
 import jinja2
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.string.conversion import to_boolean as conversion_to_boolean
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Optional
 
+logger.debug(colored("Loading zrb.helper.util", attrs=["dark"]))
+
 NON_ALPHA_NUM = re.compile(r"[^a-zA-Z0-9]+")
 
 

zrb/runner.py

@@ -1,4 +1,7 @@
 from zrb.action.runner import Runner
 from zrb.config.config import env_prefix
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 
+logger.debug(colored("Loading zrb.runner", attrs=["dark"]))
 runner = Runner(env_prefix=env_prefix)

zrb/task/any_task.py

@@ -1,5 +1,7 @@
 from abc import ABC, abstractmethod
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typing import (
     Any,
     Callable,
@@ -15,6 +17,8 @@ from zrb.task_env.env import Env
 from zrb.task_env.env_file import EnvFile
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.any_task", attrs=["dark"]))
+
 # flake8: noqa E501
 TAnyTask = TypeVar("TAnyTask", bound="AnyTask")
 

zrb/task/any_task_event_handler.py

@@ -1,6 +1,10 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typing import Any, Callable
 from zrb.task.any_task import AnyTask
 
+logger.debug(colored("Loading zrb.task.any_task_event_handler", attrs=["dark"]))
+
 _TaskEventHandler = Callable[[AnyTask], Any]
 OnTriggered = _TaskEventHandler
 OnWaiting = _TaskEventHandler

zrb/task/base_remote_cmd_task.py

@@ -2,6 +2,8 @@ import copy
 import os
 import pathlib
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import (
     Any,
@@ -31,6 +33,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.base_remote_cmd_task", attrs=["dark"]))
+
 TSingleBaseRemoteCmdTask = TypeVar(
     "TSingleBaseRemoteCmdTask", bound="SingleBaseRemoteCmdTask"
 )

zrb/task/base_task/base_task.py

@@ -140,7 +140,7 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
         xcom_dir = self.__get_xcom_dir()
         xcom_file = os.path.join(xcom_dir, key)
         with open(xcom_file, "w") as f:
-            f.write(value)
+            f.write(f"{value}")
 
     def get_xcom(self, key: str) -> str:
         self.__ensure_xcom_dir_exists()

zrb/task/checker.py

@@ -1,5 +1,7 @@
 import asyncio
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Callable, Iterable, Optional, Union
 from zrb.task.any_task import AnyTask
@@ -18,6 +20,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.checker", attrs=["dark"]))
+
 
 @typechecked
 class Checker(BaseTask):

zrb/task/cmd_task.py

@@ -8,7 +8,8 @@ import sys
 import time
 
 from zrb.config.config import default_shell
-from zrb.helper.string.conversion import to_variable_name
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import (
     Any,
@@ -36,6 +37,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.cmd_task", attrs=["dark"]))
+
 _has_stty = True
 try:
     _original_stty = subprocess.getoutput("stty -g").rstrip()

zrb/task/decorator.py

@@ -1,4 +1,6 @@
 from zrb.action.runner import Runner
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Callable, Iterable, Optional, Union
 from zrb.task.any_task import AnyTask
@@ -17,6 +19,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.decorator", attrs=["dark"]))
+
 # flake8: noqa E501
 
 

zrb/task/docker_compose_task.py

@@ -2,9 +2,11 @@ import os
 import pathlib
 
 from zrb.config.config import container_backend
+from zrb.helper.accessories.color import colored
 from zrb.helper.accessories.name import get_random_name
 from zrb.helper.docker_compose.fetch_external_env import fetch_compose_file_env_map
 from zrb.helper.docker_compose.file import read_compose_file, write_compose_file
+from zrb.helper.log import logger
 from zrb.helper.string.conversion import to_cli_name
 from zrb.helper.string.modification import double_quote
 from zrb.helper.typecheck import typechecked
@@ -37,6 +39,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.docker_compose_task", attrs=["dark"]))
+
 
 def _get_ensure_zrb_network_task(backend: str):
     CURRENT_DIR = os.path.dirname(__file__)

zrb/task/flow_task.py

@@ -1,3 +1,5 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Callable, Iterable, List, Optional, TypeVar, Union
 from zrb.task.any_task import AnyTask
@@ -16,6 +18,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.flow_task", attrs=["dark"]))
+
 TFlowTask = TypeVar("TFlowTask", bound="FlowTask")
 
 

zrb/task/http_checker.py

@@ -1,5 +1,7 @@
 from http.client import HTTPConnection, HTTPSConnection
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import (
     Any,
@@ -26,6 +28,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.http_checker", attrs=["dark"]))
+
 THTTPChecker = TypeVar("THTTPChecker", bound="HTTPChecker")
 
 

zrb/task/notifier.py

@@ -1,7 +1,9 @@
 import os
 import subprocess
 
+from zrb.helper.accessories.color import colored
 from zrb.helper.accessories.icon import get_random_icon
+from zrb.helper.log import logger
 from zrb.helper.string.modification import double_quote
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Callable, Iterable, JinjaTemplate, Optional, Union
@@ -21,6 +23,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.notifier", attrs=["dark"]))
+
 _CURRENT_DIR = os.path.dirname(__file__)
 _NOTIFY_PS1_PATH = os.path.realpath(
     os.path.abspath(

zrb/task/parallel.py

@@ -1,9 +1,13 @@
 from abc import ABC, abstractmethod
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import List, TypeVar, Union
 from zrb.task.any_task import AnyTask
 
+logger.debug(colored("Loading zrb.task.parallel", attrs=["dark"]))
+
 TParallel = TypeVar("TParallel", bound="Parallel")
 
 

zrb/task/path_checker.py

@@ -1,4 +1,6 @@
+from zrb.helper.accessories.color import colored
 from zrb.helper.file.match import get_file_names
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import (
     Any,
@@ -26,6 +28,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.path_checker", attrs=["dark"]))
+
 TPathChecker = TypeVar("TPathChecker", bound="PathChecker")
 
 

zrb/task/path_watcher.py

@@ -1,6 +1,8 @@
 import os
 
+from zrb.helper.accessories.color import colored
 from zrb.helper.file.match import get_file_names
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import (
     Any,
@@ -29,6 +31,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.path_watcher", attrs=["dark"]))
+
 TPathWatcher = TypeVar("TPathWatcher", bound="PathWatcher")
 
 

zrb/task/port_checker.py

@@ -1,5 +1,7 @@
 import socket
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import (
     Any,
@@ -26,6 +28,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.port_checker", attrs=["dark"]))
+
 TPortChecker = TypeVar("TPortChecker", bound="PortChecker")
 
 

zrb/task/recurring_task.py

@@ -1,7 +1,9 @@
 import asyncio
 import copy
 
+from zrb.helper.accessories.color import colored
 from zrb.helper.accessories.name import get_random_name
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Callable, Iterable, List, Mapping, Optional, Union
 from zrb.task.any_task import AnyTask
@@ -20,6 +22,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.recurring_task", attrs=["dark"]))
+
 
 class RunConfig:
     def __init__(
@@ -111,6 +115,7 @@ class RecurringTask(BaseTask):
         self._triggers: List[AnyTask] = [trigger.copy() for trigger in triggers]
         self._run_configs: List[RunConfig] = []
         self._single_execution = single_execution
+        self.print_err("Deprecated, please use Server instead")
 
     async def _set_keyval(self, kwargs: Mapping[str, Any], env_prefix: str):
         await super()._set_keyval(kwargs=kwargs, env_prefix=env_prefix)

zrb/task/remote_cmd_task.py

@@ -1,6 +1,8 @@
 import os
 import pathlib
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Callable, Iterable, Optional, Union
 from zrb.task.any_task import AnyTask
@@ -20,6 +22,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.remote_cmd_task", attrs=["dark"]))
+
 _CURRENT_DIR = os.path.dirname(__file__)
 _SHELL_SCRIPT_DIR = os.path.join(_CURRENT_DIR, "..", "shell-scripts")
 with open(os.path.join(_SHELL_SCRIPT_DIR, "ssh-util.sh")) as file:

zrb/task/resource_maker.py

@@ -1,4 +1,6 @@
+from zrb.helper.accessories.color import colored
 from zrb.helper.file.copy_tree import copy_tree
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import (
     Any,
@@ -35,6 +37,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.resource_maker", attrs=["dark"]))
+
 Replacement = Mapping[str, JinjaTemplate]
 ReplacementMutator = Callable[[AnyTask, Replacement], Replacement]
 TResourceMaker = TypeVar("TResourceMaker", bound="ResourceMaker")

zrb/task/rsync_task.py

@@ -1,6 +1,8 @@
 import os
 import pathlib
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Callable, Iterable, JinjaTemplate, Optional, Union
 from zrb.task.any_task import AnyTask
@@ -20,6 +22,8 @@ from zrb.task_env.env_file import EnvFile
 from zrb.task_group.group import Group
 from zrb.task_input.any_input import AnyInput
 
+logger.debug(colored("Loading zrb.task.rsync_task", attrs=["dark"]))
+
 _CURRENT_DIR = os.path.dirname(__file__)
 _SHELL_SCRIPT_DIR = os.path.join(_CURRENT_DIR, "..", "shell-scripts")
 with open(os.path.join(_SHELL_SCRIPT_DIR, "rsync-util.sh")) as file: