castor-extractor 0.23.2__tar.gz → 0.23.3__tar.gz

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## 0.23.3 - 2025-02-19
+
+* Snowflake : add --insecure-mode option to turn off OCSP checking
+
 ## 0.23.2 - 2025-02-17
 
 * support page_size in Tableau extraction command

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: castor-extractor
-Version: 0.23.2
+Version: 0.23.3
 Summary: Extract your metadata assets.
 Home-page: https://www.castordoc.com/
 License: EULA
@@ -100,14 +100,14 @@ It also embeds utilities to help you push your metadata to Castor:
 
 ## Installation
 
-Requirements: **python3.8+**
+Requirements: **python3.9+**
 <img src="https://upload.wikimedia.org/wikipedia/commons/c/c3/Python-logo-notext.svg" width=20 />
 
 ### Create castor-env
 
 We advise to create a dedicated [Python environment](https://docs.python.org/3/library/venv.html).
 
-Here's an example using `Pyenv` and Python `3.8.12`:
+Here's an example using `Pyenv`:
 
 - Install Pyenv
 
@@ -124,22 +124,24 @@ eval "$(pyenv init --path)"
 eval "$(pyenv virtualenv-init -)"
 ```
 
-- [optional] Install python 3.8+
+- [optional] Install python 3.9+ if needed
 
 ```bash
 pyenv versions # check your local python installations
 
-pyenv install -v 3.8.12 # if none of the installed versions satisfy requirements 8+
+# install Python if none of the installed versions satisfy requirements 3.9+
+# in this example, we will install Python 3.11.9
+pyenv install -v 3.11.9
 ```
 
 - Create your virtual env
 
 ```bash
-pyenv virtualenv 3.8.12 castor-env # create a dedicated env
+pyenv virtualenv 3.11.9 castor-env # create a dedicated env
 pyenv shell castor-env # activate the environment
 
 # optional checks
-python --version # should be `3.8.12`
+python --version # should be `3.11.9`
 pyenv version # should be `castor-env`
 ```
 
@@ -206,6 +208,10 @@ For any questions or bug report, contact us at [support@castordoc.com](mailto:su
 
 # Changelog
 
+## 0.23.3 - 2025-02-19
+
+* Snowflake : add --insecure-mode option to turn off OCSP checking
+
 ## 0.23.2 - 2025-02-17
 
 * support page_size in Tableau extraction command

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/README.md

@@ -34,14 +34,14 @@ It also embeds utilities to help you push your metadata to Castor:
 
 ## Installation
 
-Requirements: **python3.8+**
+Requirements: **python3.9+**
 <img src="https://upload.wikimedia.org/wikipedia/commons/c/c3/Python-logo-notext.svg" width=20 />
 
 ### Create castor-env
 
 We advise to create a dedicated [Python environment](https://docs.python.org/3/library/venv.html).
 
-Here's an example using `Pyenv` and Python `3.8.12`:
+Here's an example using `Pyenv`:
 
 - Install Pyenv
 
@@ -58,22 +58,24 @@ eval "$(pyenv init --path)"
 eval "$(pyenv virtualenv-init -)"
 ```
 
-- [optional] Install python 3.8+
+- [optional] Install python 3.9+ if needed
 
 ```bash
 pyenv versions # check your local python installations
 
-pyenv install -v 3.8.12 # if none of the installed versions satisfy requirements 8+
+# install Python if none of the installed versions satisfy requirements 3.9+
+# in this example, we will install Python 3.11.9
+pyenv install -v 3.11.9
 ```
 
 - Create your virtual env
 
 ```bash
-pyenv virtualenv 3.8.12 castor-env # create a dedicated env
+pyenv virtualenv 3.11.9 castor-env # create a dedicated env
 pyenv shell castor-env # activate the environment
 
 # optional checks
-python --version # should be `3.8.12`
+python --version # should be `3.11.9`
 pyenv version # should be `castor-env`
 ```
 

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/castor_extractor/commands/extract_snowflake.py

@@ -40,6 +40,11 @@ def main():
         action="store_true",
         help="Optional: will fetch transients tables if added",
     )
