steve-cli 0.3.0__tar.gz → 0.3.2__tar.gz

{steve_cli-0.3.0 → steve_cli-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: steve-cli
-Version: 0.3.0
+Version: 0.3.2
 Summary: A simple CLI tool to run jobs from jobs.yaml with proper environment setup
 Author: Frank
 License: MIT

{steve_cli-0.3.0 → steve_cli-0.3.2}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "steve-cli"
-version = "0.3.0"
+version = "0.3.2"
 description = "A simple CLI tool to run jobs from jobs.yaml with proper environment setup"
 readme = "README.md"
 license = {text = "MIT"}

{steve_cli-0.3.0 → steve_cli-0.3.2}/steve_cli/cli.py

@@ -233,5 +233,100 @@ def setup_env():
         click.echo(f"   💡 Run: {click.style(f'source {out_file.name}', fg='yellow')}")
 
 
+def _detect_workspaces() -> List[str]:
+    tiers = {"BRONZE", "SILVER", "GOLD"}
+    workspaces = set()
+    for key in os.environ:
+        if key.endswith("_ACCESS_KEY"):
+            prefix = key[: -len("_ACCESS_KEY")]
+            if prefix not in tiers:
+                workspaces.add(prefix)
+    return sorted(workspaces)
+
+
+def _build_tree(keys: List[str]) -> Dict[str, Any]:
+    tree: Dict[str, Any] = {}
+    for key in keys:
+        parts = key.split("/")
+        node = tree
+        for part in parts[:-1]:
+            node = node.setdefault(part, {})
+        node[parts[-1]] = None
+    return tree
+
+
+def _print_tree(node: Dict[str, Any], prefix: str = "", is_last: bool = True) -> None:
+    items = sorted(node.items())
+    for i, (name, child) in enumerate(items):
+        last = i == len(items) - 1
+        connector = "└── " if last else "├── "
+        click.echo(f"{prefix}{connector}{name}")
+        if child is not None:
+            extension = "    " if last else "│   "
+            _print_tree(child, prefix + extension, last)
+
+
+def _list_bucket(storage_kwargs: dict, label: str, bucket_name: str) -> None:
+    click.echo(f"  {click.style(label, fg='cyan')} ({bucket_name})")
+    try:
+        from steve_cli.storage import S3Storage
+        storage = S3Storage(**storage_kwargs)
+        keys = storage.list_all()
+        if not keys:
+            click.echo("    (empty)")
+        else:
+            tree = _build_tree(keys)
+            _print_tree(tree, prefix="    ")
+    except EnvironmentError as e:
+        click.echo(f"    ⚠️  {e}", err=True)
+    except Exception as e:
+        click.echo(f"    ❌ {e}", err=True)
+
+
+@main.command("buckets")
+def buckets():
+    """List all S3 buckets detected from env variables and show their files as a tree."""
+    tiers = ["bronze", "silver", "gold"]
+
+    options: List[Dict[str, Any]] = []
+
+    bare_tiers = [t for t in tiers if os.getenv(f"{t.upper()}_ACCESS_KEY")]
+    for tier in bare_tiers:
+        bucket_name = os.getenv(f"{tier.upper()}_BUCKET", "")
+        options.append({
+            "label": f"default / {tier} ({bucket_name})",
+            "kwargs": {"tier": tier},
+            "bucket_name": bucket_name,
+            "tier": tier,
+        })
+
+    for ws in _detect_workspaces():
+        for tier in tiers:
+            bucket_name = os.getenv(f"{ws}_BUCKET_{tier.upper()}")
+            if not bucket_name:
+                continue
+            options.append({
+                "label": f"{ws} / {tier} ({bucket_name})",
+                "kwargs": {"tier": tier, "workspace": ws.lower().replace("_", "-")},
+                "bucket_name": bucket_name,
+                "tier": tier,
+            })
+
+    if not options:
+        click.echo("No bucket env variables found (expected: BRONZE_ACCESS_KEY or {WORKSPACE}_ACCESS_KEY).")
+        return
+
+    choice = questionary.select(
+        "Select a bucket to list:",
+        choices=[o["label"] for o in options],
+    ).ask()
+
+    if choice is None:
+        sys.exit(0)
+
+    selected = next(o for o in options if o["label"] == choice)
+    _list_bucket(selected["kwargs"], selected["tier"], selected["bucket_name"])
+
+
 if __name__ == '__main__':
     main()

{steve_cli-0.3.0 → steve_cli-0.3.2}/steve_cli/storage.py

@@ -16,24 +16,34 @@ class Storage(Protocol):
 
 
 class S3Storage:
-    def __init__(self, tier: str = "bronze"):
+    def __init__(self, tier: str = "bronze", workspace: str | None = None):
         self.tier = tier.lower()
 
-        if self.tier == "bronze":
-            access_key = os.getenv("BRONZE_ACCESS_KEY")
-            secret_key = os.getenv("BRONZE_SECRET_KEY")
-            self.bucket = os.getenv("BRONZE_BUCKET")
-            required_vars = ["BRONZE_ACCESS_KEY", "BRONZE_SECRET_KEY", "BRONZE_BUCKET"]
-        elif self.tier == "silver":
-            access_key = os.getenv("SILVER_ACCESS_KEY")
-            secret_key = os.getenv("SILVER_SECRET_KEY")
-            self.bucket = os.getenv("SILVER_BUCKET")
-            required_vars = ["SILVER_ACCESS_KEY", "SILVER_SECRET_KEY", "SILVER_BUCKET"]
+        if workspace is not None:
+            prefix = workspace.upper().replace("-", "_")
+            tier_suffix = f"BUCKET_{tier.upper()}"
+            access_key = os.getenv(f"{prefix}_ACCESS_KEY")
+            secret_key = os.getenv(f"{prefix}_SECRET_KEY")
+            self.bucket = os.getenv(f"{prefix}_{tier_suffix}")
+            required_vars = [f"{prefix}_ACCESS_KEY", f"{prefix}_SECRET_KEY", f"{prefix}_{tier_suffix}"]
+            endpoint = os.getenv(f"{prefix}_S3_ENDPOINT", os.getenv("S3_ENDPOINT", "http://localhost:9000"))
         else:
-            access_key = os.getenv("GOLD_ACCESS_KEY")
-            secret_key = os.getenv("GOLD_SECRET_KEY")
-            self.bucket = os.getenv("GOLD_BUCKET")
-            required_vars = ["GOLD_ACCESS_KEY", "GOLD_SECRET_KEY", "GOLD_BUCKET"]
+            if self.tier == "bronze":
+                access_key = os.getenv("BRONZE_ACCESS_KEY")
+                secret_key = os.getenv("BRONZE_SECRET_KEY")
+                self.bucket = os.getenv("BRONZE_BUCKET")
+                required_vars = ["BRONZE_ACCESS_KEY", "BRONZE_SECRET_KEY", "BRONZE_BUCKET"]
+            elif self.tier == "silver":
+                access_key = os.getenv("SILVER_ACCESS_KEY")
+                secret_key = os.getenv("SILVER_SECRET_KEY")
+                self.bucket = os.getenv("SILVER_BUCKET")
+                required_vars = ["SILVER_ACCESS_KEY", "SILVER_SECRET_KEY", "SILVER_BUCKET"]
+            else:
+                access_key = os.getenv("GOLD_ACCESS_KEY")
+                secret_key = os.getenv("GOLD_SECRET_KEY")
+                self.bucket = os.getenv("GOLD_BUCKET")
+                required_vars = ["GOLD_ACCESS_KEY", "GOLD_SECRET_KEY", "GOLD_BUCKET"]
+            endpoint = os.getenv("S3_ENDPOINT", "http://localhost:9000")
 
         missing_vars = [var for var in required_vars if not os.getenv(var)]
         if missing_vars:
@@ -41,8 +51,6 @@ class S3Storage:
                 f"Missing required environment variables for {self.tier} tier: {', '.join(missing_vars)}"
             )
 
-        endpoint = os.getenv("S3_ENDPOINT", "http://localhost:9000")
-
         self.client = boto3.client(
             "s3",
             endpoint_url=endpoint,
@@ -153,6 +161,14 @@ class S3Storage:
             for obj in resp.get("Contents", [])
         ]
 
+    def list_all(self) -> List[str]:
+        keys = []
+        paginator = self.client.get_paginator("list_objects_v2")
+        for page in paginator.paginate(Bucket=self.bucket):
+            for obj in page.get("Contents", []):
+                keys.append(obj["Key"])
+        return keys
+
 
-def get_storage(tier: str = "bronze") -> Storage:
-    return S3Storage(tier=tier)
+def get_storage(tier: str = "bronze", workspace: str | None = None) -> Storage:
+    return S3Storage(tier=tier, workspace=workspace)

{steve_cli-0.3.0 → steve_cli-0.3.2}/steve_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: steve-cli
-Version: 0.3.0
+Version: 0.3.2
 Summary: A simple CLI tool to run jobs from jobs.yaml with proper environment setup
 Author: Frank
 License: MIT