outerbounds 0.3.20rc0__py3-none-any.whl → 0.3.21__py3-none-any.whl

outerbounds/__init__.py

@@ -5,16 +5,10 @@ from pathlib import Path
 import json
 import os
 import re
-import string
 import sys
 import zlib
+
 import click
-from kubeconfig import KubeConfig
-import yaml
-import requests
-import base64
-import datetime
-import hashlib
 
 
 class CheckException(Exception):
@@ -397,282 +391,3 @@ def configure(encoded_config=None, config_dir=None, profile=None, echo=None):
     except Exception as e:
         click.secho("Writing the configuration file '{}' failed.".format(writer.path()))
         click.secho("Error: {}".format(str(e)))
-
-
-def get_k8s_token_response(config_dir, profile):
-    config_path = path.join(config_dir, "config.json")
-    if profile != "":
-        config_path = path.join(config_dir, "config_{}.json".format(profile))
-    with open(config_path) as json_file:
-        config = json.load(json_file)
-        metaflow_token = config["METAFLOW_SERVICE_AUTH_KEY"]
-
-        if config["OBP_AUTH_SERVER"].startswith("https://"):
-            auth_endpoint = config["OBP_AUTH_SERVER"]
-        else:
-            auth_endpoint = "https://{}".format(config["OBP_AUTH_SERVER"])
-
-        if auth_endpoint.endswith("/"):
-            auth_endpoint = auth_endpoint[:-1]
-
-        generate_token_url = "{}/generate/k8s".format(auth_endpoint)
-        headers = {"x-api-key": metaflow_token}
-        response = requests.get(generate_token_url, headers=headers)
-        if response.status_code == 200:
-            return json.loads(response.content)
-        else:
-            raise Exception(
-                "Failed to get response. Error: {}".format(response.status_code)
-            )
-
-
-@cli.command(help="Generate a token to use your cloud workstation")
-@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="",
-    help="The named metaflow profile in which your workstation exists",
-)
-def generate_workstation_token(config_dir=None, profile=None):
-    try:
-        token = get_k8s_token_response(config_dir, profile)["token"]
-        token_data = base64.b64decode(token.split(".")[1] + "==")
-        exec_creds = {
-            "kind": "ExecCredential",
-            "apiVersion": "client.authentication.k8s.io/v1beta1",
-            "spec": {},
-            "status": {
-                "token": token,
-                "expirationTimestamp": datetime.datetime.fromtimestamp(
-                    json.loads(token_data)["exp"], datetime.timezone.utc
-                ).isoformat(),
-            },
-        }
-        click.echo(json.dumps(exec_creds))
-    except Exception as e:
-        click.secho("Failed to generate workstation token.", fg="red")
-        click.secho("Error: {}".format(str(e)))
-
-
-@cli.command(help="Configure a cloud workstation")
-@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="",
-    help="The named metaflow profile in which your workstation exists",
-)
-def configure_cloud_workstation(config_dir=None, profile=None):
-    k8s_response = get_k8s_token_response(config_dir, profile)
-    token_data = base64.b64decode(k8s_response["token"].split(".")[1] + "==")
-    ws_namespace = "ws-{}".format(
-        hashlib.md5(bytes(json.loads(token_data)["username"], "utf-8")).hexdigest()
-    )
-
-    kube_config_path = path.expanduser(os.environ.get("KUBECONFIG", "~/.kube/config"))
-    kube_config = KubeConfig(kube_config_path)
-    kube_config.set_context(
-        "outerbounds-workstations", "outerbounds-cluster", ws_namespace, "obp-user"
-    )
-
-    kube_config.set_cluster(
-        "outerbounds-cluster",
-        server=k8s_response["endpoint"],
-        insecure_skip_tls_verify=True,
-    )
-
-    with open(kube_config_path, "r") as f:
-        kube_yaml = yaml.safe_load(f)
-
-    gen_creds_args = ["generate-workstation-token"]
-    if profile != "":
-        gen_creds_args.append("--profile")
-        gen_creds_args.append(profile)
-
-    gen_creds_args.append("--config-dir")
-    gen_creds_args.append(config_dir)
-
-    user_exec_creds = {
-        "exec": {
-            "apiVersion": "client.authentication.k8s.io/v1beta1",
-            "command": "outerbounds",
-            "args": gen_creds_args,
-            "env": None,
-            "interactiveMode": "Never",
-            "provideClusterInfo": False,
-        }
-    }
-
-    user_updated = False
-
-    if kube_yaml["users"] is None:
-        kube_yaml["users"] = []
-
-    for user in kube_yaml["users"]:
-        if user["name"] == "obp-user":
-            user["user"] = user_exec_creds
-            user_updated = True
-
-    if not user_updated:
-        kube_yaml["users"].append({"name": "obp-user", "user": user_exec_creds})
-
-    with open(kube_config_path, "w") as f:
-        yaml.safe_dump(kube_yaml, f)
-
-    kube_config.use_context("outerbounds-workstations")
-
-
-def prepare_workstations_api_request(config_dir, profile, api_path):
-    config_path = path.join(config_dir, "config.json")
-    if profile != "":
-        config_path = path.join(config_dir, "config_{}.json".format(profile))
-    with open(config_path) as json_file:
-        config = json.load(json_file)
-        metaflow_token = config["METAFLOW_SERVICE_AUTH_KEY"]
-
-        if config["OBP_AUTH_SERVER"].startswith("https://"):
-            auth_endpoint = config["OBP_AUTH_SERVER"]
-        else:
-            auth_endpoint = "https://{}".format(config["OBP_AUTH_SERVER"])
-
-        if auth_endpoint.endswith("/"):
-            auth_endpoint = auth_endpoint[:-1]
-
-        api_endpoint = auth_endpoint.replace("auth", "api")
-        endpoint_url = "{}/{}".format(api_endpoint, api_path)
-        headers = {"x-api-key": metaflow_token}
-        return endpoint_url, headers
-
-
-@cli.command(help="List all existing workstations")
-@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="",
-    help="The named metaflow profile in which your workstation exists",
-)
-def list_workstations(config_dir=None, profile=None):
-    try:
-        list_workstations_path = "v1/workstations"
-        endpoint_url, headers = prepare_workstations_api_request(
-            config_dir, profile, list_workstations_path
-        )
-        response = requests.get(endpoint_url, headers=headers)
-        if response.status_code == 200:
-            # Print pretty JSON
-            click.echo(json.dumps(response.json(), indent=4))
-        else:
-            click.secho("Failed to list workstations", fg="red")
-            click.secho("Error: {}".format(json.dumps(response.json(), indent=4)))
-    except Exception as e:
-        click.secho("Failed to list workstations", fg="red")
-        click.secho("Error: {}".format(str(e)))
-
-
-@cli.command(help="Hibernate workstation")
-@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="",
-    help="The named metaflow profile in which your workstation exists",
-)
-@click.option(
-    "-w",
-    "--workstation",
-    default="",
-    help="The ID of the workstation to hibernate",
-)
-def hibernate_workstation(config_dir=None, profile=None, workstation=None):
-    if workstation is None or workstation == "":
-        click.secho("Please specify a workstation ID", fg="red")
-        return
-    try:
-        list_workstations_path = "v1/workstations/hibernate/{}".format(workstation)
-        endpoint_url, headers = prepare_workstations_api_request(
-            config_dir, profile, list_workstations_path
-        )
-        response = requests.put(endpoint_url, headers=headers)
-        if response.status_code == 200:
-            response_json = response.json()
-            if len(response_json) > 0:
-                click.echo(json.dumps(response_json, indent=4))
-            else:
-                click.secho("Success", fg="green", bold=True)
-        else:
-            click.secho("Failed to hibernate workstation", fg="red")
-            click.secho("Error: {}".format(json.dumps(response.json(), indent=4)))
-    except Exception as e:
-        click.secho("Failed to hibernate workstation", fg="red")
-        click.secho("Error: {}".format(str(e)))
-
-
-@cli.command(help="Restart workstation")
-@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="",
-    help="The named metaflow profile in which your workstation exists",
-)
-@click.option(
-    "-w",
-    "--workstation",
-    default="",
-    help="The ID of the workstation to restart",
-)
-def restart_workstation(config_dir=None, profile=None, workstation=None):
-    if workstation is None or workstation == "":
-        click.secho("Please specify a workstation ID", fg="red")
-        return
-    try:
-        list_workstations_path = "v1/workstations/restart/{}".format(workstation)
-        endpoint_url, headers = prepare_workstations_api_request(
-            config_dir, profile, list_workstations_path
-        )
-        response = requests.put(endpoint_url, headers=headers)
-        if response.status_code == 200:
-            # Print pretty JSON
-            response_json = response.json()
-            if len(response_json) > 0:
-                click.echo(json.dumps(response_json, indent=4))
-            else:
-                click.secho("Success", fg="green", bold=True)
-        else:
-            click.secho("Failed to hibernate workstation", fg="red")
-            click.secho("Error: {}".format(json.dumps(response.json(), indent=4)))
-    except Exception as e:
-        click.secho("Failed to hibernate workstation", fg="red")
-        click.secho("Error: {}".format(str(e)))

