dbt-platform-helper 13.1.2__py3-none-any.whl → 13.3.0__py3-none-any.whl

dbt_platform_helper/commands/application.py

@@ -10,9 +10,7 @@ from datetime import timedelta
 import click
 from prettytable import PrettyTable
 
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.utils.application import load_application
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 
@@ -141,7 +139,7 @@ def container_stats(env, app, storage, network):
         storage_write = field[6]["value"]
         network_read = field[7]["value"]
         network_write = field[8]["value"]
-        cpu = "%.1f%%" % float(field[9]["value"])
+        cpu = f"{float(field[9]['value']):.1f}%"
         memory = f"{field[10]['value']}M"
 
         # Nothing to compare to at start.

dbt_platform_helper/commands/codebase.py

@@ -1,11 +1,11 @@
 import click
 
 from dbt_platform_helper.domain.codebase import Codebase
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.platform_exception import PlatformException
 from dbt_platform_helper.providers.io import ClickIOProvider
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
+from dbt_platform_helper.providers.parameter_store import ParameterStore
+from dbt_platform_helper.utils.aws import get_aws_session_or_abort
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 
 
@@ -19,7 +19,7 @@ def codebase():
 def prepare():
     """Sets up an application codebase for use within a DBT platform project."""
     try:
-        Codebase().prepare()
+        Codebase(ParameterStore(get_aws_session_or_abort().client("ssm"))).prepare()
     except PlatformException as err:
         ClickIOProvider().abort_with_error(str(err))
 
@@ -35,7 +35,7 @@ def prepare():
 def list(app, with_images):
     """List available codebases for the application."""
     try:
-        Codebase().list(app, with_images)
+        Codebase(ParameterStore(get_aws_session_or_abort().client("ssm"))).list(app, with_images)
     except PlatformException as err:
         ClickIOProvider().abort_with_error(str(err))
 
@@ -51,7 +51,9 @@ def list(app, with_images):
 def build(app, codebase, commit):
     """Trigger a CodePipeline pipeline based build."""
     try:
-        Codebase().build(app, codebase, commit)
+        Codebase(ParameterStore(get_aws_session_or_abort().client("ssm"))).build(
+            app, codebase, commit
+        )
     except PlatformException as err:
         ClickIOProvider().abort_with_error(str(err))
 
@@ -67,6 +69,8 @@ def build(app, codebase, commit):
 @click.option("--commit", help="GitHub commit hash", required=True)
 def deploy(app, env, codebase, commit):
     try:
-        Codebase().deploy(app, env, codebase, commit)
+        Codebase(ParameterStore(get_aws_session_or_abort().client("ssm"))).deploy(
+            app, env, codebase, commit
+        )
     except PlatformException as err:
         ClickIOProvider().abort_with_error(str(err))

dbt_platform_helper/commands/conduit.py

@@ -1,13 +1,11 @@
 import click
 
 from dbt_platform_helper.domain.conduit import Conduit
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.platform_exception import PlatformException
 from dbt_platform_helper.providers.cloudformation import CloudFormation
 from dbt_platform_helper.providers.ecs import ECS
 from dbt_platform_helper.providers.io import ClickIOProvider
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
 from dbt_platform_helper.providers.secrets import Secrets
 from dbt_platform_helper.utils.application import load_application
 from dbt_platform_helper.utils.click import ClickDocOptCommand

dbt_platform_helper/commands/config.py

@@ -1,56 +1,11 @@
-import os
-import webbrowser
-from pathlib import Path
-from typing import Dict
-
-import boto3
-import botocore
 import click
-from prettytable import PrettyTable
 
-from dbt_platform_helper.providers.config import ConfigProvider
+from dbt_platform_helper.domain.config import Config
+from dbt_platform_helper.platform_exception import PlatformException
+from dbt_platform_helper.providers.aws.sso_auth import SSOAuthProvider
 from dbt_platform_helper.providers.io import ClickIOProvider
