icsDataValidation 1.0.280__py3-none-any.whl → 1.0.285__py3-none-any.whl

icsDataValidation/connection_setups/snowflake_connection_setup.py

@@ -2,6 +2,10 @@ import os
 
 from dotenv import load_dotenv
 from pathlib import Path
+from cryptography.hazmat.backends import default_backend
+from cryptography.hazmat.primitives.asymmetric import rsa
+from cryptography.hazmat.primitives.asymmetric import dsa
+from cryptography.hazmat.primitives import serialization
 
 #########################################################################################
 #########################################################################################
@@ -11,10 +15,36 @@ def load_snowflake_credentials(system_configs:dict,system_selection:str)->dict:
     snowflake_params = {
         "account"   : system_configs[system_selection]["ACCOUNT"],
         "user"      : system_configs[system_selection]["USER"],
-        "password"  : os.getenv(system_configs[system_selection]["PASSWORD_NAME"]),
         "warehouse" : system_configs[system_selection]["WAREHOUSE"],
         "role"      : system_configs[system_selection]["ROLE"],
         "database"  : system_configs[system_selection]["DATABASE"]
     }
 
+    password_name = os.getenv(system_configs[system_selection]["PASSWORD_NAME"])
+    private_key_name = os.getenv(system_configs[system_selection]["PRIVATE_KEY_NAME"])
+    private_key_passphrase_name = os.getenv(system_configs[system_selection]["PRIVATE_KEY_PASSPHRASE_NAME"])
+
+    if password_name:
+        snowflake_params['password'] = password_name
+    elif private_key_name:
+        if private_key_passphrase_name:
+            # if private key is encrypted it is decrypted here with provided passphrase
+            p_key = serialization.load_pem_private_key(
+                private_key_name,
+                password = private_key_passphrase_name,
+                backend = default_backend()
+            )
+
+            decrypted_p_key = p_key.private_bytes(
+            encoding = serialization.Encoding.DER,
+            format = serialization.PrivateFormat.PKCS8,
+            encryption_algorithm = serialization.NoEncryption())
+
+            snowflake_params['private_key'] = decrypted_p_key
+        else:
+            # otherwise use the provided not encrypted private key
+            snowflake_params['private_key'] = private_key_name
+    else:
+        raise ValueError("No valid authentication method found. Provide either PASSWORD_NAME or PRIVATE_KEY_NAME.")
+
     return snowflake_params

{icsDataValidation-1.0.280.dist-info → icsDataValidation-1.0.285.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: icsDataValidation
-Version: 1.0.280
+Version: 1.0.285
 Summary: ics data validation
 Home-page: https://initions.com/
 Author: initions

{icsDataValidation-1.0.280.dist-info → icsDataValidation-1.0.285.dist-info}/RECORD

@@ -7,7 +7,7 @@ icsDataValidation/connection_setups/azure_connection_setup.py,sha256=gvTyctG63ol
 icsDataValidation/connection_setups/databricks_connection_setup.py,sha256=dNEBum-8R-TUW2SCEk3CaNtCr_gLFvn456KBlENpgJU,1220
 icsDataValidation/connection_setups/exasol_connection_setup.py,sha256=RfCUsL6G-NaOW-qNK-3SfHcljbRaKD6fDIHXkNQhClk,590
 icsDataValidation/connection_setups/oracle_connection_setup.py,sha256=D-4ucC1ChE4HYm93ECIEg_yBOrn1NkknxFBgFRGFmWs,978
-icsDataValidation/connection_setups/snowflake_connection_setup.py,sha256=JDTdIM0bQ0_IV0HnCct90RC6Mq4fS1sfh9IJc-YYlMo,804
+icsDataValidation/connection_setups/snowflake_connection_setup.py,sha256=8gBtyEfU0GXwDhy1HHgV9XNJpgVH7-3CMbWo_nJYsSU,2220
 icsDataValidation/connection_setups/teradata_connection_setup.py,sha256=fIpuxz-FTqFK2vSMSuokqU9sdJkaJ4UP5piY_zIbj5k,624
 icsDataValidation/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 icsDataValidation/core/database_objects.py,sha256=2oaDaVQajSYI_HJjJy1pmc6FsoK_wMfwgu6ZgEcFvow,523
@@ -36,7 +36,7 @@ icsDataValidation/utils/logger_util.py,sha256=xS48_FFMot_hyQgJY8DUeRTn5jpdvRt5QI
 icsDataValidation/utils/pandas_util.py,sha256=D_g7Xw7BIS2E-1ZhJIvp62K5xuKjIkj-7TxH4HN_8SI,6505
 icsDataValidation/utils/parallelization_util.py,sha256=6P0YcQLmunW_fHR4f5-kdncZbOlxxqKyk6ZAFQQEd2k,2088
 icsDataValidation/utils/sql_util.py,sha256=0c-BInElSsRmXUedfLP_h9Wsiscv9aic7IIc5f15Uzo,396
-icsDataValidation-1.0.280.dist-info/METADATA,sha256=IOQsmO4gYzNV29pKG_wb8twh03GllVZMjV0lVwD9SBI,720
-icsDataValidation-1.0.280.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-icsDataValidation-1.0.280.dist-info/top_level.txt,sha256=YL9V1qreCXZeUCy-tzA4Vxv5-6mvXy5lsfAT0nQapfg,53
-icsDataValidation-1.0.280.dist-info/RECORD,,
+icsDataValidation-1.0.285.dist-info/METADATA,sha256=0ymdUuzh6Q-S-o8n5YeRa4PZOHE00py3Eu7yCwNo2V0,720
+icsDataValidation-1.0.285.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+icsDataValidation-1.0.285.dist-info/top_level.txt,sha256=YL9V1qreCXZeUCy-tzA4Vxv5-6mvXy5lsfAT0nQapfg,53
+icsDataValidation-1.0.285.dist-info/RECORD,,