bluer-objects 6.167.1__tar.gz → 6.172.1__tar.gz

{bluer_objects-6.167.1/bluer_objects.egg-info → bluer_objects-6.172.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bluer_objects
-Version: 6.167.1
+Version: 6.172.1
 Summary: 🌀 Object management in Bash.
 Home-page: https://github.com/kamangir/bluer-objects
 Author: Arash Abadpour (Kamangir)
@@ -64,6 +64,6 @@ pip install bluer-objects
 
 [![pylint](https://github.com/kamangir/bluer-objects/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/bluer-objects/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/bluer-objects/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/bluer-objects.svg)](https://pypi.org/project/bluer-objects/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/bluer-objects)](https://pypistats.org/packages/bluer-objects)
 
-built by 🌀 [`bluer README`](https://github.com/kamangir/bluer-objects/tree/main/bluer_objects/README), based on 🌀 [`bluer_objects-6.167.1`](https://github.com/kamangir/bluer-objects).
+built by 🌀 [`bluer README`](https://github.com/kamangir/bluer-objects/tree/main/bluer_objects/README), based on 🌀 [`bluer_objects-6.172.1`](https://github.com/kamangir/bluer-objects).
 
 built by 🌀 [`blueness-3.118.1`](https://github.com/kamangir/blueness).

{bluer_objects-6.167.1 → bluer_objects-6.172.1}/README.md

@@ -30,4 +30,4 @@ pip install bluer-objects
 
 [![pylint](https://github.com/kamangir/bluer-objects/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/bluer-objects/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/bluer-objects/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/bluer-objects.svg)](https://pypi.org/project/bluer-objects/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/bluer-objects)](https://pypistats.org/packages/bluer-objects)
 
-built by 🌀 [`bluer README`](https://github.com/kamangir/bluer-objects/tree/main/bluer_objects/README), based on 🌀 [`bluer_objects-6.167.1`](https://github.com/kamangir/bluer-objects).
+built by 🌀 [`bluer README`](https://github.com/kamangir/bluer-objects/tree/main/bluer_objects/README), based on 🌀 [`bluer_objects-6.172.1`](https://github.com/kamangir/bluer-objects).

{bluer_objects-6.167.1 → bluer_objects-6.172.1}/bluer_objects/.abcli/mlflow/deploy.sh

@@ -2,15 +2,28 @@
 
 function bluer_objects_mlflow_deploy() {
     local options=$1
-    local do_dryrun=$(bluer_ai_option_int "$options" dryrun 0)
-    local do_local=$(bluer_ai_option_int "$options" local 1)
-    local port=$(bluer_ai_option "$options" port 5001)
+    local do_set=$(bluer_ai_option_int "$options" set 0)
+
+    if [[ "$do_set" == 1 ]]; then
+        local url=$2
+        if [[ -z "$url" ]]; then
+            bluer_ai_log_error "url not found."
+            return 1
+        fi
+
+        pushd $abcli_path_git/bluer-objects >/dev/null
+        dotenv set \
+            MLFLOW_DEPLOYMENT \
+            $url
+        popd >/dev/null
 
-    if [[ "$do_local" == 0 ]]; then
-        bluer_ai_log_error "only local deployment is supported."
-        return 1
+        bluer_objects init
+        return
     fi
 
+    local do_dryrun=$(bluer_ai_option_int "$options" dryrun 0)
+    local port=$(bluer_ai_option "$options" port 5001)
+
     [[ "$MLFLOW_DEPLOYMENT" != "local" ]] &&
         bluer_ai_log_warning "MLFLOW_DEPLOYMENT is not local".
 

{bluer_objects-6.167.1 → bluer_objects-6.172.1}/bluer_objects/.abcli/tests/help.sh

@@ -49,6 +49,7 @@ function test_bluer_objects_help() {
         "@mlflow browse" \
         "@mlflow cache" \
         "@mlflow deploy" \
+        "@mlflow deploy set" \
         "@mlflow get_id" \
         "@mlflow get_run_id" \
         "@mlflow list_registered_models" \

{bluer_objects-6.167.1 → bluer_objects-6.172.1}/bluer_objects/__init__.py

@@ -4,7 +4,7 @@ ICON = "🌀"
 
 DESCRIPTION = f"{ICON} Object management in Bash."
 
-VERSION = "6.167.1"
+VERSION = "6.172.1"
 
 REPO_NAME = "bluer-objects"
 

{bluer_objects-6.167.1 → bluer_objects-6.172.1}/bluer_objects/help/mlflow/__init__.py

@@ -30,7 +30,7 @@ def help_deploy(
     tokens: List[str],
     mono: bool,
 ) -> str:
-    options = xtra("dryrun,~local,port=<5001>", mono=mono)
+    options = xtra("dryrun,port=<5001>", mono=mono)
 
     return show_usage(
         [
@@ -43,6 +43,22 @@ def help_deploy(
     )
 
 
+def help_deploy_set(
+    tokens: List[str],
+    mono: bool,
+) -> str:
+    return show_usage(
+        [
+            "@mlflow",
+            "deploy",
+            "set",
+            "<url>",
+        ],
+        "set mlflow deployment.",
+        mono=mono,
+    )
+
+
 def help_get_id(
     tokens: List[str],
     mono: bool,
@@ -199,7 +215,10 @@ def help_transition(
 help_functions = {
     "browse": help_browse,
     "cache": help_cache,
-    "deploy": help_deploy,
+    "deploy": {
+        "": help_deploy,
+        "set": help_deploy_set,
+    },
     "get_id": help_get_id,
     "get_run_id": help_get_run_id,
     "list_registered_models": help_list_registered_models,

{bluer_objects-6.167.1 → bluer_objects-6.172.1}/bluer_objects/storage/s3.py

@@ -16,6 +16,104 @@ from bluer_objects.logger import logger
 class S3Interface(StorageInterface):
     name = "s3"
 
+    def clear(
+        self,
+        do_dryrun: bool = True,
+        log: bool = True,
+    ) -> bool:
+        logger.info(
+            "{}.clear({})".format(
+                self.__class__.__name__,
+                "dryrun" if do_dryrun else "",
+            )
+        )
+
+        try:
+            s3 = boto3.client(
+                "s3",
+                endpoint_url=env.S3_STORAGE_ENDPOINT_URL,
+                aws_access_key_id=env.S3_STORAGE_AWS_ACCESS_KEY_ID,
+                aws_secret_access_key=env.S3_STORAGE_AWS_SECRET_ACCESS_KEY,
+            )
+
+            paginator = s3.get_paginator("list_objects_v2")
+            pages = paginator.paginate(
+                Bucket=env.S3_STORAGE_BUCKET,
+                Prefix="test",
+            )
+        except Exception as e:
+            logger.error(e)
+            return False
+
+        list_of_objects = sorted(
+            list(
+                set(
+                    reduce(
+                        lambda x, y: x + y,
+                        [
+                            [
+                                obj["Key"].split("/", 1)[0]
+                                for obj in page.get("Contents", [])
+                            ]
+                            for page in pages
+                        ],
+                        [],
+                    )
+                )
+            )
+        )
+
+        logger.info(f"{len(list_of_objects)} object(s) to delete.")
+
+        for object_name in tqdm(list_of_objects):
+            if not self.delete(
+                object_name=object_name,
+                do_dryrun=do_dryrun,
+            ):
+                return False
+
+        return True
+
+    def delete(
+        self,
+        object_name: str,
+        do_dryrun: bool = True,
+        log: bool = True,
+    ) -> bool:
+        if log:
+            logger.info(
+                "{}.delete({}){}".format(
+                    self.__class__.__name__,
+                    object_name,
+                    " dryrun" if do_dryrun else "",
+                )
+            )
+        if do_dryrun:
+            return True
+
+        try:
+            s3 = boto3.resource(
+                "s3",
+                endpoint_url=env.S3_STORAGE_ENDPOINT_URL,
+                aws_access_key_id=env.S3_STORAGE_AWS_ACCESS_KEY_ID,
+                aws_secret_access_key=env.S3_STORAGE_AWS_SECRET_ACCESS_KEY,
+            )
+            bucket = s3.Bucket(env.S3_STORAGE_BUCKET)
+
+            objects_to_delete = bucket.objects.filter(Prefix=f"{object_name}/")
+            delete_requests = [{"Key": obj.key} for obj in objects_to_delete]
+
+            if not delete_requests:
+                logger.warning(f"no files found under {object_name}.")
+                return True
+
+            bucket.delete_objects(Delete={"Objects": delete_requests})
+        except Exception as e:
+            logger.error(e)
+            return False
+
+        return True
+
     def download(
         self,
         object_name: str,

{bluer_objects-6.167.1 → bluer_objects-6.172.1/bluer_objects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bluer_objects
-Version: 6.167.1
+Version: 6.172.1
 Summary: 🌀 Object management in Bash.
 Home-page: https://github.com/kamangir/bluer-objects
 Author: Arash Abadpour (Kamangir)
@@ -64,6 +64,6 @@ pip install bluer-objects
 
 [![pylint](https://github.com/kamangir/bluer-objects/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/bluer-objects/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/bluer-objects/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/bluer-objects/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/bluer-objects.svg)](https://pypi.org/project/bluer-objects/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/bluer-objects)](https://pypistats.org/packages/bluer-objects)
 
-built by 🌀 [`bluer README`](https://github.com/kamangir/bluer-objects/tree/main/bluer_objects/README), based on 🌀 [`bluer_objects-6.167.1`](https://github.com/kamangir/bluer-objects).
+built by 🌀 [`bluer README`](https://github.com/kamangir/bluer-objects/tree/main/bluer_objects/README), based on 🌀 [`bluer_objects-6.172.1`](https://github.com/kamangir/bluer-objects).
 
 built by 🌀 [`blueness-3.118.1`](https://github.com/kamangir/blueness).