zrb 0.6.0__py3-none-any.whl → 0.6.1__py3-none-any.whl

zrb/builtin/watch_changes.py

@@ -1,42 +1,41 @@
-from zrb.task.recurring_task import RecurringTask
+from zrb.builtin.helper.reccuring_action import create_recurring_action
+from zrb.runner import runner
 from zrb.task.path_watcher import PathWatcher
+from zrb.task.recurring_task import RecurringTask
 from zrb.task_input.str_input import StrInput
-from zrb.runner import runner
-from zrb.builtin.helper.reccuring_action import create_recurring_action
-
 
 watch_changes = RecurringTask(
-    name='watch-changes',
-    icon='🕵️',
-    color='yellow',
-    description='Watch changes and show message/run command',
+    name="watch-changes",
+    icon="🕵️",
+    color="yellow",
+    description="Watch changes and show message/run command",
     inputs=[
         StrInput(
-            name='pattern',
-            default='*.*',
-            prompt='File pattern',
-            description='File pattern to be watched'
+            name="pattern",
+            default="*.*",
+            prompt="File pattern",
+            description="File pattern to be watched",
         ),
         StrInput(
-            name='ignored-pattern',
-            default='',
-            prompt='Ignored file pattern',
-            description='Ignored file pattern'
+            name="ignored-pattern",
+            default="",
+            prompt="Ignored file pattern",
+            description="Ignored file pattern",
         ),
     ],
     triggers=[
         PathWatcher(
-            name='watch-path',
-            color='cyan',
-            icon='👀',
-            path='{{input.pattern}}',
-            ignored_path='{{input.ignored_pattern}}'
+            name="watch-path",
+            color="cyan",
+            icon="👀",
+            path="{{input.pattern}}",
+            ignored_path="{{input.ignored_pattern}}",
         )
     ],
     task=create_recurring_action(
-        notif_title='Watch',
-        trigger_caption='File changes',
-        trigger_xcom_key='watch-path.file'
-    )
+        notif_title="Watch",
+        trigger_caption="File changes",
+        trigger_xcom_key="watch-path.file",
+    ),
 )
 runner.register(watch_changes)

zrb/config/config.py

@@ -1,9 +1,9 @@
+import importlib.metadata as metadata
+import os
+
 from zrb.helper.string.conversion import to_boolean, to_logging_level
 from zrb.helper.typecheck import typechecked
 
-import os
-import importlib.metadata as metadata
-
 
 @typechecked
 def get_version() -> str:
@@ -11,25 +11,26 @@ def get_version() -> str:
         return metadata.version("zrb")
     except metadata.PackageNotFoundError:
         import flit
+
         meta = flit.read_module_metadata("zrb")
         return str(meta["module"]["version"])
 
 
 @typechecked
 def get_current_shell() -> str:
-    current_shell = os.getenv('SHELL', '')
-    if current_shell.endswith('zsh'):
-        return 'zsh'
-    return 'bash'
-
-
-default_shell = os.getenv('ZRB_SHELL', get_current_shell())
-init_script_str = os.getenv('ZRB_INIT_SCRIPTS', '')
-init_scripts = init_script_str.split(':') if init_script_str != '' else []
-logging_level = to_logging_level(os.getenv('ZRB_LOGGING_LEVEL', 'WARNING'))
-should_load_builtin = to_boolean(os.getenv('ZRB_SHOULD_LOAD_BUILTIN', '1'))
-env_prefix = os.getenv('ZRB_ENV', '')
-show_advertisement = to_boolean(os.getenv('ZRB_SHOW_ADVERTISEMENT', '1'))
-show_prompt = to_boolean(os.getenv('ZRB_SHOW_PROMPT', '1'))
-show_time = to_boolean(os.getenv('ZRB_SHOW_TIME', '1'))
+    current_shell = os.getenv("SHELL", "")
+    if current_shell.endswith("zsh"):
+        return "zsh"
+    return "bash"
+
+
+default_shell = os.getenv("ZRB_SHELL", get_current_shell())
+init_script_str = os.getenv("ZRB_INIT_SCRIPTS", "")
+init_scripts = init_script_str.split(":") if init_script_str != "" else []
+logging_level = to_logging_level(os.getenv("ZRB_LOGGING_LEVEL", "WARNING"))
+should_load_builtin = to_boolean(os.getenv("ZRB_SHOULD_LOAD_BUILTIN", "1"))
+env_prefix = os.getenv("ZRB_ENV", "")
+show_advertisement = to_boolean(os.getenv("ZRB_SHOW_ADVERTISEMENT", "1"))
+show_prompt = to_boolean(os.getenv("ZRB_SHOW_PROMPT", "1"))
+show_time = to_boolean(os.getenv("ZRB_SHOW_TIME", "1"))
 version = get_version()

