This is an automated email from the ASF dual-hosted git repository.

timsaucer pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/datafusion-python.git


The following commit(s) were added to refs/heads/main by this push:
     new eaa3f79b Add use_fabric_endpoint parameter to MicrosoftAzure class 
(#1357)
eaa3f79b is described below

commit eaa3f79b998fc433e930fc74ba648372d59e6ace
Author: Mimoune <[email protected]>
AuthorDate: Sat Jan 31 01:32:45 2026 +1000

    Add use_fabric_endpoint parameter to MicrosoftAzure class (#1357)
    
    This change adds support for the use_fabric_endpoint parameter to the
    MicrosoftAzure object store class, enabling connections to Microsoft
    Fabric OneLake storage.
    
    The parameter allows users to specify that they want to use Data Lake
    Storage Gen2 endpoints (dfs.fabric.microsoft.com) instead of the default
    Azure Blob Storage endpoints (blob.core.windows.net), which is required
    for OneLake/Fabric storage access.
    
    Implementation follows the same pattern as existing boolean parameters
    (use_emulator, allow_http) by:
    - Adding the parameter to the PyO3 signature macro
    - Adding it as Option<bool> to the function parameters
    - Conditionally calling with_use_fabric_endpoint() on the builder
    
    Fixes #1356
    
    Co-authored-by: Claude Sonnet 4.5 <[email protected]>
---
 src/store.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/store.rs b/src/store.rs
index dcbcbd32..3eae866b 100644
--- a/src/store.rs
+++ b/src/store.rs
@@ -76,7 +76,7 @@ pub struct PyMicrosoftAzureContext {
 #[pymethods]
 impl PyMicrosoftAzureContext {
     #[allow(clippy::too_many_arguments)]
-    #[pyo3(signature = (container_name, account=None, access_key=None, 
bearer_token=None, client_id=None, client_secret=None, tenant_id=None, 
sas_query_pairs=None, use_emulator=None, allow_http=None))]
+    #[pyo3(signature = (container_name, account=None, access_key=None, 
bearer_token=None, client_id=None, client_secret=None, tenant_id=None, 
sas_query_pairs=None, use_emulator=None, allow_http=None, 
use_fabric_endpoint=None))]
     #[new]
     fn new(
         container_name: String,
@@ -89,6 +89,7 @@ impl PyMicrosoftAzureContext {
         sas_query_pairs: Option<Vec<(String, String)>>,
         use_emulator: Option<bool>,
         allow_http: Option<bool>,
+        use_fabric_endpoint: Option<bool>,
     ) -> Self {
         let mut builder = 
MicrosoftAzureBuilder::from_env().with_container_name(&container_name);
 
@@ -127,6 +128,10 @@ impl PyMicrosoftAzureContext {
             builder = builder.with_allow_http(allow_http);
         }
 
+        if let Some(use_fabric_endpoint) = use_fabric_endpoint {
+            builder = builder.with_use_fabric_endpoint(use_fabric_endpoint);
+        }
+
         Self {
             inner: Arc::new(
                 builder


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to