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

zrb/builtin/md5.py

@@ -1,11 +1,12 @@
-from zrb.helper.typing import Any
+import hashlib
+
+import aiofiles
+
 from zrb.builtin.group import md5_group
+from zrb.helper.typing import Any
+from zrb.runner import runner
 from zrb.task.decorator import python_task
 from zrb.task_input.str_input import StrInput
-from zrb.runner import runner
-
-import aiofiles
-import hashlib
 
 ###############################################################################
 # Task Definitions
@@ -13,44 +14,30 @@ import hashlib
 
 
 @python_task(
-    name='hash',
+    name="hash",
     group=md5_group,
-    inputs=[
-        StrInput(
-            name='text',
-            shortcut='t',
-            description='Text',
-            default=''
-        )
-    ],
-    description='Hash md5',
+    inputs=[StrInput(name="text", shortcut="t", description="Text", default="")],
+    description="Hash md5",
     retry=0,
-    runner=runner
+    runner=runner,
 )
 async def hash_text(*args: str, **kwargs: Any):
-    text: str = kwargs.get('text', '')
+    text: str = kwargs.get("text", "")
     hashed_text = hashlib.md5(text.encode()).hexdigest()
     return hashed_text
 
 
 @python_task(
-    name='sum',
+    name="sum",
     group=md5_group,
-    inputs=[
-        StrInput(
-            name='file',
-            shortcut='f',
-            description='File',
-            default=''
-        )
-    ],
-    description='Sum md5 file',
+    inputs=[StrInput(name="file", shortcut="f", description="File", default="")],
+    description="Sum md5 file",
     retry=0,
-    runner=runner
+    runner=runner,
 )
 async def sum_file(*args: str, **kwargs: Any):
-    file_path: str = kwargs.get('file', '')
-    async with aiofiles.open(file_path, mode='rb') as file:
+    file_path: str = kwargs.get("file", "")
+    async with aiofiles.open(file_path, mode="rb") as file:
         contents = await file.read()
     hashed_text = hashlib.md5(contents).hexdigest()
     return hashed_text

zrb/builtin/process.py

@@ -1,45 +1,45 @@
-from zrb.task.cmd_task import CmdTask
 from zrb.builtin.group import process_group
+from zrb.runner import runner
+from zrb.task.cmd_task import CmdTask
 from zrb.task_input.int_input import IntInput
 from zrb.task_input.str_input import StrInput
-from zrb.runner import runner
 
 ###############################################################################
 # Task Definitions
 ###############################################################################
 
 get_pid_by_port = CmdTask(
-    name='get-pid-by-port',
+    name="get-pid-by-port",
     group=process_group,
-    description='Get PID by port',
+    description="Get PID by port",
     inputs=[
         IntInput(
-            name='port',
-            shortcut='p',
-            default='8080',
-            description='Port to be checked',
-            prompt='Port to be checked',
+            name="port",
+            shortcut="p",
+            default="8080",
+            description="Port to be checked",
+            prompt="Port to be checked",
         )
     ],
     cmd="lsof -i :{{ input.port }} -n -P | awk 'NR>1 {print $2}'",
-    checking_interval=3
+    checking_interval=3,
 )
 runner.register(get_pid_by_port)
 
 get_pid_by_name = CmdTask(
-    name='get-pid-by-name',
+    name="get-pid-by-name",
     group=process_group,
-    description='Get PID by name',
+    description="Get PID by name",
     inputs=[
         StrInput(
-            name='name',
-            shortcut='n',
-            default='python',
-            description='Process name to be checked',
-            prompt='Process name to be checked',
+            name="name",
+            shortcut="n",
+            default="python",
+            description="Process name to be checked",
+            prompt="Process name to be checked",
         )
     ],
-    cmd='pgrep {{ input.name }}',
-    checking_interval=3
+    cmd="pgrep {{ input.name }}",
+    checking_interval=3,
 )
 runner.register(get_pid_by_name)

zrb/builtin/project.py

@@ -1,9 +1,9 @@
-from zrb.helper.typing import Any, Mapping
 from zrb.builtin.group import project_group