zrb/helper/accessories/color.py

@@ -1,14 +1,26 @@
-from zrb.helper.typing import Iterable, Optional
-from zrb.helper.typecheck import typechecked
-from termcolor import colored as term_colored, COLORS
 import random
 
+from termcolor import COLORS
+from termcolor import colored as term_colored
+
+from zrb.helper.typecheck import typechecked
+from zrb.helper.typing import Iterable, Optional
+
 
 @typechecked
 def get_random_color() -> str:
     colors = [
-        'green', 'yellow', 'blue', 'magenta', 'cyan', 'white', 'light_green',
-        'light_yellow', 'light_blue', 'light_magenta', 'light_cyan',
+        "green",
+        "yellow",
+        "blue",
+        "magenta",
+        "cyan",
+        "white",
+        "light_green",
+        "light_yellow",
+        "light_blue",
+        "light_magenta",
+        "light_cyan",
     ]
     return random.choice(colors)
 
@@ -23,6 +35,6 @@ def colored(
     text: str,
     color: Optional[str] = None,
     on_color: Optional[str] = None,
-    attrs: Optional[Iterable[str]] = None
+    attrs: Optional[Iterable[str]] = None,
 ) -> str:
     return term_colored(text, color, on_color, attrs)

zrb/helper/accessories/icon.py

@@ -1,11 +1,32 @@
-from zrb.helper.typecheck import typechecked
 import random
 
+from zrb.helper.typecheck import typechecked
+
 
 @typechecked
 def get_random_icon() -> str:
     icons = [
-        '🐶', '🐱', '🐭', '🐹', '🦊', '🐻', '🐨', '🐯', '🦁', '🐮', '🐷', '🍎', '🍐',
-        '🍊', '🍋', '🍌', '🍉', '🍇', '🍓', '🍈', '🍒', '🍑'
+        "🐶",
+        "🐱",
+        "🐭",
+        "🐹",
+        "🦊",
+        "🐻",
+        "🐨",
+        "🐯",
+        "🦁",
+        "🐮",
+        "🐷",
+        "🍎",
+        "🍐",
+        "🍊",
+        "🍋",
+        "🍌",
+        "🍉",
+        "🍇",
+        "🍓",
+        "🍈",
+        "🍒",
+        "🍑",
     ]
     return random.choice(icons)

zrb/helper/accessories/name.py

@@ -1,45 +1,137 @@
-from zrb.helper.typecheck import typechecked
 import random
 import string
 
