Add authenticator field to conn_args (#6983)

Co-authored-by: Dennis Tai <dennis.tai@ehealth.com>
dagster-io · Apr 11, 2022 · 7ecd4b3 · 7ecd4b3
1 parent bdb1192
commit 7ecd4b3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/python_modules/libraries/dagster-snowflake/dagster_snowflake/configs.py b/python_modules/libraries/dagster-snowflake/dagster_snowflake/configs.py
@@ -133,6 +133,12 @@ def define_snowflake_config():
         is_required=False,
     )
 
+    authenticator = Field(
+        StringSource,
+        description="Optional parameter to specify the authentication mechanism to use.",
+        is_required=False,
+    )
+
     return {
         "account": account,
         "user": user,
@@ -153,4 +159,5 @@ def define_snowflake_config():
         "connector": connector,
         "cache_column_metadata": cache_column_metadata,
         "numpy": numpy,
+        "authenticator": authenticator,
     }
diff --git a/python_modules/libraries/dagster-snowflake/dagster_snowflake/resources.py b/python_modules/libraries/dagster-snowflake/dagster_snowflake/resources.py
@@ -64,6 +64,7 @@ def __init__(self, context):  # pylint: disable=too-many-locals
                     "validate_default_parameters",
                     "paramstyle",
                     "timezone",
+                    "authenticator",
                 )
                 if context.resource_config.get(k) is not None
             }