-from dbt_platform_helper.providers.semantic_version import (
-    IncompatibleMajorVersionException,
-)
-from dbt_platform_helper.providers.validation import ValidationException
-from dbt_platform_helper.utils import versioning
+from dbt_platform_helper.utils.aws import get_aws_session_or_abort
 from dbt_platform_helper.utils.click import ClickDocOptGroup
-from dbt_platform_helper.utils.versioning import get_platform_helper_version_status
-
-yes = "\033[92m✔\033[0m"
-no = "\033[91m✖\033[0m"
-maybe = "\033[93m?\033[0m"
-
-RECOMMENDATIONS = {
-    "dbt-platform-helper-upgrade": (
-        "Upgrade dbt-platform-helper to version {version} `pip install "
-        "--upgrade dbt-platform-helper=={version}`."
-    ),
-    "dbt-platform-helper-upgrade-note": (
-        "Post upgrade, run `platform-helper copilot make-addons` to " "update your addon templates."
-    ),
-    "generic-tool-upgrade": "Upgrade {tool} to version {version}.",
-}
-
-SSO_START_URL = "https://uktrade.awsapps.com/start"
-
-AWS_CONFIG = """
-#
-# uktrade
-#
-
-[sso-session uktrade]
-sso_start_url = https://uktrade.awsapps.com/start#/
-sso_region = eu-west-2
-sso_registration_scopes = sso:account:access
-
-[default]
-sso_session = uktrade
-region = eu-west-2
-output = json
-
-"""
 
 
 @click.group(cls=ClickDocOptGroup)
@@ -61,190 +16,10 @@ def config():
 @config.command()
 def validate():
     """Validate deployment or application configuration."""