+from zrb.helper.typecheck import typechecked
+
 
 @typechecked
 def get_random_name(
-    separator: str = '-',
-    add_random_digit: bool = True,
-    digit_count: int = 4
+    separator: str = "-", add_random_digit: bool = True, digit_count: int = 4
 ) -> str:
     prefixes = [
-        'albedo', 'argent', 'argentum', 'aurora', 'aurum', 'azure',
-        'basilisk', 'cerulean', 'chimeric', 'citrin', 'coral', 'crimson',
-        'diamond', 'draco', 'dragon', 'emerald', 'ethereal', 'ferrum',
-        'flammeus', 'garnet', 'glacial', 'glimmering', 'glistening', 'golden',
-        'helios', 'igneous', 'imperial', 'jade', 'luminous', 'luna', 'lunar',
-        'mystic', 'nephrite', 'nocturnal', 'obsidian', 'opal', 'pearl',
-        'platinum', 'prismatic', 'ruby', 'sapphire', 'serpentine', 'silver',
-        'sol', 'solar', 'spiritual', 'stellar',  'tempest', 'topaz',
-        'turquoise', 'verde', 'vermillion', 'vitreous', 'zephyr', 'zircon'
+        "albedo",
+        "argent",
+        "argentum",
+        "aurora",
+        "aurum",
+        "azure",
+        "basilisk",
+        "cerulean",
+        "chimeric",
+        "citrin",
+        "coral",
+        "crimson",
+        "diamond",
+        "draco",
+        "dragon",
+        "emerald",
+        "ethereal",
+        "ferrum",
+        "flammeus",
+        "garnet",
+        "glacial",
+        "glimmering",
+        "glistening",
+        "golden",
+        "helios",
+        "igneous",
+        "imperial",
+        "jade",
+        "luminous",
+        "luna",
+        "lunar",
+        "mystic",
+        "nephrite",
+        "nocturnal",
+        "obsidian",
+        "opal",
+        "pearl",
+        "platinum",
+        "prismatic",
+        "ruby",
+        "sapphire",
+        "serpentine",
+        "silver",
+        "sol",
+        "solar",
+        "spiritual",
+        "stellar",
+        "tempest",
+        "topaz",
+        "turquoise",
+        "verde",
+        "vermillion",
+        "vitreous",
+        "zephyr",
+        "zircon",
     ]
     suffixes = [
-        'aether', 'albedo', 'alchemy', 'arcana', 'aureum', 'aetheris',
-        'anima', 'astralis', 'caelestis', 'chrysopoeia', 'cosmicum',
-        'crystallum', 'deum', 'divinitas', 'draconis', 'elementorum', 'elixir',
-        'essentia', 'eternis', 'ethereus', 'fatum', 'flamma', 'fulgur',
-        'hermetica', 'ignis', 'illuminationis', 'imperium', 'incantatum',
-        'infinitum', 'lapis', 'lux', 'magicae', 'magnum', 'materia',
-        'metallum', 'mysticum', 'natura', 'occultum', 'omnipotentis',
-        'opulentia', 'philosophia', 'philosophorum', 'praeparatum',
-        'praestantissimum', 'prima', 'primordium', 'quintessentia',
-        'regeneratio', 'ritualis', 'sanctum', 'spiritus', 'tenebris',
-        'terra', 'tinctura', 'transmutationis', 'universalis', 'vapores',
-        'venenum', 'veritas', 'vitae', 'volatus'
+        "aether",
+        "albedo",
+        "alchemy",
+        "arcana",
+        "aureum",
+        "aetheris",
+        "anima",
+        "astralis",
+        "caelestis",
+        "chrysopoeia",
+        "cosmicum",
+        "crystallum",
+        "deum",
+        "divinitas",
+        "draconis",
+        "elementorum",
+        "elixir",
+        "essentia",
+        "eternis",
+        "ethereus",
+        "fatum",
+        "flamma",
+        "fulgur",
+        "hermetica",
+        "ignis",
+        "illuminationis",
+        "imperium",
+        "incantatum",
+        "infinitum",
+        "lapis",
+        "lux",
+        "magicae",
+        "magnum",
+        "materia",
+        "metallum",
+        "mysticum",
+        "natura",
+        "occultum",
+        "omnipotentis",
+        "opulentia",
+        "philosophia",
+        "philosophorum",
+        "praeparatum",
+        "praestantissimum",
+        "prima",
+        "primordium",
+        "quintessentia",
+        "regeneratio",
+        "ritualis",
+        "sanctum",
+        "spiritus",
+        "tenebris",
+        "terra",
+        "tinctura",
+        "transmutationis",
+        "universalis",
+        "vapores",
+        "venenum",
+        "veritas",
+        "vitae",
+        "volatus",
     ]
     prefix = random.choice(prefixes)
     suffix = random.choice(suffixes)
     parts = [prefix, suffix]
     if add_random_digit:
-        random_digit = ''.join(
-            random.choices(string.digits, k=digit_count)
-        )
+        random_digit = "".join(random.choices(string.digits, k=digit_count))
         parts.append(random_digit)
     return separator.join(parts)

zrb/helper/advertisement.py

@@ -1,33 +1,29 @@
-from zrb.helper.typing import Iterable, Optional
-from zrb.helper.typecheck import typechecked
-from zrb.helper.accessories.color import colored
-
 import datetime
 import random
 import re
 import sys
 
+from zrb.helper.accessories.color import colored
+from zrb.helper.typecheck import typechecked
+from zrb.helper.typing import Iterable, Optional
+
 
 @typechecked
-class Advertisement():
-    def __init__(self, content: str, time_pattern: str = '.*'):
+class Advertisement:
+    def __init__(self, content: str, time_pattern: str = ".*"):
         self.time_pattern = time_pattern
         self.content = content
 
     def show(self):
