[tracer] Add get and set grants functions

Author: ao-picterra

src/picterra/tracer_client.py

@@ -26,6 +26,8 @@
     _download_to_file,
 )
 
+GrantKind = Literal["plots_group", "plots_analysis", "methodology"]
+
 
 class TracerClient(BaseAPIClient):
     def __init__(self, **kwargs):
@@ -576,3 +578,45 @@ def get_plots_analysis_report(
         )
         _check_resp_is_ok(resp, "Failed to get plots analysis report")
         return resp.json()
+
+    def get_authorization_grants(self, kind: GrantKind, resource_id: str):
+        """
+        **beta** function. Get the authorization grants for a given resource.
+
+        Args:
+            kind: The kind of resource to get the grants for
+            resource_id: The ID of the resource
+
+        Returns:
+            dict: A dictionary containing the authorization grants for the resource.
+                  See https://app.picterra.ch/public/apidocs/plots_analysis/v1/#tag/authorization/operation/getGrants
+        """
+        resp = self.sess.get(
+            self._full_url("authorization/grants/%s/%s/" % (kind, resource_id))
+        )
+        _check_resp_is_ok(resp, "Failed to get grants for %s %s" % (kind, resource_id))
+        return resp.json()
+
+    def set_authorization_grants(
+        self,
+        kind: GrantKind,
+        resource_id: str,
+        grants_data: dict,
+    ):
+        """
+        **beta** function. Set the authorization grants for a given resource.
+
+        Args:
+            kind: The kind of resource to set the grants for
+            resource_id: The ID of the resource
+            grants: See https://app.picterra.ch/public/apidocs/plots_analysis/v1/#tag/authorization/operation/setGrants.
+
+        Returns:
+            dict: The updated authorization grants for the resource.
+        """
+        resp = self.sess.post(
+            self._full_url("authorization/grants/%s/%s/" % (kind, resource_id)),
+            json=grants_data,
+        )
+        _check_resp_is_ok(resp, "Failed to set grants for %s %s" % (kind, resource_id))
+        return resp.json()

tests/test_tracer_client.py

@@ -494,3 +494,38 @@ def test_get_plots_analysis_report(monkeypatch):
     report = client.get_plots_analysis_report("a-report-id")
     assert report["id"] == "a-report-id"
     assert report["artifacts"][0]["name"] == "EUDR Report"
+
+
+@responses.activate
+def test_get_authorization_grants(monkeypatch):
+    client: TracerClient = _client(monkeypatch, platform="plots_analysis")
+    data = {
+        "grants": [
+            {"grantee": {"type": "user", "id": "1"}, "role": "viewer"},
+            {"grantee": {"type": "organization", "id": "2"}, "role": "editor"},
+        ]
+    }
+    _add_api_response(
+        plots_analysis_api_url("authorization/grants/plots_group/a-plots-group-id/"),
+        responses.GET,
+        data,
+    )
+    grants = client.get_authorization_grants("plots_group", "a-plots-group-id")
+    assert grants == data and len(responses.calls) == 1
+
+
+@responses.activate
+def test_set_authorization_grants(monkeypatch):
+    client: TracerClient = _client(monkeypatch, platform="plots_analysis")
+    new_grants = [
+        {"grantee": {"type": "user", "id": "1"}, "role": "viewer"},
+        {"grantee": {"type": "organization", "id": "2"}, "role": "editor"},
+    ]
+    _add_api_response(
+        plots_analysis_api_url("authorization/grants/plots_group/a-plots-group-id/"),
+        responses.POST,
+        {"grants": new_grants},
+        match=responses.matchers.json_params_matcher({"grants": new_grants}),
+    )
+    grants = client.set_authorization_grants("plots_group", "a-plots-group-id", {"grants": new_grants})
+    assert grants == {"grants": new_grants} and len(responses.calls) == 1