mdbt 0.4.27__tar.gz → 0.4.28__tar.gz

{mdbt-0.4.27 → mdbt-0.4.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mdbt
-Version: 0.4.27
+Version: 0.4.28
 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 → mdbt-0.4.28}/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 → mdbt-0.4.28}/mdbt.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mdbt
-Version: 0.4.27
+Version: 0.4.28
 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 → mdbt-0.4.28}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup
 
 setup(
     name="mdbt",
-    version="0.4.27",
+    version="0.4.28",
     description="A CLI tool to manage dbt builds with state handling and manifest management",
     author="Craig Lathrop",
     author_email="info@markimicrowave.com",