mdbt 0.4.27__py3-none-any.whl → 0.4.29__py3-none-any.whl

mdbt/core.py

@@ -5,6 +5,8 @@ import subprocess
 import sys
 import typing as t
 
+from cryptography.hazmat.backends import default_backend
+from cryptography.hazmat.primitives import serialization
 import snowflake.connector as snow
 from dotenv import find_dotenv
 from dotenv import load_dotenv
@@ -27,18 +29,27 @@ class Core:
         self.dbt_execute_command_output = ""
 
     def _create_snowflake_connection(self):
-        if not os.environ.get("SNOWFLAKE_MAIN_ACCOUNT"):
-            raise ValueError(
-                "SNOWFLAKE_MAIN_ACCOUNT environment variable is not set"
-            )
+        rsa_key_str = os.environ.get('MAIN__RSA_KEY')
+        if not rsa_key_str:
+            raise ValueError('MAIN__RSA_KEY environment variable not set')
+        # The RSA key from the environment variable may be in single line or multi-line string
+        if '\\n' in rsa_key_str:
+            rsa_key_str = rsa_key_str.replace('\\n', '\n')
+        rsa_key_bytes = rsa_key_str.encode()
+        # Load private key object
+        private_key = serialization.load_pem_private_key(
+            rsa_key_bytes,
+            password=None,
+            backend=default_backend(),
+        )
         self._conn = snow.connect(
-            account=os.environ.get("SNOWFLAKE_MAIN_ACCOUNT"),
-            password=os.environ.get("SNOWFLAKE_MAIN_PASSWORD"),
-            schema=os.environ.get("SNOWFLAKE_MAIN_SCHEMA"),
-            user=os.environ.get("SNOWFLAKE_MAIN_USER"),
-            warehouse=os.environ.get("SNOWFLAKE_MAIN_WAREHOUSE"),
-            database=os.environ.get("SNOWFLAKE_MAIN_DATABASE"),
-            role=os.environ.get("SNOWFLAKE_MAIN_ROLE"),
+            user=os.getenv('MAIN__USER'),
+            account=os.getenv('MAIN__ACCOUNT'),
+            database=os.getenv('MAIN__DATABASE'),
+            private_key=private_key,
+            schema=os.getenv('MAIN__SCHEMA'),
+            warehouse=os.getenv('MAIN__WAREHOUSE'),
+            role=os.getenv('MAIN__ROLE'),
         )
 
         self._cur = self._conn.cursor()

{mdbt-0.4.27.dist-info → mdbt-0.4.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mdbt
-Version: 0.4.27
+Version: 0.4.29
 Summary: A CLI tool to manage dbt builds with state handling and manifest management
 Author: Craig Lathrop
 Author-email: info@markimicrowave.com

{mdbt-0.4.27.dist-info → mdbt-0.4.29.dist-info}/RECORD

@@ -3,7 +3,7 @@ mdbt/ai_core.py,sha256=diJFT35pGOEIQpUYOk6GFBgQgmK2sHj_F_kfkjeaEyQ,3963
 mdbt/build_dbt_docs_ai.py,sha256=BOuFBUJyf2l7pOzw9LSTXc5Gk-8LoGqKKDRE4Jgx0p0,5309
 mdbt/build_unit_test_data_ai.py,sha256=OqJXDerhg54_QVIlVtXNxfk5TKVe68i6JnrV_8iEDrc,4292
 mdbt/cmdline.py,sha256=meNATu3BzP_4Htt5VcoT923mlh9NsfK8og0JQgn9PCE,10822
-mdbt/core.py,sha256=VZhIwJEPm33hhTyYn9v7EAjXU0aDVkRsxbozz3G6OAI,4109
+mdbt/core.py,sha256=Y1K7ecJTZ9Bfkvk7Kk7-DRO1-QJs9GD6RP6m7sfQ-oY,4522
 mdbt/expectations_output_builder.py,sha256=AXKEM-WO7FecYzfMLwzsOnQnVf7AiHBi_khyidE2lJs,3195
 mdbt/lightdash.py,sha256=qJBJ-pc5mN8GBA1MZElRhtA8aldrX-AgvHtha4iOA-Y,2745
 mdbt/main.py,sha256=UO3r7zOXmVpjnAIz2eeZVeQMFmgXa698Gm42Wo2qhRU,16939
@@ -12,9 +12,9 @@ mdbt/prompts.py,sha256=2vCvh9hamgop92kGGaMKtap11F2MZiM7hHKjcwX4lhQ,13992
 mdbt/recce.py,sha256=P14CvWd7lRgTPUW7BVMLmcclSqby-_uSgpoI3r5VjTA,2327
 mdbt/sort_yaml_fields.py,sha256=1PROsrz8KubSr0bVmPq3oAw4V-eSiZh1uGRR5B-uaA4,5648
 mdbt/sql_sorter.py,sha256=8bd6svrtcXp7ePT2g4FTGLTW55qbsVjXgUmba7L-G-4,6467
-mdbt-0.4.27.dist-info/licenses/LICENSE,sha256=DrJpgQEYhttwpwcE56BzrGZ1aEfR_tqfaxsI5NlsYOE,1072
-mdbt-0.4.27.dist-info/METADATA,sha256=nYnEobjgoL3XJcryk3GBVzLZgxniFvp6J7u-w55sZzU,920
-mdbt-0.4.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mdbt-0.4.27.dist-info/entry_points.txt,sha256=fVft1CYVP2MtZMMtsXN67S1T5RszfgKtAuaeoKLdCow,43
-mdbt-0.4.27.dist-info/top_level.txt,sha256=-PP7vAl9EgVjRTzBovElczsPNjOfja6kjZssNmv5vo0,5
-mdbt-0.4.27.dist-info/RECORD,,
+mdbt-0.4.29.dist-info/licenses/LICENSE,sha256=DrJpgQEYhttwpwcE56BzrGZ1aEfR_tqfaxsI5NlsYOE,1072
+mdbt-0.4.29.dist-info/METADATA,sha256=-vcHygz-kLA0k4vMO4LotYRbfGQci-fdOayHkQNWOng,920
+mdbt-0.4.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mdbt-0.4.29.dist-info/entry_points.txt,sha256=fVft1CYVP2MtZMMtsXN67S1T5RszfgKtAuaeoKLdCow,43
+mdbt-0.4.29.dist-info/top_level.txt,sha256=-PP7vAl9EgVjRTzBovElczsPNjOfja6kjZssNmv5vo0,5
+mdbt-0.4.29.dist-info/RECORD,,