fdc-shared-kernel 0.0.188__tar.gz → 0.0.190__tar.gz

{fdc_shared_kernel-0.0.188 → fdc_shared_kernel-0.0.190}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fdc_shared_kernel
-Version: 0.0.188
+Version: 0.0.190
 Summary: Shared library for microservice
 Author-email: Shikhil S <shikhil.s@dbizsolution.com>, Ahammed Akdham N <ahammedakdham.n@dbizsolution.com>
 Classifier: Programming Language :: Python :: 3

{fdc_shared_kernel-0.0.188 → fdc_shared_kernel-0.0.190}/fdc_shared_kernel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fdc_shared_kernel
-Version: 0.0.188
+Version: 0.0.190
 Summary: Shared library for microservice
 Author-email: Shikhil S <shikhil.s@dbizsolution.com>, Ahammed Akdham N <ahammedakdham.n@dbizsolution.com>
 Classifier: Programming Language :: Python :: 3

{fdc_shared_kernel-0.0.188 → fdc_shared_kernel-0.0.190}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fdc_shared_kernel"
-version = "0.0.188"
+version = "0.0.190"
 requires-python = ">=3.7"
 readme = "README_pypi.md"
 description = "Shared library for microservice"

{fdc_shared_kernel-0.0.188 → fdc_shared_kernel-0.0.190}/shared_kernel/data_warehouse_handlers/connections/jdbc_warehouse_connection.py

@@ -81,8 +81,14 @@ class JDBCWarehouseConnection(DataWarehouseConnection):
                 try:
                     # Note: startJVM can only be called ONCE. 
                     # We load all JARs now to support future "plug-and-play" connections.
-                    jpype.startJVM(jpype.getDefaultJVMPath(), classpath=classpath, convertStrings=True)
-                    logger.info("JVM started for JDBC connections.")
+                    jpype.startJVM(
+                        jpype.getDefaultJVMPath(),
+                        "-Xmx512m",
+                        "-Xms128m",
+                        classpath=classpath,
+                        convertStrings=True
+                    )
+                    logger.info("JVM started for JDBC connections with memory constraints.")
                 except Exception as e:
                     logger.error(f"Failed to start JVM: {e}")
 

{fdc_shared_kernel-0.0.188 → fdc_shared_kernel-0.0.190}/shared_kernel/data_warehouse_handlers/connections/snowflake_warehouse_connection.py

@@ -2,6 +2,12 @@ import time
 import snowflake.connector
 
 from contextlib import contextmanager
+from cryptography.hazmat.primitives.serialization import (
+    load_pem_private_key,
+    Encoding,
+    PrivateFormat,
+    NoEncryption,
+)
 
 from snowflake.connector.errors import (
     DatabaseError,
@@ -47,13 +53,25 @@ class SnowflakeWarehouseConnection(DataWarehouseConnection):
 
     def _create_connection(self):
         cfg = self._snowflake_config
-        self._conn = snowflake.connector.connect(
-            user=cfg.user,
-            password=cfg.password,
-            account=cfg.account,
-            database=cfg.database,
-            schema=cfg.schema,
-        )
+        if cfg.private_key:
+            p_key = load_pem_private_key(cfg.private_key.encode(), password=None)
+            pkb = p_key.private_bytes(Encoding.DER, PrivateFormat.PKCS8, NoEncryption())
+            self._conn = snowflake.connector.connect(
+                user=cfg.user,
+                account=cfg.account,
+                database=cfg.database,
+                schema=cfg.schema,
+                private_key=pkb,
+                authenticator="snowflake_jwt",
+            )
+        else:
+            self._conn = snowflake.connector.connect(
+                user=cfg.user,
+                password=cfg.password,
+                account=cfg.account,
+                database=cfg.database,
+                schema=cfg.schema,
+            )
         self._cursor = self._conn.cursor()
 
     def _create_connection_with_retry(self, retries: int = 3, delay: int = 5):

{fdc_shared_kernel-0.0.188 → fdc_shared_kernel-0.0.190}/shared_kernel/data_warehouse_handlers/snowflake_handler.py

@@ -17,16 +17,24 @@ class SnowflakeWarehouseHandler(WarehouseHandler):
         if source_config and isinstance(source_config, SnowflakeConfig):
             return source_config
 
-        required = ("user", "password", "account", "database")
+        required = ("user", "account", "database")
         missing = [f for f in required if not self.payload.get(f)]
         if missing:
             raise ValueError(
                 f"Snowflake payload is missing required fields: {missing}."
             )
 
+        has_password = bool(self.payload.get("password"))
+        has_private_key = bool(self.payload.get("private_key"))
+        if not has_password and not has_private_key:
+            raise ValueError(
+                "Snowflake payload must include either 'password' or 'private_key'."
+            )
+
         return SnowflakeConfig(
             user=str(self.payload.get("user")).strip(),
-            password=str(self.payload.get("password")).strip(),
+            password=str(self.payload.get("password")).strip() if has_password else None,
+            private_key=str(self.payload.get("private_key")).strip() if has_private_key else None,
             account=str(self.payload.get("account")).strip(),
             database=str(self.payload.get("database")).strip(),
             schema=str(self.payload.get("schema")).strip() if self.payload.get("schema") else None,

{fdc_shared_kernel-0.0.188 → fdc_shared_kernel-0.0.190}/shared_kernel/dataclasses/warehouse_configs.py

@@ -92,9 +92,10 @@ class SnowflakeConfig(BaseConfig):
     """Configuration for Snowflake connections."""
 
     user: str
-    password: str
     account: str
     database: str
+    password: Optional[str] = None
+    private_key: Optional[str] = None
     default_schema: str = "PUBLIC"