+from zrb.helper.env_map.fetch import fetch_env_map_from_group
+from zrb.helper.typing import Any, Mapping
+from zrb.runner import runner
 from zrb.task.decorator import python_task
 from zrb.task_input.bool_input import BoolInput
-from zrb.runner import runner
-from zrb.helper.env_map.fetch import fetch_env_map_from_group
 
 ###############################################################################
 # Task Definitions
@@ -11,26 +11,24 @@ from zrb.helper.env_map.fetch import fetch_env_map_from_group
 
 
 @python_task(
-    name='get-default-env',
-    description='Get default values for project environments',
+    name="get-default-env",
+    description="Get default values for project environments",
     inputs=[
         BoolInput(
-            name='export',
-            shortcut='e',
-            description='Whether add export statement or not',
-            default=True
+            name="export",
+            shortcut="e",
+            description="Whether add export statement or not",
+            default=True,
         )
     ],
     group=project_group,
-    runner=runner
+    runner=runner,
 )
 async def get_default_env(*args: Any, **kwargs: Any) -> str:
     env_map: Mapping[str, str] = {}
     env_map = fetch_env_map_from_group(env_map, project_group)
     env_keys = list(env_map.keys())
     env_keys.sort()
-    should_export = kwargs.get('export', True)
-    export_prefix = 'export ' if should_export else ''
-    return '\n'.join([
-        f'{export_prefix}{key}={env_map[key]}' for key in env_keys
-    ])
+    should_export = kwargs.get("export", True)
+    export_prefix = "export " if should_export else ""
+    return "\n".join([f"{export_prefix}{key}={env_map[key]}" for key in env_keys])

zrb/builtin/say.py

@@ -1,138 +1,119 @@
-from zrb.helper.typing import Any, List
+import datetime
+import random
+
 from zrb.helper.python_task import show_lines
+from zrb.helper.typing import Any, List
+from zrb.runner import runner
 from zrb.task.decorator import python_task
-from zrb.task_input.str_input import StrInput
 from zrb.task_input.int_input import IntInput
-from zrb.runner import runner
-import datetime
-import random
+from zrb.task_input.str_input import StrInput
 
 _MIN_WIDTH = 10
 _MOTIVATIONAL_QUOTES = [
     [
-        'The best time to plant a tree was 20 years ago.',
-        'The second best time is now.',
-        '~ Chinese Proverb'
-    ],
-    [
-        'The only way to do great work is to love what you do.',
-        '~ Steve Jobs'
+        "The best time to plant a tree was 20 years ago.",
+        "The second best time is now.",
+        "~ Chinese Proverb",
     ],
+    ["The only way to do great work is to love what you do.", "~ Steve Jobs"],
+    ["Believe you can and you're halfway there.", "~ Theodore Roosevelt"],
+    ["It does not matter how slowly you go as long as you do not stop.", "~ Confucius"],
+    ["Everything you've ever wanted is on the other side of fear.", "~ George Addair"],
     [
-        'Believe you can and you\'re halfway there.',
-        '~ Theodore Roosevelt'
+        "Success is not final, failure is not fatal:",
+        "It is the courage to continue that counts.",
+        "~ Winston Churchill",
     ],
     [
-        'It does not matter how slowly you go as long as you do not stop.',
-        '~ Confucius'
+        "Hardships often prepare ordinary people",
+        "for an extraordinary destiny.",
+        "~ C.S. Lewis",
     ],
     [
-        'Everything you\'ve ever wanted is on the other side of fear.',
-        '~ George Addair'
+        "Your time is limited, don't waste it living someone else's life.",
+        "~ Steve Jobs",
     ],
+    ["Don’t watch the clock; do what it does. Keep going.", "~ Sam Levenson"],
     [
-        'Success is not final, failure is not fatal:',
-        'It is the courage to continue that counts.',
-        '~ Winston Churchill'
+        "You are never too old to set another goal or to dream a new dream.",
+        "~ C.S. Lewis",
     ],
     [
-        'Hardships often prepare ordinary people',
-        'for an extraordinary destiny.',
-        '~ C.S. Lewis'
+        "The only limit to our realization of tomorrow",
+        "will be our doubts of today.",
+        "~ Franklin D. Roosevelt",
     ],
     [
-        'Your time is limited, don\'t waste it living someone else\'s life.',
-        '~ Steve Jobs'
+        "Believe in yourself.",
+        "You are braver than you think, more talented than you know,"
+        "and capable of more than you imagine.",
+        "~ Roy T. Bennett",
     ],
     [
-        'Don’t watch the clock; do what it does. Keep going.',
-        '~ Sam Levenson'
+        "I can't change the direction of the wind,",
+        "but I can adjust my sails to always reach my destination.",
+        "~ Jimmy Dean",
     ],
+    ["You are enough just as you are.", "~ Meghan Markle"],
     [
-        'You are never too old to set another goal or to dream a new dream.',
-        '~ C.S. Lewis'
+        "The future belongs to those",
+        "who believe in the beauty of their dreams.",
+        "~ Eleanor Roosevelt",
     ],
-    [
-        'The only limit to our realization of tomorrow',
-        'will be our doubts of today.',
-        '~ Franklin D. Roosevelt'
-    ],
-    [
-        'Believe in yourself.',
-        'You are braver than you think, more talented than you know,'
-        'and capable of more than you imagine.',
-        '~ Roy T. Bennett'
-    ],
-    [
-        'I can\'t change the direction of the wind,',
-        'but I can adjust my sails to always reach my destination.',
-        '~ Jimmy Dean'
-    ],
-    [
-        'You are enough just as you are.',
-        '~ Meghan Markle'
-    ],
-    [
-        'The future belongs to those',
-        'who believe in the beauty of their dreams.',
-        '~ Eleanor Roosevelt'
-    ]
 ]
 
 
 @python_task(
-    name='say',
-    inputs=[
-        StrInput(name='text', default=''),
-        IntInput(name='width', default=80)
-    ],
-    description='Say anything, https://www.youtube.com/watch?v=MbPr1oHO4Hw',
-    runner=runner
+    name="say",
+    inputs=[StrInput(name="text", default=""), IntInput(name="width", default=80)],
+    description="Say anything, https://www.youtube.com/watch?v=MbPr1oHO4Hw",
+    runner=runner,
 )
 def say(*args: Any, **kwargs: Any):
