diff --git a/cosmos/profiles/trino/base.py b/cosmos/profiles/trino/base.py index 1e7e84124..3ed5c7e6a 100644 --- a/cosmos/profiles/trino/base.py +++ b/cosmos/profiles/trino/base.py @@ -13,16 +13,19 @@ class TrinoBaseProfileMapping(BaseProfileMapping): dbt_profile_type: str = "trino" is_community: bool = True - required_fields = [ + base_fields = [ "host", "database", "schema", "port", ] + required_fields = base_fields + ["user"] + airflow_param_mapping = { "host": "host", "port": "port", + "user": "login", "session_properties": "extra.session_properties", } diff --git a/cosmos/profiles/trino/certificate.py b/cosmos/profiles/trino/certificate.py index b792664a4..c87e7c67c 100644 --- a/cosmos/profiles/trino/certificate.py +++ b/cosmos/profiles/trino/certificate.py @@ -15,7 +15,7 @@ class TrinoCertificateProfileMapping(TrinoBaseProfileMapping): dbt_profile_method: str = "certificate" - required_fields = TrinoBaseProfileMapping.required_fields + [ + required_fields = TrinoBaseProfileMapping.base_fields + [ "client_certificate", "client_private_key", ] diff --git a/cosmos/profiles/trino/jwt.py b/cosmos/profiles/trino/jwt.py index 5d6d98908..4f09cd3f1 100644 --- a/cosmos/profiles/trino/jwt.py +++ b/cosmos/profiles/trino/jwt.py @@ -16,7 +16,7 @@ class TrinoJWTProfileMapping(TrinoBaseProfileMapping): dbt_profile_method: str = "jwt" - required_fields = TrinoBaseProfileMapping.required_fields + [ + required_fields = TrinoBaseProfileMapping.base_fields + [ "jwt_token", ] secret_fields = [ diff --git a/cosmos/profiles/trino/ldap.py b/cosmos/profiles/trino/ldap.py index d456f7c68..5ba122b93 100644 --- a/cosmos/profiles/trino/ldap.py +++ b/cosmos/profiles/trino/ldap.py @@ -16,7 +16,7 @@ class TrinoLDAPProfileMapping(TrinoBaseProfileMapping): dbt_profile_method: str = "ldap" - required_fields = TrinoBaseProfileMapping.required_fields + [ + required_fields = TrinoBaseProfileMapping.base_fields + [ "user", "password", ] diff --git a/tests/profiles/trino/test_trino_base.py b/tests/profiles/trino/test_trino_base.py index 31f1a3166..ad1e84748 100644 --- a/tests/profiles/trino/test_trino_base.py +++ b/tests/profiles/trino/test_trino_base.py @@ -40,6 +40,7 @@ def test_profile_args() -> None: "schema": "my_schema", "host": "my_host", "port": 8080, + "user": "my_login", "session_properties": {"my_property": "my_value"}, } @@ -80,5 +81,6 @@ def test_profile_args_overrides() -> None: "schema": "my_schema", "host": "my_host_override", "port": 8080, + "user": "my_login", "session_properties": {"my_property": "my_value_override"}, }