feat: set settings (#22)

This lets us set the base url.

- Closes #21

Author: gadomski

Cargo.lock

@@ -9,8 +9,8 @@ name = "pgstacrs"
 crate-type = ["cdylib"]
 
 [dependencies]
-bb8 = "0.8.6"
-bb8-postgres = "0.8.1"
+bb8 = "0.9.0"
+bb8-postgres = "0.9.0"
 geojson = "0.24.1"
 pgstac = { version = "0.2.2", git = "https://github.com/stac-utils/stac-rs" }
 pyo3 = "0.23.2"
@@ -21,7 +21,7 @@ pyo3-async-runtimes = { version = "0.23.0", features = [
 pyo3-log = "0.12.1"
 pythonize = "0.23.0"
 serde_json = "1.0.134"
-stac-api = { version = "0.6.2", features = [
+stac-api = { version = "0.7.0", features = [
     "python",
 ], git = "https://github.com/stac-utils/stac-rs" }
 stac = { version = "0.11.0", git = "https://github.com/stac-utils/stac-rs" }

Cargo.toml

@@ -60,12 +60,21 @@ class Client:
                 It is recommended to use filter instead, if possible.
             limit: The page size returned from the server.
         """
+
     async def print_config(self) -> None:
         """Prints the postgresql configuration.
 
         Redacts the password
         """
 
+    async def set_setting(self, key: str, value: str) -> None:
+        """Sets a pgstac setting.
+
+        Args:
+            key: The setting name, e.g. `base_url`
+            value: The setting value, e.g. `http://pgstacrs.test`
+        """
+
     async def get_version(self) -> str:
         """Returns the pgstac version.
 

pgstacrs.pyi

@@ -104,6 +104,19 @@ impl Client {
         })
     }
 
+    fn set_setting<'a>(
+        &self,
+        py: Python<'a>,
+        key: String,
+        value: String,
+    ) -> PyResult<Bound<'a, PyAny>> {
+        self.run(py, |pool| async move {
+            let connection = pool.get().await?;
+            connection.set_pgstac_setting(&key, &value).await?;
+            Ok(())
+        })
+    }
+
     fn get_collection<'a>(&self, py: Python<'a>, id: String) -> PyResult<Bound<'a, PyAny>> {
         self.run(py, |pool| async move {
             let connection = pool.get().await?;

src/lib.rs

@@ -26,6 +26,30 @@ async def test_empty_search(client: Client) -> None:
         }
 
 
+async def test_base_url(client: Client) -> None:
+    await client.set_setting("base_url", "http://pgstacrs.test")
+    search = await client.search()
+    version = await client.get_version()
+    if version.startswith("0.9"):
+        assert search == {
+            "features": [],
+            "links": [
+                {
+                    "href": "http://pgstacrs.test/",
+                    "rel": "root",
+                    "type": "application/json",
+                },
+                {
+                    "href": "http://pgstacrs.test/search",
+                    "rel": "self",
+                    "type": "application/json",
+                },
+            ],
+            "numberReturned": 0,
+            "type": "FeatureCollection",
+        }
+
+
 async def test_search(
     client: Client, collection: dict[str, Any], item: dict[str, Any]
 ) -> None: