Snowflake provider not accepting private_key_file in extras

Apache Airflow Provider(s)

snowflake

Versions of Apache Airflow Providers

2.1.1

Apache Airflow version

2.1.3 (latest released)

Operating System

centos7

Deployment

Virtualenv installation

Deployment details

Created a virtualenv, did a pip install.

What happened

I am adding the dictionary below to the extras in snowflake connections. After I click save and try to access this connection, I get a response that the password field is empty. This leads me to believe that the dictionary key is not being picked up.

{ "private_key_file": "/etc/key/key.pk8" }

According to https://github.com/snowflakedb/snowflake-connector-python/blob/master/src/snowflake/connector/connection.py, there is a KEY_PAIR_AUTHENTICATOR authenticator as well, but I am not sure if this is required to be passed as using the standard sqlalchemy I do not need to pass the authenticator.

What you expected to happen

With the private_key_file provided, the session should connect without any errors.

How to reproduce

Install apache-airflow-providers-snowflake 2.1.1

Go to connections Add Snowflake In extras field, enter the dictionary, click save.

Anything else

Looks like it might be the same issue as https://github.com/apache/airflow/pull/16735.

Are you willing to submit PR?

• Yes I am willing to submit a PR!

Code of Conduct

• I agree to follow this project’s Code of Conduct

[weiminmei]

Hi,

On version 2.2.0b1, looks like this is addressed.

[eladkal]

@weiminmei Can you please check if the issue still happens in 2.2.0b1 ? This is a beta release that includes https://github.com/apache/airflow/pull/17269 which should solve your issue.