-        print(colored(self.content, attrs=['dark']), file=sys.stderr)
+        print(colored(self.content, attrs=["dark"]), file=sys.stderr)
 
 
 @typechecked
 def get_advertisement(
-    advertisements: Iterable[Advertisement]
+    advertisements: Iterable[Advertisement],
 ) -> Optional[Advertisement]:
     now = datetime.datetime.now().isoformat()
-    candidates = [
-        adv
-        for adv in advertisements
-        if re.match(adv.time_pattern, now)
-    ]
+    candidates = [adv for adv in advertisements if re.match(adv.time_pattern, now)]
     if len(candidates) == 0:
         return None
     return random.choice(candidates)

zrb/helper/callable.py

@@ -1,10 +1,8 @@
-from typing import Any, Callable
 import inspect
+from typing import Any, Callable
 
 
-async def run_async(
-    fn: Callable, *args: Any, **kwargs: Any
-) -> Any:
+async def run_async(fn: Callable, *args: Any, **kwargs: Any) -> Any:
     if inspect.iscoroutinefunction(fn):
         return await fn(*args, **kwargs)
     return fn(*args, **kwargs)

zrb/helper/cli.py

@@ -1,19 +1,18 @@
-from zrb.helper.typecheck import typechecked
-from zrb.runner import runner
-from zrb.config.config import (
-    init_scripts, should_load_builtin, version, logging_level
-)
-from zrb.helper.loader.load_module import load_module
-from zrb.helper.log import logger
-from zrb.helper.accessories.color import colored
-from functools import lru_cache
-
-import click
 import logging
 import os
 import sys
+from functools import lru_cache
 
