Add functions to retrieve a single detector or vector layer

Author: ao-picterra

src/picterra/client.py

@@ -582,6 +582,24 @@ def list_detectors(
             data["is_shared"] = is_shared
         return self._paginate_through_list("detectors", data)
 
+    def get_detector(self, detector_id: str):
+        """
+        Get detector information
+
+        Args:
+            detector_id: id of the detector
+
+        Raises:
+            APIError: There was an error while getting the detector information
+
+        Returns:
+            dict: Dictionary of the information
+        """
+        resp = self.sess.get(self._api_url("detectors/%s/" % detector_id))
+        if not resp.ok:
+            raise APIError(resp.text)
+        return resp.json()
+
     def edit_detector(
         self,
         detector_id: str,
@@ -943,6 +961,24 @@ def edit_vector_layer(
         if not resp.ok:
             raise APIError(resp.text)
 
+    def get_vector_layer(self, vector_layer_id: str):
+        """
+        Get vector layer information
+
+        Args:
+            vector_layer_id: id of the detector
+
+        Raises:
+            APIError: There was an error while getting the vector layer information
+
+        Returns:
+            dict: Dictionary of the information
+        """
+        resp = self.sess.get(self._api_url("vector_layers/%s/" % vector_layer_id))
+        if not resp.ok:
+            raise APIError(resp.text)
+        return resp.json()
+
     def delete_vector_layer(self, vector_layer_id: UUID):
         """
         Removes a vector layer

tests/test_client.py

@@ -750,6 +750,16 @@ def test_list_detectors():
     assert len(responses.calls) == 6
 
 
+@responses.activate
+def test_get_detector():
+    """Test the detector information"""
+    DETECTOR_ID = "foobar"
+    client = _client()
+    _add_api_response("detectors/%s/" % DETECTOR_ID, json={}, status=200)
+    client.get_detector(DETECTOR_ID)
+    assert len(responses.calls) == 1
+
+
 @responses.activate
 def test_delete_detector():
     DETECTOR_ID = "foobar"
@@ -871,7 +881,7 @@ def test_train_detector():
                 "training_annotations_count": 3,
                 "validation_annotations_count": 1,
             },
-        }
+        },
     )
     client = _client()
     op = client.train_detector(1)
@@ -912,6 +922,16 @@ def test_delete_vector_layer():
     assert len(responses.calls) == 1
 
 
+@responses.activate
+def test_get_vector_layer():
+    """Test the vector layer information"""
+    LAYER_ID = "foobar"
+    client = _client()
+    _add_api_response("vector_layers/%s/" % LAYER_ID, json={}, status=200)
+    client.get_vector_layer(LAYER_ID)
+    assert len(responses.calls) == 1
+
+
 @responses.activate
 def test_edit_vector_layer():
     LAYER_ID = "foobar"
@@ -945,7 +965,7 @@ def test_list_raster_markers():
 
 
 @responses.activate
-def test_list_raster_markers():
+def test_create_marker():
     client = _client()
     add_mock_marker_creation_response("spam", "foo", "bar", [12.34, 56.78], "foobar")
     marker = client.create_marker("foo", "bar", 12.34, 56.78, "foobar")