yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-24 16:35:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

parametrize get private key for snowflake (#22214)

Browse Source
This commit is contained in:
IceS2 2025-07-08 15:47:00 +02:00 committed by GitHub
parent 00b8aa03fd
commit 2d469e1b00
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ingestion/src/metadata/ingestion/source/database/snowflake/connection.py
View File

@ -151,7 +151,9 @@ class SnowflakeConnection(BaseConnection[SnowflakeConnectionConfig, Engine]):
url = f"{url}?{params}" url = f"{url}?{params}"
return url return url
def _get_private_key(self) -> Optional[bytes]: def _get_private_key(
self, encoding: serialization.Encoding = serialization.Encoding.DER
) -> Optional[bytes]:
connection = self.service_connection connection = self.service_connection
if connection.privateKey: if connection.privateKey:
snowflake_private_key_passphrase = ( snowflake_private_key_passphrase = (
@ -170,7 +172,7 @@ class SnowflakeConnection(BaseConnection[SnowflakeConnectionConfig, Engine]):
backend=default_backend(), backend=default_backend(),
) )
pkb = p_key.private_bytes( pkb = p_key.private_bytes(
encoding=serialization.Encoding.DER, encoding=encoding,
format=serialization.PrivateFormat.PKCS8, format=serialization.PrivateFormat.PKCS8,
encryption_algorithm=serialization.NoEncryption(), encryption_algorithm=serialization.NoEncryption(),
) )