feat: add config args to duckdb client

Author: gadomski

Cargo.lock

@@ -5,7 +5,7 @@ use pyo3::{
     types::{PyDict, PyList},
 };
 use stac_api::python::{StringOrDict, StringOrList};
-use stac_duckdb::Client;
+use stac_duckdb::{Client, Config};
 use std::sync::Mutex;
 
 #[pyclass(frozen)]
@@ -14,8 +14,13 @@ pub struct DuckdbClient(Mutex<Client>);
 #[pymethods]
 impl DuckdbClient {
     #[new]
-    fn new() -> Result<DuckdbClient> {
-        let client = Client::new()?;
+    #[pyo3(signature = (use_s3_credential_chain=true, use_hive_partitioning=false))]
+    fn new(use_s3_credential_chain: bool, use_hive_partitioning: bool) -> Result<DuckdbClient> {
+        let config = Config {
+            use_s3_credential_chain,
+            use_hive_partitioning,
+        };
+        let client = Client::with_config(config)?;
         Ok(DuckdbClient(Mutex::new(client)))
     }
 

src/duckdb.rs

@@ -3,6 +3,18 @@ from typing import Any, Optional, Tuple
 class DuckdbClient:
     """A client for querying stac-geoparquet with DuckDB."""
 
+    def __init__(
+        self, use_s3_credential_chain: bool = True, use_hive_partitioning: bool = False
+    ) -> None:
+        """Creates a new duckdb client.
+
+        Args:
+            use_s3_credential_chain: If true, configures DuckDB to correctly
+                handle s3:// urls.
+            use_hive_partitioning: If true, enables queries on hive partitioned
+                geoparquet files.
+        """
+
     def search(
         self,
         href: str,

stacrs.pyi

@@ -33,3 +33,7 @@ def test_search_offset(client: DuckdbClient) -> None:
 def test_get_collections(client: DuckdbClient) -> None:
     collections = client.get_collections("data/100-sentinel-2-items.parquet")
     assert len(collections) == 1
+
+
+def test_init_with_config() -> None:
+    DuckdbClient(use_s3_credential_chain=True, use_hive_partitioning=True)