outerbounds 0.3.166__py3-none-any.whl → 0.3.168rc0__py3-none-any.whl

outerbounds/command_groups/cli.py

@@ -6,6 +6,7 @@ from . import (
     apps_cli,
     tutorials_cli,
     fast_bakery_cli,
+    secrets_cli,
 )
 
 
@@ -18,6 +19,7 @@ from . import (
         apps_cli.cli,
         tutorials_cli.cli,
         fast_bakery_cli.cli,
+        secrets_cli.cli,
     ],
 )
 def cli(**kwargs):

outerbounds/command_groups/secrets_cli.py

@@ -0,0 +1,119 @@
+import json
+import os
+from os import path
+from typing import Any, Dict
+from outerbounds._vendor import click
+
+from ..utils.utils import safe_write_to_disk
+
+
+def _get_ob_provider():
+    from metaflow.plugins import SECRETS_PROVIDERS
+
+    ob_provider = [i for i in SECRETS_PROVIDERS if i.TYPE == "outerbounds"][0]
+    return ob_provider()
+
+
+class SecretOutput:
+    def __init__(self, type):
+        self.type = type
+        self._secrets = []
+
+    def consume(self, secret_dict):
+        self._secrets.append(secret_dict)
+
+    def parse(self):
+        if self.type == "text":
+            return "\n".join(
+                [f"{k}={v}" for secret in self._secrets for k, v in secret.items()]
+            )
+        elif self.type == "json":
+            secrets = {}
+            for secret in self._secrets:
+                secrets.update(secret)
+            return json.dumps(secrets)
+        elif self.type == "shell":
+            return "\n".join(
+                [
+                    f"export {k}={v}"
+                    for secret in self._secrets
+                    for k, v in secret.items()
+                ]
+            )
+
+
+@click.group()
+def cli(**kwargs):
+    pass
+
+
+@click.group(help="Manage secrets")
+def secrets(**kwargs):
+    pass
+
+
+FORMAT_HELP = """
+Format of the output. When you specify:
+    - `text`: The output will be in `key=value` format. All secrets will be printed on new lines.
+    - `json`: The output will be in JSON format. All secrets will be combined into a single dictionary so if secrets have overlapping keys, the last one will take precedence.
+    - `shell`: The output will be a shell script that exports all secrets as environment variables.
+"""
+
+
+@secrets.command(
+    help="Get secrets. The `secret_ids` correspond to the name of the integrations configured on the Outerbounds platform."
+)
+# command should be like :
+# `outerbounds secrets get <secret-id-0> <secret-id-1> ... --output [text|json] --config-dir <config-dir> --profile <profile> --role <role> --output-file <output-file>`
+@click.argument("secret_ids", type=str, nargs=-1)
+@click.option(
+    "--format",
+    type=click.Choice(["text", "json", "shell"]),
+    default="text",
+    help=FORMAT_HELP,
+)
+@click.option(
+    "--profile",
+    type=str,
+    default=None,
+    help="The named metaflow profile to use.",
+)
+@click.option(
+    "--role",
+    type=str,
+    default=None,
+    help="Any additional IAM role required to access the secrets.",
+)
+@click.option(
+    "--file",
+    "-f",
+    "output_file",
+    type=str,
+    default=None,
+    help="The file to write the output to. If not specified, the output will be printed to standard out.",
+)
+def get(
+    secret_ids,
+    format,
+    profile,
+    role,
+    output_file,
+):
+
+    if profile != "" and profile is not None:
+        os.environ["METAFLOW_PROFILE"] = profile
+
+    provider = _get_ob_provider()
+    output = SecretOutput(format)
+    for secret_id in secret_ids:
+        secret = provider.get_secret_as_dict(secret_id, role=role)
+        output.consume(secret)
+    output_str = output.parse()
+    if output_file:
+        output_file = os.path.abspath(output_file)
+        safe_write_to_disk(output_file, output_str)
+    else:
+        print(output_str)
+
+
+cli.add_command(secrets, name="secrets")