-    width: int = kwargs.get('width', 80)
+    width: int = kwargs.get("width", 80)
     if width < _MIN_WIDTH:
         width = _MIN_WIDTH
-    text: str = kwargs.get('text', '')
-    top_border = '┌' + '─' * (width + 2) + '┐'
-    content = [
-        '| ' + line + ' |' for line in _get_content(text, width)
-    ]
-    bottom_border = '└' + '─' * (width + 2) + '┘'
-    lines = [top_border] + content + [bottom_border] + [
-        '      \\',
-        '       \\',
-        '  o    ___    o',
-        '  | ┌-------┐ |',
-        '  |(| o   o |)|',
-        '    | └---┘ |',
-        '    └-------┘',
-    ]
-    show_lines(kwargs['_task'], *lines)
+    text: str = kwargs.get("text", "")
+    top_border = "┌" + "─" * (width + 2) + "┐"
+    content = ["| " + line + " |" for line in _get_content(text, width)]
+    bottom_border = "└" + "─" * (width + 2) + "┘"
+    lines = (
+        [top_border]
+        + content
+        + [bottom_border]
+        + [
+            "      \\",
+            "       \\",
+            "  o    ___    o",
+            "  | ┌-------┐ |",
+            "  |(| o   o |)|",
+            "    | └---┘ |",
+            "    └-------┘",
+        ]
+    )
+    show_lines(kwargs["_task"], *lines)
 
 
 def _get_content(text: str, width: int) -> List[str]:
-    if text == '':
+    if text == "":
         now = datetime.datetime.now()
-        today = 'Today is ' + now.strftime('%A, %B %d, %Y')
-        current_time = 'Current time is ' + now.strftime('%I:%M %p')
+        today = "Today is " + now.strftime("%A, %B %d, %Y")
+        current_time = "Current time is " + now.strftime("%I:%M %p")
         motivational_quote = random.choice(_MOTIVATIONAL_QUOTES)
         return [
             today.ljust(width),
             current_time.ljust(width),
-            ''.ljust(width),
-        ] + [
-            line.ljust(width) for line in motivational_quote
-        ]
+            "".ljust(width),
+        ] + [line.ljust(width) for line in motivational_quote]
     return _split_text_by_width(text, width)
 
 
 def _split_text_by_width(text: str, width: int) -> List[str]:
