outerbounds 0.3.151__py3-none-any.whl → 0.3.152__py3-none-any.whl

outerbounds/command_groups/apps_cli.py

@@ -4,6 +4,8 @@ from outerbounds._vendor import click
 import requests
 import time
 import random
+import shutil
+import subprocess
 
 from ..utils import metaflowconfig
 
@@ -113,7 +115,12 @@ def start(config_dir=None, profile=None, port=-1, name=""):
                                     err=True,
                                 )
                                 click.secho(
-                                    f"App URL: {api_url.replace('api', 'ui')}/apps/{workstation_id}/{name}/",
+                                    f"Browser URL: {api_url.replace('api', 'ui')}/apps/{workstation_id}/{name}/",
+                                    fg="green",
+                                    err=True,
+                                )
+                                click.secho(
+                                    f"App URL: {api_url}/apps/{workstation_id}/{name}/",
                                     fg="green",
                                     err=True,
                                 )
@@ -145,7 +152,12 @@ def start(config_dir=None, profile=None, port=-1, name=""):
                                             err=True,
                                         )
                                         click.secho(
-                                            f"App URL: {api_url.replace('api', 'ui')}/apps/{os.environ['WORKSTATION_ID']}/{name}/",
+                                            f"Browser URL: {api_url.replace('api', 'ui')}/apps/{os.environ['WORKSTATION_ID']}/{name}/",
+                                            fg="green",
+                                            err=True,
+                                        )
+                                        click.secho(
+                                            f"App URL: {api_url}/apps/{os.environ['WORKSTATION_ID']}/{name}/",
                                             fg="green",
                                             err=True,
                                         )
@@ -301,6 +313,93 @@ def stop(config_dir=None, profile=None, port=-1, name=""):
         click.secho(f"Failed to stop app on port {port}!", fg="red", err=True)
 
 
