hydopt-client 0.2.0__tar.gz → 0.3.0__tar.gz

{hydopt_client-0.2.0 → hydopt_client-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: hydopt-client
-Version: 0.2.0
+Version: 0.3.0
 Summary: Python client for the HydOpt hydropower optimization API
 Requires-Python: >=3.13
 Requires-Dist: google-auth>=2.56.3

{hydopt_client-0.2.0 → hydopt_client-0.3.0}/docker-compose.yml

@@ -2,7 +2,7 @@ name: hydopt-api
 
 services:
   api:
-    image: us-central1-docker.pkg.dev/hydopt/hydopt-backend/hydopt-api:20260910T125042-abf20a2
+    image: us-central1-docker.pkg.dev/hydopt/hydopt-backend/hydopt-api:20260922T111447-86bcc45
     ports:
       - "${PORT:-8083}:8083"
     environment:

{hydopt_client-0.2.0 → hydopt_client-0.3.0}/src/hydopt_client/client.py

@@ -17,6 +17,7 @@ from hydopt_client.models import (
     GraphLayers,
     HydoptRecord,
     KeyNumbers,
+    Link,
     ObjectStatReport,
     Order,
     parse_records,
@@ -264,6 +265,11 @@ class Client:
         response.raise_for_status()
         return response.json()["availableDates"]
 
+    def links(self) -> list[Link]:
+        response = self._request("GET", "/v1/links")
+        response.raise_for_status()
+        return [Link.model_validate(item) for item in response.json().get("links") or []]
+
     def catchments(self, asof: str) -> list[CatchmentRecord]:
         response = self._request("GET", f"/v1/catchments/{asof}")
         response.raise_for_status()

{hydopt_client-0.2.0 → hydopt_client-0.3.0}/src/hydopt_client/models.py

@@ -45,6 +45,19 @@ def parse_records(items: list[dict]) -> list[HydoptRecord]:
     return [HydoptRecord.model_validate(item) for item in items]
 
 
+class Link(BaseModel):
+    model_config = ConfigDict(alias_generator=to_camel, populate_by_name=True)
+
+    id: str
+    up: str
+    down: str
+    edge_type: str = ""
+    lat_up: float = 0.0
+    lon_up: float = 0.0
+    lat_down: float = 0.0
+    lon_down: float = 0.0
+
+
 class CatchmentRecord(BaseModel):
     model_config = ConfigDict(alias_generator=to_camel, populate_by_name=True)
 

{hydopt_client-0.2.0 → hydopt_client-0.3.0}/tests/test_client_unit.py

@@ -129,6 +129,36 @@ def test_hierarchy_layers_parses_graph_layers() -> None:
     assert layers.point_ids == ["p1"]
 
 
+def test_links_parses_watercourse_edges() -> None:
+    def handler(request: httpx.Request) -> httpx.Response:
+        assert request.url.path == "/v1/links"
+        return httpx.Response(
+            200,
+            json={
+                "links": [
+                    {
+                        "id": "a//b",
+                        "up": "A::R::Bergsdalen",
+                        "down": "B::G::Bergsdalen",
+                        "edgeType": "Release",
+                        "latUp": 60.5,
+                        "lonUp": 6.1,
+                        "latDown": 60.4,
+                        "lonDown": 6.0,
+                    }
+                ]
+            },
+        )
+
+    client, _ = make_client(handler)
+    links = client.links()
+    assert len(links) == 1
+    assert links[0].up == "A::R::Bergsdalen"
+    assert links[0].down == "B::G::Bergsdalen"
+    assert links[0].edge_type == "Release"
+    assert links[0].lat_up == 60.5
+
+
 def test_hierarchy_layers_sends_selected_nodes() -> None:
     captured = {}