-    original_lines = text.split('\n')
+    original_lines = text.split("\n")
     new_lines = []
     for original_line in original_lines:
         new_lines += [
-            original_line[i:i+width].ljust(width)
+            original_line[i : i + width].ljust(width)
             for i in range(0, len(original_line), width)
         ]
     return new_lines

zrb/builtin/schedule.py

@@ -1,35 +1,31 @@
+from zrb.builtin.helper.reccuring_action import create_recurring_action
+from zrb.runner import runner
 from zrb.task.recurring_task import RecurringTask
 from zrb.task.time_watcher import TimeWatcher
 from zrb.task_input.str_input import StrInput
-from zrb.runner import runner
-from zrb.builtin.helper.reccuring_action import create_recurring_action
-
 
 schedule = RecurringTask(
-    name='schedule',
-    icon='📅',
-    color='yellow',
-    description='Show message/run command periodically',
+    name="schedule",
+    icon="📅",
+    color="yellow",
+    description="Show message/run command periodically",
     inputs=[
         StrInput(
-            name='schedule',
-            default='* * * * *',
-            prompt='Schedule cron pattern (minute hour day(month) month day(week)',  # noqa
-            description='Schedule cron pattern to show the message'
+            name="schedule",
+            default="* * * * *",
+            prompt="Schedule cron pattern (minute hour day(month) month day(week)",  # noqa
+            description="Schedule cron pattern to show the message",
         ),
     ],
     triggers=[
         TimeWatcher(
-            name='watch-time',
-            color='cyan',
-            icon='⏰',
-            schedule='{{input.schedule}}'
+            name="watch-time", color="cyan", icon="⏰", schedule="{{input.schedule}}"
         )
     ],
     task=create_recurring_action(
-        notif_title='Schedule',
-        trigger_caption='Schedule',
-        trigger_xcom_key='watch-time.scheduled-time'
-    )
+        notif_title="Schedule",
+        trigger_caption="Schedule",
+        trigger_xcom_key="watch-time.scheduled-time",
+    ),
 )
 runner.register(schedule)

zrb/builtin/ubuntu.py

@@ -1,12 +1,21 @@
+from zrb.builtin.devtool.devtool_install import (
+    install_aws,
+    install_docker,
+    install_gcloud,
+    install_gvm,
+    install_helix,
+    install_helm,
+    install_kubectl,
+    install_nvm,
+    install_pyenv,
+    install_sdkman,
+    install_tmux,
+    install_zsh,
+)
 from zrb.builtin.group import ubuntu_group, ubuntu_install_group
+from zrb.runner import runner
 from zrb.task.cmd_task import CmdTask
 from zrb.task.flow_task import FlowTask
-from zrb.runner import runner
-from zrb.builtin.devtool.devtool_install import (
-    install_tmux, install_zsh, install_gvm, install_nvm, install_pyenv,
-    install_sdkman, install_aws, install_gcloud, install_docker,
-    install_kubectl, install_helm, install_helix
-)
 
 ###############################################################################
 # Task Definitions
