amazon-sp-cli 0.3.2__tar.gz → 0.3.3__tar.gz

{amazon_sp_cli-0.3.2/amazon_sp_cli.egg-info → amazon_sp_cli-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amazon-sp-cli
-Version: 0.3.2
+Version: 0.3.3
 Summary: CLI tool for Amazon Selling Partner API (SP-API) operations
 Home-page: https://github.com/stellaraether/amazon-sp-cli
 Author: Lunan Li

{amazon_sp_cli-0.3.2 → amazon_sp_cli-0.3.3}/amazon_sp_cli/__init__.py

@@ -1,3 +1,3 @@
 # Amazon SP-API CLI tool
 
-__version__ = "0.3.2"
+__version__ = "0.3.3"

{amazon_sp_cli-0.3.2 → amazon_sp_cli-0.3.3}/amazon_sp_cli/auth.py

@@ -17,7 +17,8 @@ class SPAPIAuth:
     CACHE_FILE = Path.home() / ".config" / "amazon-sp-cli" / "token-cache.json"
     BUFFER_SECONDS = 60
 
-    def __init__(self, credentials_path: str = None):
+    def __init__(self, credentials_path: str = None, profile: str = "default"):
+        self.profile = profile
         self.credentials = self._load_credentials(credentials_path)
         self._ensure_cache_dir()
 
@@ -32,7 +33,7 @@ class SPAPIAuth:
         with open(path, "r") as f:
             config = yaml.safe_load(f)
 
-        return config.get("default", config)
+        return config.get(self.profile, config)
 
     def _ensure_cache_dir(self):
         """Ensure cache directory exists."""

{amazon_sp_cli-0.3.2 → amazon_sp_cli-0.3.3}/amazon_sp_cli/cli.py

@@ -58,7 +58,7 @@ def _check_path():
 def _ensure_auth_client(ctx):
     """Lazily create auth and client if not already present."""
     if "client" not in ctx.obj:
-        auth = SPAPIAuth(ctx.obj.get("credentials_path"))
+        auth = SPAPIAuth(ctx.obj.get("credentials_path"), ctx.obj.get("profile", "default"))
         if auth.credentials is None:
             click.echo("Error: No credentials found. Run 'amz-sp auth setup' first.", err=True)
             raise click.Abort()
@@ -86,9 +86,11 @@ def handle_errors(f):
 @click.version_option(version=__version__, prog_name="amz-sp")
 @click.group()
 @click.option("--credentials", "-c", help="Path to credentials YAML file")
+@click.option("--profile", "-p", default="default", help="Credentials profile to use")
 @click.pass_context
-def cli(ctx, credentials):
+def cli(ctx, credentials, profile):
     """Amazon SP-API CLI - Manage listings, pricing, inventory, and more."""
     _check_path()
     ctx.ensure_object(dict)
     ctx.obj["credentials_path"] = credentials
+    ctx.obj["profile"] = profile

{amazon_sp_cli-0.3.2 → amazon_sp_cli-0.3.3/amazon_sp_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amazon-sp-cli
-Version: 0.3.2
+Version: 0.3.3
 Summary: CLI tool for Amazon Selling Partner API (SP-API) operations
 Home-page: https://github.com/stellaraether/amazon-sp-cli
 Author: Lunan Li

{amazon_sp_cli-0.3.2 → amazon_sp_cli-0.3.3}/tests/test_auth.py

@@ -23,6 +23,10 @@ default:
   refresh_token: "test-refresh-token"
   client_id: "test-client-id"
   client_secret: "test-client-secret"
+prod:
+  refresh_token: "prod-refresh-token"
+  client_id: "prod-client-id"
+  client_secret: "prod-client-secret"
 """
             )
             path = f.name
@@ -44,6 +48,20 @@ default:
         assert auth.credentials["client_id"] == "test-client-id"
         assert auth.credentials["client_secret"] == "test-client-secret"
 
+    def test_load_credentials_profile(self, temp_credentials):
+        """Test credentials loading with a specific profile."""
+        auth = SPAPIAuth(temp_credentials, profile="prod")
+        assert auth.credentials["refresh_token"] == "prod-refresh-token"
+        assert auth.credentials["client_id"] == "prod-client-id"
+        assert auth.credentials["client_secret"] == "prod-client-secret"
+
+    def test_load_credentials_missing_profile_fallback(self, temp_credentials):
+        """Test fallback when profile is missing."""
+        auth = SPAPIAuth(temp_credentials, profile="nonexistent")
+        # Falls back to the whole config dict
+        assert "default" in auth.credentials
+        assert "prod" in auth.credentials
+
     def test_token_valid(self, temp_credentials, temp_cache_dir):
         """Test token validation."""
         auth = SPAPIAuth(temp_credentials)