zrb 0.28.1__py3-none-any.whl → 1.0.0a1__py3-none-any.whl

zrb/builtin/monorepo/push.py

@@ -1,24 +0,0 @@
-from datetime import datetime
-
-from zrb.builtin.monorepo._group import monorepo_group
-from zrb.builtin.monorepo._task import PUSH_SUBREPO_UPSTREAM
-from zrb.runner import runner
-from zrb.task.task import Task
-from zrb.task_input.str_input import StrInput
-
-push_to_monorepo = Task(
-    name="push",
-    group=monorepo_group,
-    description="Pushing to subrepos",
-    inputs=[
-        StrInput(
-            name="message",
-            shortcut="m",
-            prompt="Commit Messsage",
-            default=lambda m: f"Synchronize subrepos at {datetime.now()}",
-        )
-    ],
-    retry=0,
-)
-PUSH_SUBREPO_UPSTREAM >> push_to_monorepo
-runner.register(push_to_monorepo)

zrb/builtin/process/__init__.py

@@ -1,11 +0,0 @@
-from zrb.builtin.process._group import process_group
-from zrb.builtin.process.pid import (
-    get_process_pid_by_name,
-    get_process_pid_by_port,
-    process_pid_group,
-)
-
-assert process_group
-assert process_pid_group
-assert get_process_pid_by_name
-assert get_process_pid_by_port

zrb/builtin/process/_group.py

@@ -1,3 +0,0 @@
-from zrb.task_group.group import Group
-
-process_group = Group(name="process", description="Process related commands")

zrb/builtin/process/pid/__init__.py

@@ -1,7 +0,0 @@
-from zrb.builtin.process.pid._group import process_pid_group
-from zrb.builtin.process.pid.get_by_name import get_process_pid_by_name
-from zrb.builtin.process.pid.get_by_port import get_process_pid_by_port
-
-assert process_pid_group
-assert get_process_pid_by_name
-assert get_process_pid_by_port

zrb/builtin/process/pid/_group.py

@@ -1,6 +0,0 @@
-from zrb.builtin.process._group import process_group
-from zrb.task_group.group import Group
-
-process_pid_group = Group(
-    name="pid", parent=process_group, description="PID related commands"
-)

zrb/builtin/process/pid/get_by_name.py

@@ -1,22 +0,0 @@
-from zrb.builtin.process._group import process_group
-from zrb.runner import runner
-from zrb.task.cmd_task import CmdTask
-from zrb.task_input.str_input import StrInput
-
-get_process_pid_by_name = CmdTask(
-    name="get-pid-by-name",
-    group=process_group,
-    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",
-        )
-    ],
-    cmd='pgrep "{{ input.name }}" || exit 0',
-    checking_interval=3,
-)
-runner.register(get_process_pid_by_name)

zrb/builtin/process/pid/get_by_port.py