{outerbounds-0.3.166.dist-info → outerbounds-0.3.168rc0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: outerbounds
-Version: 0.3.166
+Version: 0.3.168rc0
 Summary: More Data Science, Less Administration
 License: Proprietary
 Keywords: data science,machine learning,MLOps
@@ -29,8 +29,8 @@ 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.2.1)
 Requires-Dist: ob-metaflow (==2.15.11.2)
-Requires-Dist: ob-metaflow-extensions (==1.1.153)
-Requires-Dist: ob-metaflow-stubs (==6.0.3.166)
+Requires-Dist: ob-metaflow-extensions (==1.1.155rc0)
+Requires-Dist: ob-metaflow-stubs (==6.0.3.168rc0)
 Requires-Dist: opentelemetry-distro (>=0.41b0) ; extra == "otel"
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.20.0) ; extra == "otel"
 Requires-Dist: opentelemetry-instrumentation-requests (>=0.41b0) ; extra == "otel"

{outerbounds-0.3.166.dist-info → outerbounds-0.3.168rc0.dist-info}/RECORD

@@ -42,10 +42,11 @@ outerbounds/_vendor/yaml/tokens.py,sha256=JBSu38wihGr4l73JwbfMA7Ks1-X84g8-NskTz7
 outerbounds/cli_main.py,sha256=e9UMnPysmc7gbrimq2I4KfltggyU7pw59Cn9aEguVcU,74
 outerbounds/command_groups/__init__.py,sha256=QPWtj5wDRTINDxVUL7XPqG3HoxHNvYOg08EnuSZB2Hc,21
 outerbounds/command_groups/apps_cli.py,sha256=v9OlQ1b4BGB-cBZiHB6W5gDocDoMmrQ7zdK11QiJ-B8,20847
-outerbounds/command_groups/cli.py,sha256=YsDS-AKhhwwSH65mgNsmjeg1U2bk202V5NpIYnXEdb8,440
+outerbounds/command_groups/cli.py,sha256=de4_QY1UeoKX6y-IXIbmklAi6bz0DsdBSmAoCg6lq1o,482
 outerbounds/command_groups/fast_bakery_cli.py,sha256=5kja7v6C651XAY6dsP_IkBPJQgfU4hA4S9yTOiVPhW0,6213
 outerbounds/command_groups/local_setup_cli.py,sha256=tuuqJRXQ_guEwOuQSIf9wkUU0yg8yAs31myGViAK15s,36364
 outerbounds/command_groups/perimeters_cli.py,sha256=iF_Uw7ROiSctf6FgoJEy30iDBLVE1j9FKuR3shgJRmc,19050
+outerbounds/command_groups/secrets_cli.py,sha256=PRV_pV1JEGkgSKQop6Qq_W4o2lcK3b7rNH2P-S8nY-8,3250
 outerbounds/command_groups/tutorials_cli.py,sha256=UInFyiMqtscHFfi8YQwiY_6Sdw9quJOtRu5OukEBccw,3522
 outerbounds/command_groups/workstations_cli.py,sha256=V5Jbj1cVb4IRllI7fOgNgL6OekRpuFDv6CEhDb4xC6w,22016
 outerbounds/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -54,7 +55,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.166.dist-info/METADATA,sha256=TZ52Cxpwyuzj6z8TWn-wJa--mYiy7tpSWX_AkOOLoXQ,1837
-outerbounds-0.3.166.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-outerbounds-0.3.166.dist-info/entry_points.txt,sha256=7ye0281PKlvqxu15rjw60zKg2pMsXI49_A8BmGqIqBw,47
-outerbounds-0.3.166.dist-info/RECORD,,
+outerbounds-0.3.168rc0.dist-info/METADATA,sha256=cQyn3eWluj6w2YUCcpNTWXr6vZk7DncpRZu4LM6wvAU,1846
+outerbounds-0.3.168rc0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+outerbounds-0.3.168rc0.dist-info/entry_points.txt,sha256=7ye0281PKlvqxu15rjw60zKg2pMsXI49_A8BmGqIqBw,47
+outerbounds-0.3.168rc0.dist-info/RECORD,,