From 01336853ef1bf33bd922325dcf91db61905f1196 Mon Sep 17 00:00:00 2001
From: codingwithabhi <63392662+codingwithabhi@users.noreply.github.com>
Date: Mon, 11 Apr 2022 19:27:50 +0530
Subject: [PATCH] Snowflake private key fix (#4031)

* snowflake-private-key-fixed

* ran-core_bump_version_dev
---
 ingestion-core/src/metadata/_version.py              | 2 +-
 ingestion/examples/workflows/snowflake.json          | 5 ++++-
 ingestion/src/metadata/ingestion/source/snowflake.py | 6 +++---
 3 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/ingestion-core/src/metadata/_version.py b/ingestion-core/src/metadata/_version.py
index d8870de4bd2..315c1541753 100644
--- a/ingestion-core/src/metadata/_version.py
+++ b/ingestion-core/src/metadata/_version.py
@@ -7,5 +7,5 @@ Provides metadata version information.
 
 from incremental import Version
 
-__version__ = Version("metadata", 0, 9, 0, dev=20)
+__version__ = Version("metadata", 0, 9, 0, dev=21)
 __all__ = ["__version__"]
diff --git a/ingestion/examples/workflows/snowflake.json b/ingestion/examples/workflows/snowflake.json
index 314f95385c4..bc4daa0aeb0 100644
--- a/ingestion/examples/workflows/snowflake.json
+++ b/ingestion/examples/workflows/snowflake.json
@@ -10,7 +10,10 @@
         "password": "password",
         "database": "database_name",
         "warehouse": "warehouse_name",
-        "account": "account.region_name.cloud_service"
+        "account": "account.region_name.cloud_service",
+        "connectionArguments": {
+          "private_key": "private_key" 
+        }
       }
     },
     "sourceConfig": {
diff --git a/ingestion/src/metadata/ingestion/source/snowflake.py b/ingestion/src/metadata/ingestion/source/snowflake.py
index dfe439b71e4..3683982a424 100644
--- a/ingestion/src/metadata/ingestion/source/snowflake.py
+++ b/ingestion/src/metadata/ingestion/source/snowflake.py
@@ -62,9 +62,9 @@ class SnowflakeSource(SQLSource):
                     format=serialization.PrivateFormat.PKCS8,
                     encryption_algorithm=serialization.NoEncryption(),
                 )
-            config.serviceConnection.__root__.config.connectionArguments.private_key = (
-                pkb
-            )
+                config.serviceConnection.__root__.config.connectionArguments.private_key = (
+                    pkb
+                )
 
         super().__init__(config, metadata_config)