@@ -1,22 +0,0 @@
-from zrb.builtin.process._group import process_group
-from zrb.runner import runner
-from zrb.task.cmd_task import CmdTask
-from zrb.task_input.int_input import IntInput
-
-get_process_pid_by_port = CmdTask(
-    name="get-pid-by-port",
-    group=process_group,
-    description="Get PID by port",
-    inputs=[
-        IntInput(
-            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,
-)
-runner.register(get_process_pid_by_port)

zrb/builtin/project/__init__.py

@@ -1,35 +0,0 @@
-from zrb.builtin.project._group import project_group
-from zrb.builtin.project.add import (
-    add_cmd_task,
-    add_docker_compose_task,
-    add_fastapp_application,
-    add_fastapp_crud,
-    add_fastapp_field,
-    add_fastapp_module,
-    add_plugin,
-    add_project_tasks,
-    add_python_app,
-    add_python_task,
-    project_add_app_group,
-    project_add_fastapp_group,
-    project_add_group,
-    project_add_task_group,
-)
-from zrb.builtin.project.create import create_project
-
-assert project_group
-assert project_add_group
-assert project_add_app_group
-assert project_add_fastapp_group
-assert project_add_task_group
-assert create_project
-assert add_plugin
-assert add_project_tasks
-assert add_cmd_task
-assert add_docker_compose_task
-assert add_fastapp_application
-assert add_fastapp_crud
-assert add_fastapp_module
-assert add_fastapp_field
-assert add_python_task
-assert add_python_app

zrb/builtin/project/_group.py

@@ -1,3 +0,0 @@
-from zrb.task_group.group import Group
-
-project_group = Group(name="project", description="Project management")

zrb/builtin/project/_helper.py

@@ -1,98 +0,0 @@
-import os
-from typing import Any, Optional
-
-from zrb.builtin.project._input import project_dir_input
-from zrb.helper import util
-from zrb.helper.accessories.color import colored
-from zrb.helper.codemod.add_assert_resource import add_assert_resource
-from zrb.helper.codemod.add_import_module import add_import_module
-from zrb.helper.file.text import read_text_file_async, write_text_file_async
-from zrb.helper.typecheck import typechecked
-from zrb.task.any_task import AnyTask
-from zrb.task.decorator import python_task
-from zrb.task.task import Task
-from zrb.task_input.any_input import AnyInput
-
-
-@typechecked
-def validate_existing_project_dir(project_dir: str):
-    if not os.path.isfile(os.path.join(project_dir, "zrb_init.py")):
-        raise Exception(f"Not a project: {project_dir}")
-
-
-@typechecked
-def validate_inexisting_automation(project_dir: str, automation_name: str):
-    validate_inexisting_automation_package(project_dir, automation_name)
-    validate_inexisting_automation_module(project_dir, automation_name)
-
-
-@typechecked
-def validate_inexisting_automation_package(project_dir: str, package_name: str):
-    package_dir = os.path.join(
-        project_dir, "_automate", f"{util.to_snake_case(package_name)}"
-    )
-    if os.path.exists(package_dir):
-        raise Exception(f"Automation package already exists: {package_dir}")
-
-
-@typechecked
-def validate_inexisting_automation_module(project_dir: str, module_name: str):
-    module_file = os.path.join(
-        project_dir, "_automate", f"{util.to_snake_case(module_name)}.py"
-    )
-    if os.path.exists(module_file):
-        raise Exception(f"Automation module already exists: {module_file}")
-
-
-@typechecked
-async def register_module_to_project(
-    project_dir: str, module_path: str, alias: Optional[str] = None
-):
-    zrb_init_path = os.path.join(project_dir, "zrb_init.py")
-    code = await read_text_file_async(zrb_init_path)
-    if alias is None:
-        alias = module_path.split(".")[-1]
-    code = add_import_module(code=code, module_path=module_path, alias=alias)
-    code = add_assert_resource(code, alias)
-    await write_text_file_async(zrb_init_path, code)
-    return True
-
-
-@typechecked
-def create_register_module(
-    module_path: str,
-    inputs: Optional[list[AnyInput]] = None,
-    upstreams: Optional[list[AnyTask]] = None,
-    alias: Optional[str] = None,
-    task_name: str = "register-module",
-) -> Task:
-    @python_task(
-        name=task_name,
-        inputs=[project_dir_input] + inputs if inputs is not None else [],
-        upstreams=upstreams if upstreams is not None else [],
-    )
-    async def register_module(*args: Any, **kwargs: Any):
-        task: Task = kwargs.get("_task")
-        project_dir = kwargs.get("project_dir")
-        validate_existing_project_dir(project_dir)
-        rendered_module_path = task.render_str(module_path)
-        rendered_alias: Optional[str] = None
-        if alias is not None:
-            rendered_alias = task.render_str(alias)
-        module_import_label = _get_module_import_label(
-            module_path=rendered_module_path, alias=rendered_alias
-        )
-        task.print_out(colored(module_import_label, color="yellow"))
-        await register_module_to_project(
-            project_dir=project_dir,
-            module_path=rendered_module_path,
-            alias=rendered_alias,
-        )
-
-    return register_module
-
-
-def _get_module_import_label(module_path: str, alias: Optional[str]):
-    if alias == "" or alias is None:
-        return module_path
-    return f"{module_path} as {alias}"

zrb/builtin/project/_input.py

@@ -1,13 +0,0 @@
-import os
-
-from zrb.task_input.str_input import StrInput
-
-_PROJECT_DIR = os.getenv("ZRB_PROJECT_DIR", ".")
-
-project_dir_input = StrInput(
-    name="project-dir",
-    shortcut="d",
-    description="Project directory",
-    prompt="Project directory",
-    default=_PROJECT_DIR,
-)

zrb/builtin/project/add/__init__.py

@@ -1,37 +0,0 @@
-from zrb.builtin.project.add._group import project_add_group
-from zrb.builtin.project.add.app import (
-    add_app_generator,
-    add_python_app,
-    project_add_app_group,
-)
-from zrb.builtin.project.add.fastapp import (
-    add_fastapp_application,
-    add_fastapp_crud,
-    add_fastapp_field,
-    add_fastapp_module,
-    project_add_fastapp_group,
-)
-from zrb.builtin.project.add.plugin import add_plugin
-from zrb.builtin.project.add.project_task import add_project_tasks
-from zrb.builtin.project.add.task import (
-    add_cmd_task,
-    add_docker_compose_task,
-    add_python_task,
-    project_add_task_group,
-)
-
-assert project_add_group
-assert project_add_app_group
-assert project_add_fastapp_group
-assert project_add_task_group
-assert add_app_generator
-assert add_python_app
-assert add_fastapp_application
-assert add_fastapp_crud
-assert add_fastapp_field
-assert add_fastapp_module
-assert add_plugin
-assert add_project_tasks
-assert add_cmd_task
-assert add_docker_compose_task
-assert add_python_task

zrb/builtin/project/add/_group.py

@@ -1,6 +0,0 @@
-from zrb.builtin.project._group import project_group
-from zrb.task_group.group import Group
-
-project_add_group = Group(
-    name="add", description="Add resources to project", parent=project_group
-)

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

@@ -1,7 +0,0 @@
-from zrb.builtin.project.add.app._group import project_add_app_group
-from zrb.builtin.project.add.app.generator import add_app_generator
-from zrb.builtin.project.add.app.python import add_python_app
-
-assert project_add_app_group
-assert add_app_generator
-assert add_python_app

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

@@ -1,6 +0,0 @@
-from zrb.builtin.project.add._group import project_add_group
-from zrb.task_group.group import Group
-
-project_add_app_group = Group(
-    name="app", description="Add app/generator to project", parent=project_add_group
-)

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

@@ -1,3 +0,0 @@
-from zrb.builtin.project.add.app.generator.generator import add_app_generator
-
-assert add_app_generator

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

@@ -1,15 +0,0 @@
-import os
-
-from zrb.helper.typecheck import typechecked
-from zrb.helper.util import to_kebab_case, to_snake_case
-
-
-@typechecked
-def validate_existing_package(project_dir: str, package_name: str):
-    kebab_package_name = to_kebab_case(package_name)
-    snake_package_name = to_snake_case(package_name)
-    package_init_path = os.path.join(
-        project_dir, "src", kebab_package_name, "src", snake_package_name, "__init__.py"
-    )
-    if not os.path.isfile(package_init_path):
-        raise Exception(f"Package not exists: {package_init_path}")

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

@@ -1,31 +0,0 @@
-from zrb.helper.accessories.name import get_random_name
-from zrb.task_input.int_input import IntInput
-from zrb.task_input.str_input import StrInput
-
-package_name_input = StrInput(
-    name="package-name",
-    description="Package name",
-    prompt="Package name",
-    default=get_random_name(),
-)
-
-generator_name_input = StrInput(
-    name="generator-name",
-    description="Generator name",
-    prompt="Generator name",
-    default=get_random_name(),
-)
-
-generator_base_image_input = StrInput(
-    name="generator-base-image",
-    description="Base image",
-    prompt="Base image",
-    default="python:3.10-slim",
-)
-
-generator_app_port_input = IntInput(
-    name="generator-app-port",
-    description="Default app port",
-    prompt="Default app port",
-    default="8080",
-)

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

@@ -1,107 +0,0 @@
-import os
-from typing import Any
-
-from zrb.builtin.project._helper import validate_existing_project_dir
-from zrb.builtin.project._input import project_dir_input
-from zrb.builtin.project.add.app._group import project_add_app_group
-from zrb.builtin.project.add.app.generator._helper import validate_existing_package
-from zrb.builtin.project.add.app.generator._input import (
-    generator_app_port_input,
-    generator_base_image_input,
-    generator_name_input,
-    package_name_input,
-)
-from zrb.helper.accessories.color import colored
-from zrb.helper.codemod.add_assert_resource import add_assert_resource
-from zrb.helper.codemod.add_import_module import add_import_module
-from zrb.helper.util import to_kebab_case, to_snake_case
-from zrb.runner import runner
-from zrb.task.decorator import python_task
-from zrb.task.resource_maker import ResourceMaker
-from zrb.task.task import Task
-
-_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")
-    validate_existing_project_dir(project_dir)
-    package_name = kwargs.get("package_name")
-    validate_existing_package(project_dir, package_name)
-
-
-copy_resource = ResourceMaker(
-    name="copy-resource",
-    inputs=[
-        project_dir_input,
-        package_name_input,
-        generator_name_input,
-        generator_base_image_input,
-        generator_app_port_input,
-    ],
-    upstreams=[validate],
-    replacements={
-        "zrbPackageName": "{{input.package_name}}",
-        "zrbGeneratorName": "{{input.generator_name}}",
-        "zrbGeneratorBaseImage": "{{input.generator_base_image}}",
-        "zrbGeneratorAppPort": "{{input.generator_app_port}}",
-    },
-    template_path=os.path.join(_CURRENT_DIR, "template"),
-    destination_path="{{ input.project_dir }}",
-    excludes=[
-        "*/deployment/venv",
-        "*/__pycache__",
-    ],
-)
-
-
-@python_task(
-    name="register-generator",
-    inputs=[
-        project_dir_input,
-        package_name_input,
-        generator_name_input,
-    ],
-    upstreams=[copy_resource],
-)
-def register_generator(*args: Any, **kwargs: Any):
-    task = kwargs.get("_task")
-    project_dir = kwargs.get("project_dir")
-    package_name = kwargs.get("package_name")
-    generator_name = kwargs.get("generator_name")
-    kebab_package_name = to_kebab_case(package_name)
-    snake_package_name = to_snake_case(package_name)
-    snake_generator_name = to_snake_case(generator_name)
-    package_init_path = os.path.join(
-        project_dir, "src", kebab_package_name, "src", snake_package_name, "__init__.py"
-    )
-    with open(package_init_path, "r") as f:
-        code = f.read()
-    new_code = add_import_module(
-        code=code,
-        module_path=f"{snake_package_name}.{snake_generator_name}",
-        alias=snake_generator_name,
-    )
-    new_code = add_assert_resource(code=new_code, resource=snake_generator_name)
-    with open(package_init_path, "w") as f:
-        f.write(new_code)
-    task.print_out(colored("Register generator", color="yellow"))
-
-
-@python_task(
-    name="generator",
-    group=project_add_app_group,
-    description="Add app generator",
-    upstreams=[
-        register_generator,
-    ],
-    runner=runner,
-)
-async def add_app_generator(*args: Any, **kwargs: Any):
-    task: Task = kwargs.get("_task")
-    task.print_out(colored("Appp generator added", color="yellow"))

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

@@ -1,3 +0,0 @@
-from .snake_zrb_generator_name import add_snake_zrb_generator_name
-
-assert add_snake_zrb_generator_name

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

@@ -1,65 +0,0 @@
-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
-
-SYSTEM_USER = os.getenv("USER", "incognito")
-
-app_name_input = StrInput(
-    name="app-name",
-    shortcut="a",
-    description="App name",
-    prompt="App name",
-    default="app",
-)
-
-app_author_name_input = StrInput(
-    name="app-author-name",
-    prompt="App author name",
-    description="App author name",
-    default=SYSTEM_USER,
-)
-
-app_author_email_input = StrInput(
-    name="app-author-email",
-    prompt="App author email",
-    description="App author email",
-    default=f"{SYSTEM_USER}@gmail.com",
-)
-
-app_description_input = StrInput(
-    name="app-description",
-    description="App description",
-    prompt="App description",
-    default="Just another app",
-)
-
-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=lambda m: "/".join(
-        [app_image_default_namespace, to_kebab_case(m.get("app_name"))]
-    ),
-)
-
-http_port_input = IntInput(
-    name="http-port",
-    shortcut="p",
-    description="HTTP port",
-    prompt="HTTP port",
-    default="zrbGeneratorAppPort",
-)
-
-env_prefix_input = StrInput(
-    name="env-prefix",
-    description="OS environment prefix",
-    prompt="OS environment prefix",
-    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

@@ -1,106 +0,0 @@
-import os
-from typing import Any
-
-from zrb.builtin.project._helper import (
-    create_register_module,
-    validate_existing_project_dir,
-    validate_inexisting_automation,
-)
-from zrb.builtin.project._input import project_dir_input
-from zrb.builtin.project.add.app._group import project_add_app_group
-from zrb.helper.accessories.color import colored
-from zrb.helper.util import to_kebab_case
-from zrb.runner import runner
-from zrb.task.decorator import python_task
-from zrb.task.resource_maker import ResourceMaker
-from zrb.task.task import Task
-
-from ._input import (
-    app_author_email_input,
-    app_author_name_input,
-    app_description_input,
-    app_image_name_input,
-    app_name_input,
-    env_prefix_input,
-    http_port_input,
-)
-
-_CURRENT_DIR = os.path.dirname(__file__)
-_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")
-    validate_existing_project_dir(project_dir)
-    app_name = kwargs.get("app_name")
-    validate_inexisting_automation(project_dir, app_name)
-    source_dir = os.path.join(project_dir, "src", f"{to_kebab_case(app_name)}")
-    if os.path.exists(source_dir):
-        raise Exception(f"Source already exists: {source_dir}")
-
-
-copy_resource = ResourceMaker(
-    name="copy-resource",
-    inputs=[
-        project_dir_input,
-        app_name_input,
-        app_description_input,
-        app_author_name_input,
-        app_author_email_input,
-        app_image_name_input,
-        http_port_input,
-        env_prefix_input,
-    ],
-    upstreams=[validate],
-    replacements={
-        "zrbAppName": "{{input.app_name}}",
-        "zrbAppDescription": "{{input.app_description}}",
-        "zrbAppAuthorName": "{{input.app_author_name}}",
-        "zrbAppAuthorEmail": "{{input.app_author_email}}",
-        "zrbAppHttpPort": '{{util.coalesce(input.http_port, "3000")}}',
-        "ZRB_ENV_PREFIX": '{{util.coalesce(input.env_prefix, "MY").upper()}}',
-        "zrb-app-image-name": "{{input.app_image_name}}",
-    },
-    template_path=os.path.join(_CURRENT_DIR, "template"),
-    destination_path="{{ input.project_dir }}",
-    excludes=[
-        "*/deployment/venv",
-        "*/__pycache__",
-    ],
-)
-
-register_module = create_register_module(
-    module_path=f"_automate.{_snake_app_name_tpl}",
-    alias=f"_automate_{_snake_app_name_tpl}",
-    inputs=[app_name_input],
-    upstreams=[copy_resource],
-)
-
-
-@python_task(
-    name="snake_zrb_generator_name",
-    group=project_add_app_group,
-    description="Add human readable zrb generator name",
-    inputs=[
-        project_dir_input,
-        app_name_input,
-        app_description_input,
-        app_author_name_input,
-        app_author_email_input,
-        app_image_name_input,
-        http_port_input,
-        env_prefix_input,
-    ],
-    upstreams=[
-        register_module,
-    ],
-    runner=runner,
-)
-async def add_snake_zrb_generator_name(*args: Any, **kwargs: Any):
-    task: Task = kwargs.get("_task")
-    task.print_out(colored("Human readable zrb generator name added", color="yellow"))