Merge pull request #7 from keboola/fix/publishing

Fix/publishing

Author: AlbSed

src/dbstorage/snowflake_client.py

@@ -70,7 +70,7 @@ def _create_snfk_connection(self, config: dict, session_parameters: dict = None)
         auth_type = config.get("auth_type", "key_pair")
         logging.info(f"Using authentication type: {auth_type}")
 
-        if auth_type == "password":
+        if auth_type != "key_pair" or not config.get("private_key"):
             try:
                 connection = snowflake.connector.connect(
                     user=config["user"],
@@ -110,7 +110,6 @@ def _create_snfk_connection(self, config: dict, session_parameters: dict = None)
                 format=serialization.PrivateFormat.PKCS8,
                 encryption_algorithm=serialization.NoEncryption(),
             )
-
             try:
                 connection = snowflake.connector.connect(
                     user=config["user"],
@@ -126,9 +125,11 @@ def _create_snfk_connection(self, config: dict, session_parameters: dict = None)
                     "Snowflake connection created successfully with key_pair authentication"
                 )
                 return connection
+
             except Exception as e:
                 logging.error(
-                    "Failed to create Snowflake connection with key_pair: %s", str(e)
+                    "Failed to create Snowflake connection with key_pair: %s",
+                    str(e),
                 )
                 raise