@@ -14,72 +23,72 @@ from zrb.builtin.devtool.devtool_install import (
 
 
 update = CmdTask(
-    name='update',
+    name="update",
     group=ubuntu_group,
-    description='Update ubuntu',
+    description="Update ubuntu",
     cmd=[
-        'sudo apt update',
-        'sudo apt upgrade -y',
+        "sudo apt update",
+        "sudo apt upgrade -y",
     ],
     retry_interval=3,
-    preexec_fn=None
+    preexec_fn=None,
 )
 runner.register(update)
 
 install_toys = CmdTask(
-    name='toys',
+    name="toys",
     group=ubuntu_install_group,
-    description='Install ubuntu toy packages',
+    description="Install ubuntu toy packages",
     cmd=[
-        'sudo apt install -y lolcat cowsay figlet neofetch',
+        "sudo apt install -y lolcat cowsay figlet neofetch",
     ],
     retry_interval=3,
-    preexec_fn=None
+    preexec_fn=None,
 )
 update_and_install_toys: CmdTask = install_toys.copy()
 update_and_install_toys.add_upstream(update)
 runner.register(update_and_install_toys)
 
 install_essentials = CmdTask(
-    name='essentials',
+    name="essentials",
     group=ubuntu_install_group,
-    description='Install ubuntu essential packages',
+    description="Install ubuntu essential packages",
     cmd=[
-        'sudo apt install -y \\',
-        'build-essential python3-distutils libssl-dev zlib1g-dev \\'
-        'libbz2-dev libreadline-dev libsqlite3-dev libpq-dev python3-dev \\',
-        'llvm libncurses5-dev libncursesw5-dev xz-utils tk-dev libffi-dev \\',
-        'liblzma-dev python3-openssl libblas-dev liblapack-dev rustc \\',
-        'golang gfortran fd-find ripgrep wget curl git ncat zip unzip \\',
-        'cmake make tree tmux zsh neovim xdotool xsel'
+        "sudo apt install -y \\",
+        "build-essential python3-distutils libssl-dev zlib1g-dev \\"
+        "libbz2-dev libreadline-dev libsqlite3-dev libpq-dev python3-dev \\",
+        "llvm libncurses5-dev libncursesw5-dev xz-utils tk-dev libffi-dev \\",
+        "liblzma-dev python3-openssl libblas-dev liblapack-dev rustc \\",
+        "golang gfortran fd-find ripgrep wget curl git ncat zip unzip \\",
+        "cmake make tree tmux zsh neovim xdotool xsel",
     ],
     retry_interval=3,
-    preexec_fn=None
+    preexec_fn=None,
 )
 update_and_install_essentials: CmdTask = install_essentials.copy()
 update_and_install_essentials.add_upstream(update)
 runner.register(update_and_install_essentials)
 
 install_tex = CmdTask(
-    name='tex',
+    name="tex",
     group=ubuntu_install_group,
-    description='Install ubuntu tex packages',
+    description="Install ubuntu tex packages",
     cmd=[
-        'sudo apt install -y \\',
-        'texlive-full texlive-latex-base texlive-fonts-recommended \\',
-        'texlive-fonts-extra texlive-latex-extra'
+        "sudo apt install -y \\",
+        "texlive-full texlive-latex-base texlive-fonts-recommended \\",
+        "texlive-fonts-extra texlive-latex-extra",
     ],
     retry_interval=3,
-    preexec_fn=None
+    preexec_fn=None,
 )
 update_and_install_tex: CmdTask = install_tex.copy()
 update_and_install_tex.add_upstream(update)
 runner.register(update_and_install_tex)
 
 install_all = FlowTask(
-    name='all',
+    name="all",
     group=ubuntu_install_group,
-    description='Install all ubuntu packages',
+    description="Install all ubuntu packages",
     steps=[
         update,
         install_essentials,
@@ -96,7 +105,7 @@ install_all = FlowTask(
         install_docker,
         install_kubectl,
         install_helm,
-        install_helix
-    ]
+        install_helix,
+    ],
 )
 runner.register(install_all)

zrb/builtin/update.py

@@ -1,14 +1,14 @@
-from zrb.task.cmd_task import CmdTask
 from zrb.runner import runner
+from zrb.task.cmd_task import CmdTask
 
 ###############################################################################
 # Task Definitions
 ###############################################################################
 
 update = CmdTask(
-    name='update',
-    description='Update zrb',
-    cmd='pip install zrb -U',
-    checking_interval=3
+    name="update",
+    description="Update zrb",
+    cmd="pip install zrb -U",
+    checking_interval=3,
 )
 runner.register(update)

zrb/builtin/version.py

@@ -1,17 +1,13 @@
+from zrb.config.config import version
 from zrb.helper.typing import Any
-from zrb.task.decorator import python_task
 from zrb.runner import runner
-from zrb.config.config import version
+from zrb.task.decorator import python_task
 
 ###############################################################################
 # Task Definitions
 ###############################################################################
 
 
-@python_task(
-    name='version',
-    description='Get Zrb version',
-    runner=runner
-)
+@python_task(name="version", description="Get Zrb version", runner=runner)
 async def get_version(*args: Any, **kwargs: Any) -> str:
     return version