+@app.command(help="Kill the process associated with an app.")
+@click.option(
+    "-d",
+    "--config-dir",
+    default=path.expanduser(os.environ.get("METAFLOW_HOME", "~/.metaflowconfig")),
+    help="Path to Metaflow configuration directory",
+    show_default=True,
+)
+@click.option(
+    "-p",
+    "--profile",
+    default=os.environ.get("METAFLOW_PROFILE", ""),
+    help="The named metaflow profile in which your workstation exists",
+)
+@click.option(
+    "--name",
+    required=False,
+    help="Name of your app",
+    default="",
+    type=str,
+)
+def kill_process(config_dir=None, profile=None, port=-1, name=""):
+    if port > 0 and name:
+        click.secho(
+            "Please provide either a port number or a name to stop the app, not both.",
+            fg="red",
+            err=True,
+        )
+        return
+
+    if "WORKSTATION_ID" not in os.environ:
+        click.secho(
+            "All outerbounds app commands can only be run from a workstation.",
+            fg="red",
+            err=True,
+        )
+
+        return
+
+    supervisorctl_exists = shutil.which("supervisorctl")
+    if not supervisorctl_exists:
+        click.secho(
+            "This workstation does not support automated app deployment and management. Pleasr reach out to outerbounds support!",
+            fg="red",
+            err=True,
+        )
+        return
+
+    supervisorctl_command = ["supervisorctl", "stop", name]
+
+    # This is somewhat ugly way of doing this, but there aren't any better ways.
+    # Since we use supervisorctl to manage our apps, we have to rely on external calls to
+    # kill the process.
+    result = subprocess.run(supervisorctl_command, capture_output=True, text=True)
+    if result.returncode == 0:
+        # It gets uglier.
+        # When the process is not running, which is likely due to a user error, the output looks like:
+        # myapp: ERROR (not running)
+        # but the return code is still 0.
+        # If the stop is successful, the output looks like:
+        # myapp: stopped
+
+        if result.stdout.startswith(f"{name}: ERROR (not running)"):
+            click.secho(
+                f"Process {name} is not in a running state to kill!",
+                fg="yellow",
+                err=True,
+            )
+            return
+        elif result.stdout.startswith(f"{name}: stopped"):
+            click.secho(f"Process {name} killed successfully!", fg="green", err=True)
+            return
+        else:
+            click.secho(f"Process {name} stopped!", fg="red", err=True)
+            return
+    elif result.returncode == 1:
+        # One of the cases where this can happen is if the app name itself is wrong.
+        # In this case, the output looks like:
+        # 'mya111pp: ERROR (no such process)\n'
+        if result.stdout.startswith(f"{name}: ERROR (no such process)"):
+            click.secho(f"Process {name} does not exist!", fg="red", err=True)
+            return
+        else:
+            click.secho(f"Failed to kill process {name}!", fg="red", err=True)
+            return
+
+
 @app.command(help="List all apps on the workstation")
 @click.option(
     "-d",

{outerbounds-0.3.151.dist-info → outerbounds-0.3.152.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: outerbounds
-Version: 0.3.151
+Version: 0.3.152
 Summary: More Data Science, Less Administration
 License: Proprietary
 Keywords: data science,machine learning,MLOps
@@ -27,9 +27,9 @@ Requires-Dist: google-auth (>=2.27.0,<3.0.0) ; extra == "gcp"
 Requires-Dist: google-cloud-secret-manager (>=2.20.0,<3.0.0) ; extra == "gcp"
 Requires-Dist: google-cloud-storage (>=2.14.0,<3.0.0) ; extra == "gcp"
 Requires-Dist: metaflow-checkpoint (==0.1.9)
-Requires-Dist: ob-metaflow (==2.15.6.1)
-Requires-Dist: ob-metaflow-extensions (==1.1.138)
-Requires-Dist: ob-metaflow-stubs (==6.0.3.151)
+Requires-Dist: ob-metaflow (==2.15.7.1)
+Requires-Dist: ob-metaflow-extensions (==1.1.140)
+Requires-Dist: ob-metaflow-stubs (==6.0.3.152)
 Requires-Dist: opentelemetry-distro (==0.41b0)
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (==1.20.0)
 Requires-Dist: opentelemetry-instrumentation-requests (==0.41b0)

{outerbounds-0.3.151.dist-info → outerbounds-0.3.152.dist-info}/RECORD

@@ -41,7 +41,7 @@ outerbounds/_vendor/yaml/serializer.py,sha256=8wFZRy9SsQSktF_f9OOroroqsh4qVUe53r
 outerbounds/_vendor/yaml/tokens.py,sha256=JBSu38wihGr4l73JwbfMA7Ks1-X84g8-NskTz7KwPmA,2578
 outerbounds/cli_main.py,sha256=e9UMnPysmc7gbrimq2I4KfltggyU7pw59Cn9aEguVcU,74
 outerbounds/command_groups/__init__.py,sha256=QPWtj5wDRTINDxVUL7XPqG3HoxHNvYOg08EnuSZB2Hc,21
-outerbounds/command_groups/apps_cli.py,sha256=8jmQufa0bK2sfRfs7DiWjoJ1oWiqZAixsL4Dte_KY4Y,17201
+outerbounds/command_groups/apps_cli.py,sha256=v9OlQ1b4BGB-cBZiHB6W5gDocDoMmrQ7zdK11QiJ-B8,20847
 outerbounds/command_groups/cli.py,sha256=YsDS-AKhhwwSH65mgNsmjeg1U2bk202V5NpIYnXEdb8,440
 outerbounds/command_groups/fast_bakery_cli.py,sha256=5kja7v6C651XAY6dsP_IkBPJQgfU4hA4S9yTOiVPhW0,6213
 outerbounds/command_groups/local_setup_cli.py,sha256=tuuqJRXQ_guEwOuQSIf9wkUU0yg8yAs31myGViAK15s,36364
@@ -54,7 +54,7 @@ outerbounds/utils/metaflowconfig.py,sha256=l2vJbgPkLISU-XPGZFaC8ZKmYFyJemlD6bwB-
 outerbounds/utils/schema.py,sha256=lMUr9kNgn9wy-sO_t_Tlxmbt63yLeN4b0xQXbDUDj4A,2331
 outerbounds/utils/utils.py,sha256=4Z8cszNob_8kDYCLNTrP-wWads_S_MdL3Uj3ju4mEsk,501
 outerbounds/vendor.py,sha256=gRLRJNXtZBeUpPEog0LOeIsl6GosaFFbCxUvR4bW6IQ,5093
-outerbounds-0.3.151.dist-info/METADATA,sha256=HJd3SOqWMsX2oJ28gzynFOj-0QSTSeVDR076OpX9HfY,1761
-outerbounds-0.3.151.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-outerbounds-0.3.151.dist-info/entry_points.txt,sha256=7ye0281PKlvqxu15rjw60zKg2pMsXI49_A8BmGqIqBw,47
-outerbounds-0.3.151.dist-info/RECORD,,
+outerbounds-0.3.152.dist-info/METADATA,sha256=fNqe7vT9wc3QsEGzA5HiOYvCqTtBvu2BNgPjfAyABEA,1761
+outerbounds-0.3.152.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+outerbounds-0.3.152.dist-info/entry_points.txt,sha256=7ye0281PKlvqxu15rjw60zKg2pMsXI49_A8BmGqIqBw,47
+outerbounds-0.3.152.dist-info/RECORD,,