blue-assistant 4.32.1__py3-none-any.whl → 4.43.1__py3-none-any.whl

blue_assistant/.abcli/script/run.sh

@@ -3,18 +3,24 @@
 function blue_assistant_script_run() {
     local options=$1
     local do_dryrun=$(abcli_option_int "$options" dryrun 0)
-    local do_download=$(abcli_option_int "$options" download $(abcli_not $do_dryrun))
+    local do_download=$(abcli_option_int "$options" download 0)
     local do_upload=$(abcli_option_int "$options" upload $(abcli_not $do_dryrun))
 
-    local object_name=$(abcli_clarify_object $2 .)
+    local script_options=$2
+    local script_name=$(abcli_option "$script_options" script generic)
+
+    local object_name=$(abcli_clarify_object $3 $script_name-$(abcli_string_timestamp_short))
     [[ "$do_download" == 1 ]] &&
         abcli_download - $object_name
 
+    abcli_log "running $script_name -> $object_name ..."
+
     abcli_eval dryrun=$do_dryrun \
         python3 -m blue_assistant.script \
         run \
+        --script_name $script_name \
         --object_name $object_name \
-        "${@:3}"
+        "${@:4}"
     [[ $? -ne 0 ]] && return 1
 
     [[ "$do_upload" == 1 ]] &&

blue_assistant/.abcli/tests/script_run.sh

@@ -2,10 +2,26 @@
 
 function test_blue_assistant_script_run() {
     local options=$1
+    local list_of_script_name=$(python3 -m blue_assistant.script \
+        get_list_of \
+        --delim +)
+    list_of_script_name=$(abcli_option "$options" script $list_of_script_name)
 
-    abcli_eval ,$options \
-        blue_assistant_script_run \
-        ~upload,$options \
-        $BLUE_ASSISTANT_TEST_OBJECT \
-        "${@:2}"
+    local script_name
+    for script_name in $(echo "$list_of_script_name" | tr + " "); do
+        abcli_log "testing $script_name ..."
+
+        abcli_eval ,$options \
+            blue_assistant_script_run \
+            ~upload,$options \
+            script=$script_name \
+            test_blue_assistant_script_run-$(abcli_string_timestamp_short) \
+            "${@:2}"
+
+        [[ $? -ne 0 ]] && return 1
+
+        abcli_hr
+    done
+
+    return 0
 }

blue_assistant/__init__.py

@@ -1,10 +1,10 @@
 NAME = "blue_assistant"
 
-ICON = "🌀"
+ICON = "🧠"
 
 DESCRIPTION = f"{ICON} An AI Assistant."
 
-VERSION = "4.32.1"
+VERSION = "4.43.1"
 
 REPO_NAME = "blue-assistant"
 

blue_assistant/help/script.py

@@ -2,12 +2,16 @@ from typing import List
 
 from blue_options.terminal import show_usage, xtra
 
+from blue_assistant.script.repository import list_of_script_names
+
 
 def help_run(
     tokens: List[str],
     mono: bool,
 ) -> str:
-    options = xtra("~download,dryrun,~upload", mono=mono)
+    options = xtra("download,dryrun,~upload", mono=mono)
+
+    script_options = "script=<script>"
 
     args = ["--verbose 1"]
 
@@ -17,10 +21,14 @@ def help_run(
             "script",
             "run",
             f"[{options}]",
-            "[.|<object-name>]",
+            f"[{script_options}]",
+            "[-|<object-name>]",
         ]
         + args,
         "run <object-name>.",
+        {
+            "script: {}".format(" | ".join(list_of_script_names)): [],
+        },
         mono=mono,
     )
 

blue_assistant/script/__main__.py

@@ -4,7 +4,8 @@ from blueness import module
 from blueness.argparse.generic import sys_exit
 
 from blue_assistant import NAME
-from blue_assistant.script.functions import load_script
+from blue_assistant.script.load import load_script
+from blue_assistant.script.repository import list_of_script_names
 from blue_assistant.logger import logger
 
 NAME = module.name(__file__, NAME)
