dbt-platform-helper 12.3.0__py3-none-any.whl → 12.4.1__py3-none-any.whl

dbt_platform_helper/providers/cloudformation.py

@@ -4,7 +4,7 @@ import botocore
 from cfn_tools import dump_yaml
 from cfn_tools import load_yaml
 
-from dbt_platform_helper.exceptions import CloudFormationException
+from dbt_platform_helper.platform_exception import PlatformException
 
 
 class CloudFormation:
@@ -125,3 +125,10 @@ class CloudFormation:
                 raise CloudFormationException(
                     stack_name, f"Error while waiting for stack status: {str(err)}"
                 )
+
+
+class CloudFormationException(PlatformException):
+    def __init__(self, stack_name: str, current_status: str):
+        super().__init__(
+            f"The CloudFormation stack '{stack_name}' is not in a good state: {current_status}"
+        )

dbt_platform_helper/providers/copilot.py

@@ -4,7 +4,7 @@ import time
 from botocore.exceptions import ClientError
 
 from dbt_platform_helper.constants import CONDUIT_DOCKER_IMAGE_LOCATION
-from dbt_platform_helper.exceptions import CreateTaskTimeoutError
+from dbt_platform_helper.providers.aws import CreateTaskTimeoutException
 from dbt_platform_helper.providers.secrets import Secrets
 from dbt_platform_helper.utils.application import Application
 from dbt_platform_helper.utils.messages import abort_with_error
@@ -13,7 +13,6 @@ from dbt_platform_helper.utils.messages import abort_with_error
 def create_addon_client_task(
     iam_client,
     ssm_client,
-    secrets_manager_client,
     subprocess,
     application: Application,
     env: str,
@@ -32,7 +31,6 @@ def create_addon_client_task(
         elif access == "admin":
             create_postgres_admin_task(
                 ssm_client,
-                secrets_manager_client,
                 subprocess,
                 application,
                 addon_name,
@@ -74,7 +72,6 @@ def create_addon_client_task(
 
 def create_postgres_admin_task(
     ssm_client,
-    secrets_manager_client,
     subprocess,
     app: Application,
     addon_name: str,
@@ -147,7 +144,7 @@ def connect_to_addon_client_task(
         time.sleep(1)
 
     if not running:
-        raise CreateTaskTimeoutError(task_name, application_name, env)
+        raise CreateTaskTimeoutException(task_name, application_name, env)
 
 
 def _normalise_secret_name(addon_name: str) -> str:

dbt_platform_helper/providers/ecs.py

@@ -3,8 +3,7 @@ import string
 import time
 from typing import List
 
-from dbt_platform_helper.exceptions import ECSAgentNotRunning
-from dbt_platform_helper.exceptions import NoClusterError
+from dbt_platform_helper.platform_exception import PlatformException
 
 
 class ECS:
@@ -36,7 +35,7 @@ class ECS:
             if app_key_found and env_key_found and cluster_key_found:
                 return cluster_arn
 
-        raise NoClusterError(self.application_name, self.env)
+        raise NoClusterException(self.application_name, self.env)
 
     def get_or_create_task_name(self, addon_name: str, parameter_name: str) -> str:
         """Fetches the task name from SSM or creates a new one if not found."""
@@ -84,4 +83,20 @@ class ECS:
                 time.sleep(1)
 
         if execute_command_agent_status != "RUNNING":
-            raise ECSAgentNotRunning
+            raise ECSAgentNotRunningException
+
+
+class ECSException(PlatformException):
+    pass
+
+
+class ECSAgentNotRunningException(ECSException):
+    def __init__(self):
+        super().__init__("""ECS exec agent never reached "RUNNING" status""")
+
+
+class NoClusterException(ECSException):
+    def __init__(self, application_name: str, environment: str):
+        super().__init__(
+            f"""No ECS cluster found for "{application_name}" in "{environment}" environment."""
+        )

dbt_platform_helper/providers/load_balancers.py

@@ -1,5 +1,7 @@
 import boto3
 
+from dbt_platform_helper.platform_exception import PlatformException
+
 
 def find_load_balancer(session: boto3.Session, app: str, env: str) -> str:
     lb_client = session.client("elbv2")
@@ -16,7 +18,7 @@ def find_load_balancer(session: boto3.Session, app: str, env: str) -> str:
             load_balancer_arn = lb["ResourceArn"]
 
     if not load_balancer_arn:
-        raise LoadBalancerNotFoundError()
+        raise LoadBalancerNotFoundException()
 
     return load_balancer_arn
 
@@ -34,18 +36,22 @@ def find_https_listener(session: boto3.Session, app: str, env: str) -> str:
         pass
 
     if not listener_arn:
-        raise ListenerNotFoundError()
+        raise ListenerNotFoundException()
 
     return listener_arn
 
 
-class LoadBalancerNotFoundError(Exception):
+class LoadBalancerException(PlatformException):
+    pass
+
+
+class LoadBalancerNotFoundException(LoadBalancerException):
     pass
 
 
-class ListenerNotFoundError(Exception):
+class ListenerNotFoundException(LoadBalancerException):
     pass
 
 
-class ListenerRuleNotFoundError(Exception):
+class ListenerRuleNotFoundException(LoadBalancerException):
     pass