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

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/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__(

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:

zrb/task/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.base_task.base_task import BaseTask
 
+logger.debug(colored("Loading zrb.task.task", attrs=["dark"]))
+
 
 @typechecked
 class Task(BaseTask):

zrb/task/time_watcher.py

@@ -2,6 +2,8 @@ import datetime
 
 import croniter
 
+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,
@@ -28,6 +30,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.time_watcher", attrs=["dark"]))
+
 TTimeWatcher = TypeVar("TTimeWatcher", bound="TimeWatcher")
 
 

zrb/task/wiki_task.py

@@ -1,6 +1,8 @@
 import os
 
 from zrb.action.runner import Runner
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.task import show_lines
 from zrb.helper.typecheck import typechecked
 from zrb.helper.typing import Any, Callable, Iterable, List, Mapping, 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.wiki_task", attrs=["dark"]))
+
 
 @typechecked
 def create_wiki_tasks(

zrb/task_env/constant.py

@@ -1 +1,6 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
+
+logger.debug(colored("Loading zrb.task_env.constant", attrs=["dark"]))
+
 RESERVED_ENV_NAMES = ("_ZRB_EXECUTION_ID", "_ZRB_EXECUTION_ID")

zrb/task_env/env.py

@@ -1,10 +1,14 @@
 import os
 
+from zrb.helper.accessories.color import colored
+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 JinjaTemplate, Optional
 from zrb.task_env.constant import RESERVED_ENV_NAMES
 
+logger.debug(colored("Loading zrb.task_env.env", attrs=["dark"]))
+
 # flake8: noqa E501
 
 

zrb/task_env/env_file.py

@@ -1,11 +1,15 @@
 from dotenv import dotenv_values
 
+from zrb.helper.accessories.color import colored
+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 List, Optional
 from zrb.task_env.constant import RESERVED_ENV_NAMES
 from zrb.task_env.env import Env
 
+logger.debug(colored("Loading zrb.task_env.env_file", attrs=["dark"]))
+
 # flake8: noqa E501
 
 

zrb/task_group/group.py

@@ -1,9 +1,13 @@
+from zrb.helper.accessories.color import colored
+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
 from zrb.helper.typing import List, Optional, TypeVar
 from zrb.task.any_task import AnyTask
 
+logger.debug(colored("Loading zrb.task_group.group", attrs=["dark"]))
+
 # flake8: noqa E501
 TGroup = TypeVar("TGroup", bound="Group")
 

zrb/task_input/any_input.py

@@ -1,7 +1,11 @@
 from abc import ABC, abstractmethod
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
 from zrb.helper.typing import Any, List, Mapping
 
+logger.debug(colored("Loading zrb.task_input.any_input", attrs=["dark"]))
+
 # flake8: noqa E501
 
 

zrb/task_input/base_input.py

@@ -1,9 +1,13 @@
 from zrb.config.config import show_prompt
+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, List, Mapping, Optional, Union
 from zrb.task_input.any_input import AnyInput
 from zrb.task_input.constant import RESERVED_INPUT_NAMES
 
+logger.debug(colored("Loading zrb.task_input.base_input", attrs=["dark"]))
+
 # flake8: noqa E501
 
 

zrb/task_input/bool_input.py

@@ -1,7 +1,11 @@
+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, Optional, Union
 from zrb.task_input.base_input import BaseInput
 
+logger.debug(colored("Loading zrb.task_input.bool_input", attrs=["dark"]))
+
 # flake8: noqa E501
 
 

zrb/task_input/choice_input.py

@@ -1,9 +1,13 @@
 import click
 
+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, Iterable, Optional, Union
 from zrb.task_input.base_input import BaseInput
 
+logger.debug(colored("Loading zrb.task_input.choice_input", attrs=["dark"]))
+
 # flake8: noqa E501
 
 

zrb/task_input/constant.py

@@ -1 +1,6 @@
+from zrb.helper.accessories.color import colored
+from zrb.helper.log import logger
+
+logger.debug(colored("Loading zrb.task_input.constant", attrs=["dark"]))
+
 RESERVED_INPUT_NAMES = ("_task", "_args")

zrb/task_input/float_input.py

@@ -1,7 +1,11 @@
+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, Optional, Union
 from zrb.task_input.base_input import BaseInput
 
+logger.debug(colored("Loading zrb.task_input.float_input", attrs=["dark"]))
+
 # flake8: noqa E501