feat: support enable_url_table config (apache#980)

chenkovsky · timsaucer · web-flow · commit 389164aa90c8 · 2025-01-08T16:39:31.000-05:00
* feat: support enable_url_table config

* change enable_url_table as method

* Remove whitespace

---------

Co-authored-by: Tim Saucer &lt;timsaucer@gmail.com&gt;
diff --git a/examples/create-context.py b/examples/create-context.py
@@ -37,3 +37,6 @@
 )
 ctx = SessionContext(config, runtime)
 print(ctx)
+
+ctx = ctx.enable_url_table()
+print(ctx)
diff --git a/python/datafusion/context.py b/python/datafusion/context.py
@@ -481,6 +481,17 @@ def __init__(
 
         self.ctx = SessionContextInternal(config, runtime)
 
+    def enable_url_table(self) -> "SessionContext":
+        """Control if local files can be queried as tables.
+
+        Returns:
+            A new :py:class:`SessionContext` object with url table enabled.
+        """
+        klass = self.__class__
+        obj = klass.__new__(klass)
+        obj.ctx = self.ctx.enable_url_table()
+        return obj
+
     def register_object_store(
         self, schema: str, store: Any, host: str | None = None
     ) -> None:
diff --git a/src/context.rs b/src/context.rs
@@ -299,6 +299,12 @@ impl PySessionContext {
         })
     }
 
+    pub fn enable_url_table(&self) -> PyResult<Self> {
+        Ok(PySessionContext {
+            ctx: self.ctx.clone().enable_url_table(),
+        })
+    }
+
     /// Register an object store with the given name
     #[pyo3(signature = (scheme, store, host=None))]
     pub fn register_object_store(

Original file line number	Diff line number	Diff line change
`@@ -37,3 +37,6 @@`
`37`	`37`	`)`
`38`	`38`	`ctx = SessionContext(config, runtime)`
`39`	`39`	`print(ctx)`
	`40`	`+`
	`41`	`+ctx = ctx.enable_url_table()`
	`42`	`+print(ctx)`