-HELP = f'''
+import click
+
+from zrb.config.config import init_scripts, logging_level, should_load_builtin, version
+from zrb.helper.accessories.color import colored
+from zrb.helper.loader.load_module import load_module
+from zrb.helper.log import logger
+from zrb.helper.typecheck import typechecked
+from zrb.runner import runner
+
+HELP = f"""
                 bb
    zzzzz rr rr  bb
      zz  rrr  r bbbbbb
@@ -26,7 +25,7 @@ Super framework for your super app.
 ☕ Donate at: https://stalchmst.com/donation
 🐙 Submit issues/PR at: https://github.com/state-alchemists/zrb
 🐤 Follow us at: https://twitter.com/zarubastalchmst
-'''
+"""
 
 
 class MultilineHelpClickGroup(click.Group):
@@ -38,27 +37,26 @@ class MultilineHelpClickGroup(click.Group):
 @typechecked
 def create_cli() -> click.Group:
     if logging_level <= logging.INFO:
-        logger.info(colored('Prepare CLI', attrs=['dark']))
-    zrb_cli_group = MultilineHelpClickGroup(name='zrb', help=HELP)
+        logger.info(colored("Prepare CLI", attrs=["dark"]))
+    zrb_cli_group = MultilineHelpClickGroup(name="zrb", help=HELP)
     # load from ZRB_INIT_SCRIPTS environment
     for init_script in init_scripts:
         if logging_level <= logging.INFO:
-            logger.info(colored(
-                f'Load modules from {init_script}', attrs=['dark']
-            ))
+            logger.info(colored(f"Load modules from {init_script}", attrs=["dark"]))
         load_module(script_path=init_script)
     # Load default tasks
     if should_load_builtin:
         if logging_level <= logging.INFO:
-            logger.info(colored('Load builtins', attrs=['dark']))
+            logger.info(colored("Load builtins", attrs=["dark"]))
         from zrb import builtin
+
         assert builtin
     # Load zrb_init.py
-    project_dir = os.getenv('ZRB_PROJECT_DIR', os.getcwd())
+    project_dir = os.getenv("ZRB_PROJECT_DIR", os.getcwd())
     _load_zrb_init(project_dir)
     # Serve all tasks registered to runner
     if logging_level <= logging.INFO:
-        logger.info(colored('Serve CLI', attrs=['dark']))
+        logger.info(colored("Serve CLI", attrs=["dark"]))
     cli = runner.serve(zrb_cli_group)
     return cli
 
@@ -66,26 +64,24 @@ def create_cli() -> click.Group:
 @lru_cache
 @typechecked
 def _load_zrb_init(project_dir: str):
-    project_script = os.path.join(project_dir, 'zrb_init.py')
+    project_script = os.path.join(project_dir, "zrb_init.py")
     if not os.path.isfile(project_script):
         return
     sys.path.append(project_dir)
     if logging_level <= logging.INFO:
-        logger.info(colored(f'Set sys.path to {sys.path}', attrs=['dark']))
+        logger.info(colored(f"Set sys.path to {sys.path}", attrs=["dark"]))
     python_path = _get_new_python_path(project_dir)
     if logging_level <= logging.INFO:
-        logger.info(
-            colored(f'Set PYTHONPATH to {python_path}', attrs=['dark'])
-        )
-    os.environ['PYTHONPATH'] = python_path
+        logger.info(colored(f"Set PYTHONPATH to {python_path}", attrs=["dark"]))
+    os.environ["PYTHONPATH"] = python_path
     if logging_level <= logging.INFO:
-        logger.info(colored(f'Load modules from {project_script}', attrs=['dark']))
+        logger.info(colored(f"Load modules from {project_script}", attrs=["dark"]))
     load_module(script_path=project_script)
 
 
 @typechecked
 def _get_new_python_path(project_dir: str) -> str:
-    current_python_path = os.getenv('PYTHONPATH')
-    if current_python_path is None or current_python_path == '':
+    current_python_path = os.getenv("PYTHONPATH")
+    if current_python_path is None or current_python_path == "":
         return project_dir
-    return ':'.join([current_python_path, project_dir])
+    return ":".join([current_python_path, project_dir])

zrb/helper/codemod/add_argument_to_function.py

@@ -1,12 +1,11 @@
-from zrb.helper.typecheck import typechecked
 import libcst as cst
 
+from zrb.helper.typecheck import typechecked
+
 
 @typechecked
 class AddArgumentTransformer(cst.CSTTransformer):
-    def __init__(
-        self, function_name: str, argument_name: str, argument_type: str
-    ):
+    def __init__(self, function_name: str, argument_name: str, argument_type: str):
         self.function_name = function_name
         self.argument_name = argument_name
         self.argument_type = argument_type

zrb/helper/codemod/add_argument_to_function_call.py

@@ -1,6 +1,7 @@
-from zrb.helper.typecheck import typechecked
 import libcst as cst
 
+from zrb.helper.typecheck import typechecked
+
 
 @typechecked
 class AddArgumentTransformer(cst.CSTTransformer):
@@ -9,16 +10,12 @@ class AddArgumentTransformer(cst.CSTTransformer):
         self.argument_name = argument_name
         super().__init__()
 
-    def leave_Call(
-        self, original_node: cst.Call, updated_node: cst.Call
-    ) -> cst.Call:
+    def leave_Call(self, original_node: cst.Call, updated_node: cst.Call) -> cst.Call:
         if (
             isinstance(updated_node.func, cst.Name)
             and updated_node.func.value == self.function_name
         ):
-            new_args = updated_node.args + (
-                cst.Arg(cst.Name(self.argument_name)),
-            )
+            new_args = updated_node.args + (cst.Arg(cst.Name(self.argument_name)),)
             return updated_node.with_changes(args=new_args)
         return updated_node
 

zrb/helper/codemod/add_assert_resource.py

@@ -1,6 +1,7 @@
-from zrb.helper.typecheck import typechecked
 import libcst as cst
 
+from zrb.helper.typecheck import typechecked
+
 
 @typechecked
 def add_assert_resource(code: str, resource: str) -> str:
@@ -19,12 +20,11 @@ def add_assert_resource(code: str, resource: str) -> str:
     module = cst.parse_module(code)
 
     # Create the new assertion statement
-    new_assert = cst.SimpleStatementLine([
-        cst.Assert(
-            test=cst.Name(value=resource),
-            msg=None
-        ),
-    ])
+    new_assert = cst.SimpleStatementLine(
+        [
+            cst.Assert(test=cst.Name(value=resource), msg=None),
+        ]
+    )
 
     module.body.append(new_assert)
 

zrb/helper/codemod/add_function_call.py

@@ -1,6 +1,7 @@
-from zrb.helper.typecheck import typechecked
 import libcst as cst
 
+from zrb.helper.typecheck import typechecked
+
 
 @typechecked
 def add_function_call(code: str, function_name: str, parameters: list) -> str:
@@ -23,10 +24,7 @@ def add_function_call(code: str, function_name: str, parameters: list) -> str:
     new_function_call = cst.Expr(
         value=cst.Call(
             func=cst.Name(value=function_name),
-            args=[
-                cst.Arg(value=cst.parse_expression(param))
-                for param in parameters
-            ]
+            args=[cst.Arg(value=cst.parse_expression(param)) for param in parameters],
         )
     )