-    ran_checks = False
-    if Path("copilot").exists():
-        click.secho("\nDetected a deployment repository", fg="blue")
-        deployment()
-        ran_checks = True
-
-    if not ran_checks:
-        click.secho("Could not find a deployment repository, no checks to run.", fg="red")
-        exit(1)
-
-
-def deployment():
-    click.secho()
-
-    compatible = True
-    platform_helper_version_status = get_platform_helper_version_status()
-    ClickIOProvider().process_messages(platform_helper_version_status.validate())
-    copilot_versions = versioning.get_copilot_versions()
-    aws_versions = versioning.get_aws_versions()
-    _check_tool_versions(platform_helper_version_status, copilot_versions, aws_versions)
-    click.secho("Checking addons templates versions...", fg="blue")
-
-    local_version = platform_helper_version_status.local
-    latest_release = platform_helper_version_status.latest
-    addons_templates_table = PrettyTable()
-    addons_templates_table.field_names = [
-        "Addons Template File",
-        "Generated with",
-        "Compatible with local?",
-        "Compatible with latest?",
-    ]
-    addons_templates_table.align["Addons Template File"] = "l"
-
-    addons_templates = list(Path("./copilot").glob("**/addons/*"))
-    # Sort by template file path
-    addons_templates.sort(key=lambda e: str(e))
-    # Bring environment addons to the top
-    addons_templates.sort(key=lambda e: "environments/" not in str(e))
-
-    recommendations = {}
-
-    ConfigProvider().config_file_check()
-
-    for template_file in addons_templates:
-        generated_with_version = maybe
-        local_compatible_symbol = yes
-        latest_compatible_symbol = yes
-
-        try:
-            generated_with_version = versioning.get_template_generated_with_version(
-                str(template_file.resolve())
-            )
-            versioning.validate_template_version(local_version, str(template_file.resolve()))
-        except IncompatibleMajorVersionException:
-            local_compatible_symbol = no
-            compatible = False
-            recommendations["dbt-platform-helper-upgrade"] = RECOMMENDATIONS[
-                "dbt-platform-helper-upgrade"
-            ].format(version=latest_release)
-            recommendations["dbt-platform-helper-upgrade-note"] = RECOMMENDATIONS[
-                "dbt-platform-helper-upgrade-note"
-            ]
-        except ValidationException:
-            local_compatible_symbol = maybe
-            compatible = False
-            recommendations["dbt-platform-helper-upgrade"] = RECOMMENDATIONS[
-                "dbt-platform-helper-upgrade"
-            ].format(version=latest_release)
-            recommendations["dbt-platform-helper-upgrade-note"] = RECOMMENDATIONS[
-                "dbt-platform-helper-upgrade-note"
-            ]
-
-        try:
-            generated_with_version = versioning.get_template_generated_with_version(
-                str(template_file.resolve())
-            )
-            versioning.validate_template_version(latest_release, str(template_file.resolve()))
-        except IncompatibleMajorVersionException:
-            latest_compatible_symbol = no
-            compatible = False
-        except ValidationException:
-            latest_compatible_symbol = maybe
-            compatible = False
-
-        addons_templates_table.add_row(
-            [
-                template_file.relative_to("."),
-                (maybe if latest_compatible_symbol is maybe else str(generated_with_version)),
-                local_compatible_symbol,
-                latest_compatible_symbol,
-            ]
-        )
-
-    click.secho(addons_templates_table)
-    render_recommendations(recommendations)
-
-    exit(0 if compatible else 1)
-
-
-def _check_tool_versions(platform_helper_versions, copilot_versions, aws_versions):
-    click.secho("Checking tooling versions...", fg="blue")
-    recommendations = {}
-
-    local_copilot_version = copilot_versions.local
-    copilot_latest_release = copilot_versions.latest
-    if local_copilot_version is None:
-        recommendations["install-copilot"] = (
-            "Install AWS Copilot https://aws.github.io/copilot-cli/"
-        )
-
-    if aws_versions.local is None:
-        recommendations["install-aws"] = "Install AWS CLI https://aws.amazon.com/cli/"
-
-    tool_versions_table = PrettyTable()
-    tool_versions_table.field_names = [
-        "Tool",
-        "Local version",
-        "Released version",
-        "Running latest?",
-    ]
-    tool_versions_table.align["Tool"] = "l"
-
-    tool_versions_table.add_row(
-        [
-            "aws",
-            str(aws_versions.local),
-            str(aws_versions.latest),
-            no if aws_versions.is_outdated() else yes,
-        ]
-    )
-    tool_versions_table.add_row(
-        [
-            "copilot",
-            str(copilot_versions.local),
-            str(copilot_versions.latest),
-            no if copilot_versions.is_outdated() else yes,
-        ]
-    )
-    tool_versions_table.add_row(
-        [
-            "dbt-platform-helper",
-            str(platform_helper_versions.local),
-            str(platform_helper_versions.latest),
-            no if platform_helper_versions.is_outdated() else yes,
-        ]
-    )
-
-    click.secho(tool_versions_table)
-
-    if aws_versions.is_outdated() and "install-aws" not in recommendations:
-        recommendations["aws-upgrade"] = RECOMMENDATIONS["generic-tool-upgrade"].format(
-            tool="AWS CLI",
-            version=str(aws_versions.latest),
-        )
-
-    if copilot_versions.is_outdated() and "install-copilot" not in recommendations:
-        recommendations["copilot-upgrade"] = RECOMMENDATIONS["generic-tool-upgrade"].format(
-            tool="AWS Copilot",
-            version=str(copilot_latest_release),
-        )
-
-    if platform_helper_versions.is_outdated():
-        recommendations["dbt-platform-helper-upgrade"] = RECOMMENDATIONS[
-            "dbt-platform-helper-upgrade"
-        ].format(version=str(platform_helper_versions.latest))
-        recommendations["dbt-platform-helper-upgrade-note"] = RECOMMENDATIONS[
-            "dbt-platform-helper-upgrade-note"
-        ]
-
-    render_recommendations(recommendations)
-
-
-def render_recommendations(recommendations: Dict[str, str]):
-    if recommendations:
-        click.secho("\nRecommendations:\n", bold=True)
-
-        for name, recommendation in recommendations.items():
-            if name.endswith("-note"):
-                continue
-            click.secho(f"  - {recommendation}")
-            if recommendations.get(f"{name}-note", False):
-                click.secho(f"    {recommendations.get(f'{name}-note')}")
-
-        click.secho()
+    try:
+        Config().validate()
+    except PlatformException as err:
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @config.command()
@@ -256,85 +31,8 @@ def aws(file_path):
 
     If no `--file-path` is specified, defaults to `~/.aws/config`.
     """
-    sso_oidc_client = boto3.client("sso-oidc", region_name="eu-west-2")
-    sso_client = boto3.client("sso", region_name="eu-west-2")
-    oidc_app = create_oidc_application(sso_oidc_client)
-    verification_url, device_code = get_device_code(sso_oidc_client, oidc_app, SSO_START_URL)
-
-    if click.confirm(
-        "You are about to be redirected to a verification page. You will need to complete sign-in before returning to the command line. Do you want to continue?",
-        abort=True,
-    ):
-        webbrowser.open(verification_url)
-
-    if click.confirm("Have you completed the sign-in process in your browser?", abort=True):
-        access_token = get_access_token(device_code, sso_oidc_client, oidc_app)
-
-    aws_config_path = os.path.expanduser(file_path)
-
-    if click.confirm(
-        f"This command is destructive and will overwrite file contents at {file_path}. Are you sure you want to continue?",
-        abort=True,
-    ):
-        with open(aws_config_path, "w") as config_file:
-            config_file.write(AWS_CONFIG)
-
-            for account in retrieve_aws_accounts(sso_client, access_token):
-                config_file.write(f"[profile {account['accountName']}]\n")
-                config_file.write("sso_session = uktrade\n")
-                config_file.write(f"sso_account_id = {account['accountId']}\n")
-                config_file.write("sso_role_name = AdministratorAccess\n")
-                config_file.write("region = eu-west-2\n")
-                config_file.write("output = json\n")
-                config_file.write("\n")
-
-
-def create_oidc_application(sso_oidc_client):
-    print("Creating temporary AWS SSO OIDC application")
-    client = sso_oidc_client.register_client(
-        clientName="platform-helper",
-        clientType="public",
-    )
-    client_id = client.get("clientId")
-    client_secret = client.get("clientSecret")
-
-    return client_id, client_secret
-
-
-def get_device_code(sso_oidc_client, oidc_application, start_url):
-    print("Initiating device code flow")
-    authz = sso_oidc_client.start_device_authorization(
-        clientId=oidc_application[0],
-        clientSecret=oidc_application[1],
-        startUrl=start_url,
-    )
-    url = authz.get("verificationUriComplete")
-    deviceCode = authz.get("deviceCode")
-
-    return url, deviceCode
-
-
-def retrieve_aws_accounts(sso_client, aws_sso_token):
-    aws_accounts_response = sso_client.list_accounts(
-        accessToken=aws_sso_token,
-        maxResults=100,
-    )
-    if len(aws_accounts_response.get("accountList", [])) == 0:
-        raise RuntimeError("Unable to retrieve AWS SSO account list\n")
-    return aws_accounts_response.get("accountList")
-
-
-def get_access_token(device_code, sso_oidc_client, oidc_app):
     try:
-        token_response = sso_oidc_client.create_token(
-            clientId=oidc_app[0],
-            clientSecret=oidc_app[1],
-            grantType="urn:ietf:params:oauth:grant-type:device_code",
-            deviceCode=device_code,
-        )
-
-        return token_response.get("accessToken")
-
-    except botocore.exceptions.ClientError as e:
-        if e.response["Error"]["Code"] != "AuthorizationPendingException":
-            raise e
+        session = get_aws_session_or_abort()
+        Config(sso=SSOAuthProvider(session)).generate_aws(file_path)
+    except PlatformException as err:
+        ClickIOProvider().abort_with_error(str(err))

dbt_platform_helper/commands/copilot.py

@@ -4,19 +4,16 @@ import click
 
 from dbt_platform_helper.domain.copilot import Copilot
 from dbt_platform_helper.domain.copilot_environment import CopilotTemplating
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.providers.config import ConfigProvider
 from dbt_platform_helper.providers.config_validator import ConfigValidator
 from dbt_platform_helper.providers.files import FileProvider
 from dbt_platform_helper.providers.io import ClickIOProvider
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
+from dbt_platform_helper.providers.kms import KMSProvider
+from dbt_platform_helper.providers.parameter_store import ParameterStore
+from dbt_platform_helper.utils.aws import get_aws_session_or_abort
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 
-# TODOs
-# Figure out a pattern for copilot templating and the new copilot domain - probably a lot of overlap here that really belongs in the copilottemplating domain instead (atleast whatever is concerned with "templating")
-# Check for E2E test coverage.
-
 
 @click.group(chain=True, cls=ClickDocOptGroup)
 def copilot():
@@ -27,7 +24,16 @@ def copilot():
 def make_addons():
     """Generate addons CloudFormation for each environment."""
     try:
+        session = get_aws_session_or_abort()
+        parameter_provider = ParameterStore(session.client("ssm"))
         config_provider = ConfigProvider(ConfigValidator())
-        Copilot(config_provider, FileProvider(), CopilotTemplating()).make_addons()
+        Copilot(
+            config_provider,
+            parameter_provider,
+            FileProvider(),
+            CopilotTemplating(),
+            KMSProvider,
+            session,
+        ).make_addons()
     except Exception as err:
         ClickIOProvider().abort_with_error(str(err))

dbt_platform_helper/commands/database.py

@@ -2,6 +2,8 @@ import click
 
 from dbt_platform_helper.commands.environment import AVAILABLE_TEMPLATES
 from dbt_platform_helper.domain.database_copy import DatabaseCopy
+from dbt_platform_helper.platform_exception import PlatformException
+from dbt_platform_helper.providers.io import ClickIOProvider
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 
 
@@ -38,9 +40,11 @@ def database():
 )
 def dump(app, from_env, database, from_vpc, filename):
     """Dump a database into an S3 bucket."""
-    data_copy = DatabaseCopy(app, database)
-    data_copy.dump(from_env, from_vpc, filename)
-    # Todo: Catch expected errors and output message
+    try:
+        data_copy = DatabaseCopy(app, database)
+        data_copy.dump(from_env, from_vpc, filename)
+    except PlatformException as err:
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @database.command(name="load")
@@ -68,9 +72,11 @@ def dump(app, from_env, database, from_vpc, filename):
 )
 def load(app, to_env, database, to_vpc, auto_approve, filename):
     """Load a database from an S3 bucket."""
-    data_copy = DatabaseCopy(app, database, auto_approve)
-    data_copy.load(to_env, to_vpc, filename)
-    # Todo: Catch expected errors and output message
+    try:
+        data_copy = DatabaseCopy(app, database, auto_approve)
+        data_copy.load(to_env, to_vpc, filename)
+    except PlatformException as err:
+        ClickIOProvider().abort_with_error(str(err))
 
 
 @database.command(name="copy")
@@ -120,6 +126,8 @@ def copy(
     no_maintenance_page,
 ):
     """Copy a database between environments."""
-    data_copy = DatabaseCopy(app, database, auto_approve)
-    data_copy.copy(from_env, to_env, from_vpc, to_vpc, svc, template, no_maintenance_page)
-    # Todo: Catch expected errors and output message
+    try:
+        data_copy = DatabaseCopy(app, database, auto_approve)
+        data_copy.copy(from_env, to_env, from_vpc, to_vpc, svc, template, no_maintenance_page)
+    except PlatformException as err:
+        ClickIOProvider().abort_with_error(str(err))

dbt_platform_helper/commands/environment.py

@@ -4,14 +4,12 @@ from dbt_platform_helper.constants import DEFAULT_TERRAFORM_PLATFORM_MODULES_VER
 from dbt_platform_helper.domain.copilot_environment import CopilotEnvironment
 from dbt_platform_helper.domain.maintenance_page import MaintenancePage
 from dbt_platform_helper.domain.terraform_environment import TerraformEnvironment
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.platform_exception import PlatformException
 from dbt_platform_helper.providers.cloudformation import CloudFormation
 from dbt_platform_helper.providers.config import ConfigProvider
 from dbt_platform_helper.providers.config_validator import ConfigValidator
 from dbt_platform_helper.providers.io import ClickIOProvider
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
 from dbt_platform_helper.providers.vpc import VpcProvider
 from dbt_platform_helper.utils.application import load_application
 from dbt_platform_helper.utils.aws import get_aws_session_or_abort
@@ -94,9 +92,10 @@ def generate(name):
     help=f"Override the default version of terraform-platform-modules. (Default version is '{DEFAULT_TERRAFORM_PLATFORM_MODULES_VERSION}').",
 )
 def generate_terraform(name, terraform_platform_modules_version):
-
+    click_io = ClickIOProvider()
     try:
-        config_provider = ConfigProvider(ConfigValidator())
+        session = get_aws_session_or_abort()
+        config_provider = ConfigProvider(ConfigValidator(session=session))
         TerraformEnvironment(config_provider).generate(name, terraform_platform_modules_version)
     except PlatformException as err:
-        ClickIOProvider().abort_with_error(str(err))
+        click_io.abort_with_error(str(err))

dbt_platform_helper/commands/generate.py

@@ -3,7 +3,7 @@ import click
 
 from dbt_platform_helper.commands.copilot import make_addons
 from dbt_platform_helper.commands.pipeline import generate as pipeline_generate
-from dbt_platform_helper.domain.versioning import RequiredVersion
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.utils.click import ClickDocOptCommand
 
 
@@ -16,7 +16,6 @@ def generate(ctx: click.Context):
 
     Wraps pipeline generate and make-addons.
     """
