@@ -6,15 +6,30 @@ class DuckdbClient:
66 """A client for querying stac-geoparquet with DuckDB."""
77
88 def __init__ (
9- self , use_s3_credential_chain : bool = True , use_hive_partitioning : bool = False
9+ self ,
10+ * ,
11+ use_s3_credential_chain : bool = True ,
12+ use_azure_credential_chain : bool = True ,
13+ use_httpfs : bool = True ,
14+ use_hive_partitioning : bool = False ,
15+ install_extensions : bool = True ,
16+ custom_extension_repository : str | None = None ,
17+ extension_directory : str | None = None ,
1018 ) -> None :
1119 """Creates a new duckdb client.
1220
1321 Args:
1422 use_s3_credential_chain: If true, configures DuckDB to correctly
1523 handle s3:// urls.
24+ use_azure_credential_chain: If true, configures DuckDB to correctly
25+ handle azure urls.
26+ use_https_credential_chain: If true, configures DuckDB to correctly
27+ handle https urls.
1628 use_hive_partitioning: If true, enables queries on hive partitioned
1729 geoparquet files.
30+ install_extensions: If true, installs extensions before loading them.
31+ custom_extension_repository: A custom extension repository to use.
32+ extension_directory: A non-standard extension directory to use.
1833 """
1934
2035 def search (
@@ -30,7 +45,7 @@ class DuckdbClient:
3045 datetime : Optional [str ] = None ,
3146 include : Optional [str | list [str ]] = None ,
3247 exclude : Optional [str | list [str ]] = None ,
33- sortby : Optional [str | list [str ]] = None ,
48+ sortby : Optional [str | list [str | dict [ str , str ] ]] = None ,
3449 filter : Optional [str | dict [str , Any ]] = None ,
3550 query : Optional [dict [str , Any ]] = None ,
3651 ** kwargs : str ,
@@ -79,7 +94,7 @@ class DuckdbClient:
7994 datetime : Optional [str ] = None ,
8095 include : Optional [str | list [str ]] = None ,
8196 exclude : Optional [str | list [str ]] = None ,
82- sortby : Optional [str | list [str ]] = None ,
97+ sortby : Optional [str | list [str | dict [ str , str ] ]] = None ,
8398 filter : Optional [str | dict [str , Any ]] = None ,
8499 query : Optional [dict [str , Any ]] = None ,
85100 ** kwargs : str ,
@@ -260,7 +275,7 @@ async def search(
260275 datetime : Optional [str ] = None ,
261276 include : Optional [str | list [str ]] = None ,
262277 exclude : Optional [str | list [str ]] = None ,
263- sortby : Optional [str | list [str ]] = None ,
278+ sortby : Optional [str | list [str | dict [ str , str ] ]] = None ,
264279 filter : Optional [str | dict [str , Any ]] = None ,
265280 query : Optional [dict [str , Any ]] = None ,
266281 use_duckdb : Optional [bool ] = None ,
@@ -329,7 +344,7 @@ async def search_to(
329344 datetime : Optional [str ] = None ,
330345 include : Optional [str | list [str ]] = None ,
331346 exclude : Optional [str | list [str ]] = None ,
332- sortby : Optional [str | list [str ]] = None ,
347+ sortby : Optional [str | list [str | dict [ str , str ] ]] = None ,
333348 filter : Optional [str | dict [str , Any ]] = None ,
334349 query : Optional [dict [str , Any ]] = None ,
335350 format : Optional [str ] = None ,
0 commit comments