Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 36 additions & 0 deletions src/picterra/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -598,6 +598,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,
Expand Down Expand Up @@ -953,6 +971,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: str):
"""
Removes a vector layer
Expand Down
22 changes: 21 additions & 1 deletion tests/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -770,6 +770,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"
Expand Down Expand Up @@ -930,6 +940,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"
Expand Down Expand Up @@ -963,7 +983,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")
Expand Down