Create custom exceptions and link it to website

Author: abuabraham-ttd

scripts/aws/ec2.py

@@ -9,7 +9,7 @@
 import requests
 import signal
 import argparse
-from botocore.exceptions import ClientError
+from botocore.exceptions import ClientError, NoCredentialsError
 from typing import Dict
 import sys
 import time
@@ -97,14 +97,13 @@ def add_defaults(configs: Dict[str, any]) -> AWSConfidentialComputeConfig:
 
         region = self.__get_current_region()
         print(f"Running in {region}")
-        try:
-            client = boto3.client("secretsmanager", region_name=region)
-        except Exception as e:
-            raise MissingInstanceProfile(self.__class__.__name__)
+        client = boto3.client("secretsmanager", region_name=region)
         try:
             secret = add_defaults(json.loads(client.get_secret_value(SecretId=secret_identifier)["SecretString"]))
             self.__validate_aws_specific_config(secret)
             return secret
+        except NoCredentialsError as _:
+            raise MissingInstanceProfile(self.__class__.__name__)
         except ClientError as _:
             raise ConfigNotFound(self.__class__.__name__, f"{secret_identifier} in {region}")
 

scripts/confidential_compute.py

@@ -22,7 +22,7 @@ def __init__(self, error_name, provider, extra_message):
             "GCP": "https://unifiedid.com/docs/guides/operator-private-gcp-confidential-space#uid2-operator-error-codes",
         }
         url = urls.get(provider)
-        super().__init__(f"{error_name} \n" + (extra_message if extra_message else "") + f" \n Visit {url} for more details")
+        super().__init__(f"{error_name} \n" + (extra_message if extra_message else "") + f"\nVisit {url} for more details")
 
 class MissingInstanceProfile(ConfidentialComputeStartupException):
     def __init__(self, cls):