{outerbounds-0.3.20rc0.dist-info → outerbounds-0.3.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: outerbounds
-Version: 0.3.20rc0
+Version: 0.3.21
 Summary: More Data Science, Less Administration
 License: Proprietary
 Keywords: data science,machine learning,MLOps
@@ -13,14 +13,10 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Dist: PyYAML (>=6.0,<7.0)
 Requires-Dist: click (>=8.1.3,<9.0.0)
-Requires-Dist: kubeconfig (>=1.1.1,<2.0.0)
-Requires-Dist: ob-metaflow (==2.8.0.1)
-Requires-Dist: ob-metaflow-extensions (==1.1.16)
+Requires-Dist: ob-metaflow (==2.8.4.2)
+Requires-Dist: ob-metaflow-extensions (==1.1.19)
 Requires-Dist: opentelemetry-distro (==0.37b0)
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (==1.16.0)
 Requires-Dist: opentelemetry-instrumentation-requests (==0.37b0)
-Requires-Dist: types-PyYAML (>=6.0.12.8,<7.0.0.0)
-Requires-Dist: types-requests (>=2.28.11.15,<3.0.0.0)
 Project-URL: Documentation, https://docs.metaflow.org

outerbounds-0.3.21.dist-info/RECORD

@@ -0,0 +1,6 @@
+outerbounds/__init__.py,sha256=8DqpohQHW-bb7fMvLEHjnM5DlcLGRw7IthmARDfciKY,13620
+outerbounds/cli_main.py,sha256=k9CgQa-3xvfVcIK_ja2WKsKtwjQma9kYJyQY7LagNe4,56
+outerbounds-0.3.21.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+outerbounds-0.3.21.dist-info/entry_points.txt,sha256=7ye0281PKlvqxu15rjw60zKg2pMsXI49_A8BmGqIqBw,47
+outerbounds-0.3.21.dist-info/METADATA,sha256=V3Ja9j2HBR2NzsYKOQUsxKSfSBCdB5szScQIzR3knTQ,924
+outerbounds-0.3.21.dist-info/RECORD,,

outerbounds-0.3.20rc0.dist-info/RECORD

@@ -1,6 +0,0 @@
-outerbounds/__init__.py,sha256=nbc4Q5XfGhJ5IH6TpQhbRxq4RycpCZdza884KItq0Ak,23393
-outerbounds/cli_main.py,sha256=k9CgQa-3xvfVcIK_ja2WKsKtwjQma9kYJyQY7LagNe4,56
-outerbounds-0.3.20rc0.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-outerbounds-0.3.20rc0.dist-info/entry_points.txt,sha256=7ye0281PKlvqxu15rjw60zKg2pMsXI49_A8BmGqIqBw,47
-outerbounds-0.3.20rc0.dist-info/METADATA,sha256=Jipw5msg0tZxYBUJ5JNpmM5S-C4EarDySSmtgjXt0GA,1109
-outerbounds-0.3.20rc0.dist-info/RECORD,,