From c4b4b687f8519c83b10ce9f77da4daf653875470 Mon Sep 17 00:00:00 2001 From: Dan Date: Fri, 31 Mar 2023 16:31:27 -0600 Subject: [PATCH] snowflake: add support for key password --- sqeleton/databases/snowflake.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/sqeleton/databases/snowflake.py b/sqeleton/databases/snowflake.py index 8ba6639..43098f1 100644 --- a/sqeleton/databases/snowflake.py +++ b/sqeleton/databases/snowflake.py @@ -167,9 +167,13 @@ def __init__(self, *, schema: str, **kw): with open(kw.get("key"), "rb") as key: if "password" in kw: raise ConnectError("Cannot use password and key at the same time") + if kw.get("private_key_passphrase"): + encoded_passphrase = kw.get("private_key_passphrase").encode() + else: + encoded_passphrase = None p_key = serialization.load_pem_private_key( key.read(), - password=None, + password=encoded_passphrase, backend=default_backend(), )