+    parser.add_argument(
+        "--insecure-mode",
+        action="store_true",
+        help="Optional: turns off OCSP checking",
+    )
 
     parser.add_argument("-o", "--output", help="Directory to write to")
 
@@ -65,4 +70,5 @@ def main():
         output_directory=args.output,
         skip_existing=args.skip_existing,
         fetch_transient=args.fetch_transient,
+        insecure_mode=args.insecure_mode,
     )

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/castor_extractor/warehouse/snowflake/client.py

@@ -112,6 +112,7 @@ class SnowflakeClient(SqlalchemyClient):
         role: Optional[str] = None,
     ):
         self.private_key = _private_key(credentials.get("private_key"))
+        self.insecure_mode = credentials.get("insecure_mode", False)
         super().__init__(credentials)
         self._role = role
         self._warehouse = warehouse
@@ -121,13 +122,22 @@ class SnowflakeClient(SqlalchemyClient):
         return "Snowflake"
 
     def _engine_options(self, credentials: dict) -> dict:
-        if not self.private_key:
-            return {}
-        return {
-            "connect_args": {
-                "private_key": self.private_key,
-            }
-        }
+        """
+        Sets the engine options.
+        * `insecure_mode` = True turns off OCSP checking.
+            See https://community.snowflake.com/s/article/How-to-turn-off-OCSP-checking-in-Snowflake-client-drivers
+        * `private_key` = Snowflake private key for key pair authentication.
+            See https://docs.snowflake.com/en/user-guide/key-pair-auth
+        """
+        connection_args: dict = dict()
+
+        if self.insecure_mode:
+            connection_args["insecure_mode"] = True
+
+        if self.private_key:
+            connection_args["private_key"] = self.private_key
+
+        return {"connect_args": connection_args}
 
     def _build_uri(self, credentials: dict) -> str:
         if self.private_key:

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/castor_extractor/warehouse/snowflake/credentials.py

@@ -17,6 +17,7 @@ class SnowflakeCredentials:
     account: str
     user: str
 
+    insecure_mode: Optional[bool] = field(default=False)
     password: Optional[str] = field(metadata={"sensitive": True}, default=None)
     private_key: Optional[str] = field(
         metadata={"sensitive": True}, default=None

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/castor_extractor/warehouse/snowflake/extract.py

@@ -38,6 +38,7 @@ SNOWFLAKE_ACCOUNT = "CASTOR_SNOWFLAKE_ACCOUNT"
 SNOWFLAKE_USER = "CASTOR_SNOWFLAKE_USER"
 SNOWFLAKE_PASSWORD = "CASTOR_SNOWFLAKE_PASSWORD"  # noqa: S105
 SNOWFLAKE_PRIVATE_KEY = "CASTOR_SNOWFLAKE_PRIVATE_KEY"
+SNOWFLAKE_INSECURE_MODE = "CASTOR_SNOWFLAKE_INSECURE_MODE"
 
 
 def _credentials(params: dict) -> dict:
@@ -46,9 +47,15 @@ def _credentials(params: dict) -> dict:
     private_key = params.get("private_key") or from_env(
         SNOWFLAKE_PRIVATE_KEY, True
     )
+    insecure_mode = (
+        params.get("insecure_mode")
+        or from_env(SNOWFLAKE_INSECURE_MODE, allow_missing=True)
+        or False
+    )
     common = {
         "account": params.get("account") or from_env(SNOWFLAKE_ACCOUNT),
         "user": params.get("user") or from_env(SNOWFLAKE_USER),
+        "insecure_mode": insecure_mode,
     }
     if password:
         return {**common, "password": password}

{castor_extractor-0.23.2 → castor_extractor-0.23.3}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools>=61.2"]
 
 [tool.poetry]
 name = "castor-extractor"
-version = "0.23.2"
+version = "0.23.3"
 description = "Extract your metadata assets."
 authors = ["Castor <support@castordoc.com>"]
 license = "EULA"