Merge pull request #35 from stax-labs/feat/status_code

feat(exception) add status_code to exception

Author: ZXYmania

examples/api_tokens/create_api_token.py

@@ -0,0 +1,20 @@
+import os
+
+from staxapp.config import Config
+from staxapp.openapi import StaxClient
+
+Config.access_key = os.getenv("STAX_ACCESS_KEY")
+Config.secret_key = os.getenv("STAX_SECRET_KEY")
+
+# The name of the token
+token_name = <Api Token Name>
+# What access the token will have to the api
+token_role = <Api Token Role>
+# Whether the secret to store the token in ssm params
+# Otherwise it will be returned in the response
+store_token = <Store Token?>
+
+# Create an Api token with access to your Stax Organisation
+teams = StaxClient("teams")
+response = teams.CreateApiToken(Name=token_name, Role=token_role, StoreToken=store_token)
+print(response)

examples/api_tokens/delete_api_token.py

@@ -0,0 +1,16 @@
+import os
+
+from staxapp.config import Config
+from staxapp.openapi import StaxClient
+
+Config.access_key = os.getenv("STAX_ACCESS_KEY")
+Config.secret_key = os.getenv("STAX_SECRET_KEY")
+
+# The token to delete's AccessKey
+access_key = <Access Key>
+
+# Delete an Api Token
+teams = StaxClient("teams")
+response = teams.DeleteApiToken(AccessKey=access_key)
+
+print(response)

staxapp/exceptions.py

@@ -5,6 +5,7 @@
 class ApiException(Exception):
     def __init__(self, message, response, detail=""):
         try:
+            self.status_code = response.status_code
             if response.json().get("Error"):
                 logging.error(f"{response.status_code}: {response.json()}")
                 self.message = f"Api Exception: {response.status_code} -{detail} {response.json()['Error']}"