@@ -13,7 +14,12 @@ parser = argparse.ArgumentParser(NAME)
 parser.add_argument(
     "task",
     type=str,
-    help="run",
+    help="get_list_of | run",
+)
+parser.add_argument(
+    "--script_name",
+    type=str,
+    help=" | ".join(list_of_script_names),
 )
 parser.add_argument(
     "--object_name",
@@ -25,17 +31,27 @@ parser.add_argument(
     default=0,
     help="0 | 1",
 )
+parser.add_argument(
+    "--delim",
+    type=str,
+    default="+",
+)
 args = parser.parse_args()
 
+delim = " " if args.delim == "space" else args.delim
+
 success = False
-if args.task == "run":
+if args.task == "get_list_of":
+    success = True
+    print(delim.join(list_of_script_names))
+elif args.task == "run":
     success, script = load_script(
-        object_name=args.object_name,
+        script_name=args.script_name,
         verbose=args.verbose == 1,
     )
 
     if success:
-        success = script.run()
+        success = script.run(object_name=args.object_name)
 else:
     success = None
 

blue_assistant/script/load.py

@@ -0,0 +1,17 @@
+from typing import Tuple
+
+from blue_assistant.script.repository import list_of_script_classes
+from blue_assistant.script.repository.generic import GenericScript
+from blue_assistant.logger import logger
+
+
+def load_script(
+    script_name: str,
+    verbose: bool = False,
+) -> Tuple[bool, GenericScript]:
+    for script_class in list_of_script_classes:
+        if script_class.name == script_name:
+            return True, script_class(verbose=verbose)
+
+    logger.error(f"{script_name}: script not found.")
+    return False, GenericScript(verbose=verbose)

blue_assistant/script/repository/__init__.py

@@ -0,0 +1,15 @@
+from typing import List, Type
+
+from blue_assistant.script.repository.generic import GenericScript
+from blue_assistant.script.repository.blue_amo.classes import BlueAmoScript
+from blue_assistant.script.repository.moon_datasets.classes import MiningOnMoonScript
+
+list_of_script_classes: List[Type[GenericScript]] = [
+    GenericScript,
+    BlueAmoScript,
+    MiningOnMoonScript,
+]
+
+list_of_script_names: List[str] = [
+    script_class.name for script_class in list_of_script_classes
+]

blue_assistant/script/repository/blue_amo/__init__.py

@@ -0,0 +1 @@
+from blue_assistant.script.repository.blue_amo.classes import BlueAmoScript

blue_assistant/script/repository/blue_amo/classes.py

@@ -0,0 +1,42 @@
+from typing import Dict
+from tqdm import tqdm
+
+from blue_objects import file, path
+from blue_objects.metadata import post_to_object
+
+from blue_assistant.script.repository.generic.classes import GenericScript
+from blue_assistant.logger import logger
+
+
+class BlueAmoScript(GenericScript):
+    name = path.name(file.path(__file__))
+
+    def __init__(
+        self,
+        verbose: bool = False,
+    ):
+        super().__init__(verbose=verbose)
+
+    def run(
+        self,
+        object_name: str,
+    ) -> bool:
+        if not super().run(object_name=object_name):
+            return False
+
+        metadata: Dict[Dict] = {"nodes": {}}
+        for node_name, node in tqdm(self.script["nodes"].items()):
+            logger.info(
+                "{}{}".format(
+                    node_name,
+                    f": {node}" if self.verbose else " ...",
+                )
+            )
+
+            metadata["nodes"][node_name] = "..."
+
+        return post_to_object(
+            object_name,
+            "output",
+            metadata,
+        )

blue_assistant/script/repository/generic/__init__.py

@@ -0,0 +1 @@
+from blue_assistant.script.repository.generic.classes import GenericScript

blue_assistant/script/repository/generic/classes.py

@@ -0,0 +1,51 @@
+import os
+
+from blueness import module
+from blue_objects import file, path
+from blue_objects.metadata import post_to_object
+
+from blue_assistant import NAME
+from blue_assistant.logger import logger
+
+
+NAME = module.name(__file__, NAME)
+
+
+class GenericScript:
+    name = path.name(file.path(__file__))
+
+    def __init__(
+        self,
+        verbose: bool = False,
+    ):
+        self.verbose = verbose
+
+        metadata_filename = os.path.join(
+            file.path(__file__),
+            f"../{self.name}",
+            "metadata.yaml",
+        )
+        success, self.metadata = file.load_yaml(metadata_filename)
+
+        self.script = self.metadata.get("script", [])
+
+        assert success, f"cannot load {self.name}/metadata.yaml"
+
+    def run(
+        self,
+        object_name: str,
+    ) -> bool:
+        logger.info(
+            "{}.run: {}:{} -> {}".format(
+                NAME,
+                self.__class__.__name__,
+                self.name,
+                object_name,
+            )
+        )
+
+        return post_to_object(
+            object_name,
+            "input",
+            self.script,
+        )

blue_assistant/script/repository/moon_datasets/__init__.py

@@ -0,0 +1 @@
+from blue_assistant.script.repository.moon_datasets.classes import MiningOnMoonScript

blue_assistant/script/repository/moon_datasets/classes.py

@@ -0,0 +1,16 @@
+from typing import Dict
+from tqdm import tqdm
+
+from blue_objects import file, path
+
+from blue_assistant.script.repository.generic.classes import GenericScript
+
+
+class MiningOnMoonScript(GenericScript):
+    name = path.name(file.path(__file__))
+
+    def __init__(
+        self,
+        verbose: bool = False,
+    ):
+        super().__init__(verbose=verbose)

{blue_assistant-4.32.1.dist-info → blue_assistant-4.43.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.32.1
-Summary: 🌀 An AI Assistant.
+Version: 4.43.1
+Summary: 🧠 An AI Assistant.
 Home-page: https://github.com/kamangir/blue-assistant
 Author: Arash Abadpour (Kamangir)
 Author-email: arash@kamangir.net
@@ -38,9 +38,9 @@ Dynamic: license
 Dynamic: requires-dist
 Dynamic: summary
 
-# 🌀 blue-assistant
+# 🧠 blue-assistant
 
-🌀 `@assistant` is an AI assistant.
+🧠 `@assistant` is an AI assistant.
 
 ```bash
 pip install blue-assistant
@@ -48,10 +48,12 @@ pip install blue-assistant
 
 ```mermaid
 graph LR
-    assistant_script_run["@assistant<br>script<br>run -<br>&lt;object-name&gt;"]
+    assistant_script_run["@assistant<br>script<br>run -<br>&lt;script&gt;<br>&lt;object-name&gt;"]
 
+    script["📜 script"]:::folder
     object["📂 object"]:::folder
 
+    script --> assistant_script_run
     object --> assistant_script_run
     assistant_script_run --> object
 
@@ -63,4 +65,4 @@ graph LR
 
 [![pylint](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/blue-assistant.svg)](https://pypi.org/project/blue-assistant/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/blue-assistant)](https://pypistats.org/packages/blue-assistant)
 
-built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🌀 [`blue_assistant-4.32.1`](https://github.com/kamangir/blue-assistant).
+built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.43.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.32.1.dist-info → blue_assistant-4.43.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 blue_assistant/README.py,sha256=q6EWCy7zojQESGDF-2xEoXS1Bl0wRJTe2jw_1QM_WYw,600
-blue_assistant/__init__.py,sha256=ZuCUkTSAl4hWl6Blzyia-k5zUb2pzyiKZFta3LVI31A,310
+blue_assistant/__init__.py,sha256=BvFZ2zMLuhu23_eXZnSReXQU0tuNy0WUnaYXWAfEiNw,310
 blue_assistant/__main__.py,sha256=URtal70XZc0--3FDTYWcLtnGOqBYjMX9gt-L1k8hDXI,361
 blue_assistant/config.env,sha256=es6XIU6yOA69IV0WXtLru5uTM5Dvvmq-q6BJmcE8NyY,36
 blue_assistant/env.py,sha256=TG2YL5jn7qJApHIGi7IuXvqcPFT771zXKJHomGC63Z4,189
@@ -14,23 +14,27 @@ blue_assistant/.abcli/alias.sh,sha256=MCJzXaDnX1QMllWsZJJkDePBYt1nY9ZWa3o4msfGD2
 blue_assistant/.abcli/blue_assistant.sh,sha256=plLTQQerVmfb_SNlOkv0MEaQCF7YdsOHzCq0M3FWT4c,239
 blue_assistant/.abcli/browse.sh,sha256=qZ_RK_WnsjmF-hfWKiMEOnnv22QtZh9HQ0VFJUbP6aI,294
 blue_assistant/.abcli/script.sh,sha256=XIkY4eZyFPKLi_mLoPMbnq76E4K1GG3xxha8VYJC2zI,356
-blue_assistant/.abcli/script/run.sh,sha256=015jWqKd5MFjW4KVuS2ZwW62uvutXkOEYIeQ6FwBtyI,695
+blue_assistant/.abcli/script/run.sh,sha256=kSXmyM9NUj2X2orSGyu5t_P5frG-gyumbRq-xqF692c,911
 blue_assistant/.abcli/tests/README.sh,sha256=Qs0YUxVB1OZZ70Nqw2kT1LKXeUnC5-XfQRMfqb8Cbwg,152
 blue_assistant/.abcli/tests/help.sh,sha256=bxocRb83kKjrA7eeCmI-UaAlVPJZ_xfKy-woJciig_Y,689
-blue_assistant/.abcli/tests/script_run.sh,sha256=xta0oxjIL5fmJ6vZzvWU1O2i_4AdBPNUGYHZPPnlYi4,235
+blue_assistant/.abcli/tests/script_run.sh,sha256=UXOvYOBe2FcqWywZvjyazDly6j9_GYAwuxLmmM4GLMM,725
 blue_assistant/.abcli/tests/version.sh,sha256=oR2rvYR8zi-0VDPIdPJsmsmWwYaamT8dmNTqUh3-8Gw,154
 blue_assistant/help/__init__.py,sha256=ajz1GSNU9xYVrFEDSz6Xwg7amWQ_yvW75tQa1ZvRIWc,3
 blue_assistant/help/__main__.py,sha256=cVejR7OpoWPg0qLbm-PZf5TuJS27x49jzfiyCLyzEns,241
 blue_assistant/help/functions.py,sha256=9WsmXGMN-R7sqlkGLK0nY90Peg8Gah4rIu75QbLhImo,689
-blue_assistant/help/script.py,sha256=0uPtQYrAqCebPGrVdwuC17VHCQh2fQldBfM3fPHx2eU,507
+blue_assistant/help/script.py,sha256=knI6aNntFh8OxXuIaWYgcswLqHXz6_cppTvClvuSK-I,740
 blue_assistant/script/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-blue_assistant/script/__main__.py,sha256=JD-UxKck9Dcu8EUxw-AZcgvSN9Z9ouEkCAIe7GbHOZ4,806
-blue_assistant/script/blue_amo.py,sha256=DPMJDeS_HwkpwhQf_fp3j3iO7oMQesmYXArOaRdNOX8,447
-blue_assistant/script/functions.py,sha256=pZ_b3a0fPQm5S5eIn7B2ckVW0YPRrGFbjh6zBRmqXYc,1026
-blue_assistant/script/generic.py,sha256=nw0FJeeT0H3DcsnAr8OYqehNNPU5CL3V4phvcvJRFWQ,899
-blue_assistant/script/kinds.py,sha256=jKrekJPbWVOJBOqjHCCOfFfQ_dEv4ElI-0LFwShH4Jc,97
-blue_assistant-4.32.1.dist-info/LICENSE,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
-blue_assistant-4.32.1.dist-info/METADATA,sha256=tYgoZgcoGOV1hVY3GuYDFR3WI3V82_cfCTtUsw9fAT8,2349
-blue_assistant-4.32.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-blue_assistant-4.32.1.dist-info/top_level.txt,sha256=ud0BkBbdOVze13bNqHuhZj1rwCztaBtDf5ChEYzASOs,15
-blue_assistant-4.32.1.dist-info/RECORD,,
+blue_assistant/script/__main__.py,sha256=LAbStVfkw1-kCa7GcutCmimwLB37dTx1hG4LNU-6hXg,1229
+blue_assistant/script/load.py,sha256=9hzXmHzJ9t8VcWna-oZbo7GWYDUnHou5LUMWD9l0HmA,555
+blue_assistant/script/repository/__init__.py,sha256=iuiwMKpqTjL4iaVnj4bS5Nev_MgiWKJmeJgZFlM5-lU,483
+blue_assistant/script/repository/blue_amo/__init__.py,sha256=WjL9GIlN-DBnbUMJ8O_FxTp0rcVGlsIS3H9YtXEefTk,76
+blue_assistant/script/repository/blue_amo/classes.py,sha256=yhCeP1vHUUQe_iyUzRFcpiELnLi1Bo5qPnVh3J1QvM8,1043
+blue_assistant/script/repository/generic/__init__.py,sha256=kLffGsQMQAFJTw6IZBE5eBxvshP1x9wwHHR4hsDJblo,75
+blue_assistant/script/repository/generic/classes.py,sha256=eQl4guWRy0_Xy5FJpmdY740XPkgx4ShYIZ6Jhz0lfiw,1124
+blue_assistant/script/repository/moon_datasets/__init__.py,sha256=aCtmP2avh3yKAJ668S3GsLR9vbBOm5zt9FSFCqy_tAs,86
+blue_assistant/script/repository/moon_datasets/classes.py,sha256=9LFbLAo06Jux9lWBRNOv_8QaXznPgqTqxmcZBWaWPDI,357
+blue_assistant-4.43.1.dist-info/LICENSE,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
+blue_assistant-4.43.1.dist-info/METADATA,sha256=iPMa_AwFKj5RGpWJKeC4mkqjhw69CI1z9bqC-QoWPNk,2438
+blue_assistant-4.43.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+blue_assistant-4.43.1.dist-info/top_level.txt,sha256=ud0BkBbdOVze13bNqHuhZj1rwCztaBtDf5ChEYzASOs,15
+blue_assistant-4.43.1.dist-info/RECORD,,

blue_assistant/script/blue_amo.py

@@ -1,18 +0,0 @@
-from blue_assistant.script.generic import GenericScript
-from blue_assistant.script.kinds import ScriptKind
-
-
-class BlueAmoScript(GenericScript):
-    def __init__(
-        self,
-        object_name: str,
-        verbose: bool = False,
-        download: bool = False,
-    ):
-        super().__init__(
-            object_name=object_name,
-            verbose=verbose,
-            download=download,
-        )
-
-        self.kind = ScriptKind.BLUE_AMO

blue_assistant/script/functions.py

@@ -1,40 +0,0 @@
-from typing import Tuple, Union, Type
-
-from blueness import module
-from blue_objects.metadata import get_from_object
-
-from blue_assistant import NAME
-from blue_assistant.script.generic import GenericScript
-from blue_assistant.script.blue_amo import BlueAmoScript
-from blue_assistant.logger import logger
-
-
-NAME = module.name(__file__, NAME)
-
-
-def load_script(
-    object_name: str,
-    verbose: bool = False,
-    download: bool = False,
-) -> Tuple[bool, Union[GenericScript, None]]:
-    script_kind = get_from_object(
-        object_name=object_name,
-        key="script.kind",
-        default="generic",
-    )
-
-    script_class: Type[GenericScript] = GenericScript
-    if script_kind == "blue_amo":
-        script_class = BlueAmoScript
-    if verbose:
-        logger.info(f"script class: {script_class.__name__}")
-
-    script = script_class(
-        object_name=object_name,
-        verbose=verbose,
-        download=download,
-    )
-    if verbose:
-        logger.info(f"script kind: {script.kind}")
-
-    return True, script

blue_assistant/script/generic.py

@@ -1,41 +0,0 @@
-from blueness import module
-from blue_objects.metadata import get_from_object
-
-from blue_assistant import NAME
-from blue_assistant.script.kinds import ScriptKind
-from blue_assistant.logger import logger
-
-
-NAME = module.name(__file__, NAME)
-
-
-class GenericScript:
-    def __init__(
-        self,
-        object_name: str,
-        verbose: bool = False,
-        download: bool = False,
-    ):
-        self.verbose = verbose
-
-        self.object_name = object_name
-
-        self.kind = ScriptKind.GENERIC
-
-        self.script = get_from_object(
-            object_name=object_name,
-            key="script",
-            default={},
-            download=download,
-        )
-
-    def run(self) -> bool:
-        logger.info(
-            "{}.run({} @ {})".format(
-                NAME,
-                self.__class__.__name__,
-                self.object_name,
-            )
-        )
-
-        return True

blue_assistant/script/kinds.py

@@ -1,6 +0,0 @@
-from enum import Enum, auto
-
-
-class ScriptKind(Enum):
-    GENERIC = auto()
-    BLUE_AMO = auto()