-
-    RequiredVersion().check_platform_helper_version_mismatch()
+    PlatformHelperVersioning().check_platform_helper_version_mismatch()
     ctx.invoke(pipeline_generate)
     ctx.invoke(make_addons)

dbt_platform_helper/commands/notify.py

@@ -2,9 +2,7 @@ import click
 from slack_sdk import WebClient
 from slack_sdk.models import blocks
 
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.utils.arn_parser import ARN
 from dbt_platform_helper.utils.click import ClickDocOptGroup
 

dbt_platform_helper/commands/pipeline.py

@@ -3,13 +3,11 @@ import click
 
 from dbt_platform_helper.constants import DEFAULT_TERRAFORM_PLATFORM_MODULES_VERSION
 from dbt_platform_helper.domain.pipelines import Pipelines
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.providers.config import ConfigProvider
 from dbt_platform_helper.providers.config_validator import ConfigValidator
 from dbt_platform_helper.providers.ecr import ECRProvider
 from dbt_platform_helper.providers.io import ClickIOProvider
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
 from dbt_platform_helper.providers.terraform_manifest import TerraformManifestProvider
 from dbt_platform_helper.utils.aws import get_codestar_connection_arn
 from dbt_platform_helper.utils.click import ClickDocOptGroup

dbt_platform_helper/commands/secrets.py

@@ -6,9 +6,7 @@ import click
 from botocore.exceptions import ClientError
 from cloudfoundry_client.client import CloudFoundryClient
 
-from dbt_platform_helper.providers.platform_helper_versioning import (
-    PlatformHelperVersioning,
-)
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.utils.application import get_application_name
 from dbt_platform_helper.utils.aws import SSM_BASE_PATH
 from dbt_platform_helper.utils.aws import get_aws_session_or_abort

dbt_platform_helper/commands/version.py

@@ -1,6 +1,6 @@
 import click
 
-from dbt_platform_helper.domain.versioning import RequiredVersion
+from dbt_platform_helper.domain.versioning import PlatformHelperVersioning
 from dbt_platform_helper.platform_exception import PlatformException
 from dbt_platform_helper.providers.io import ClickIOProvider
 from dbt_platform_helper.utils.click import ClickDocOptGroup
@@ -32,6 +32,6 @@ def get_platform_helper_for_project(pipeline):
         - Fall back on the version in the deprecated '.platform-helper-version' file
     """
     try:
-        RequiredVersion().get_required_version(pipeline)
+        PlatformHelperVersioning().get_required_version(pipeline)
     except PlatformException as err:
         ClickIOProvider().abort_with_error(str(err))