Browse Source

enhance json api param error reporting

tags/v1.1.0
Ng Ngai Fai 5 years ago
parent
commit
b66f83d622
1 changed files with 9 additions and 1 deletions
  1. +9
    -1
      mindinsight/backend/explainer/explainer_api.py

+ 9
- 1
mindinsight/backend/explainer/explainer_api.py View File

@@ -25,6 +25,7 @@ from flask import request
from mindinsight.conf import settings from mindinsight.conf import settings
from mindinsight.utils.exceptions import ParamMissError from mindinsight.utils.exceptions import ParamMissError
from mindinsight.utils.exceptions import ParamValueError from mindinsight.utils.exceptions import ParamValueError
from mindinsight.datavisual.common.exceptions import ImageNotExistError
from mindinsight.datavisual.common.validation import Validation from mindinsight.datavisual.common.validation import Validation
from mindinsight.datavisual.data_transform.summary_watcher import SummaryWatcher from mindinsight.datavisual.data_transform.summary_watcher import SummaryWatcher
from mindinsight.datavisual.utils.tools import get_train_id from mindinsight.datavisual.utils.tools import get_train_id
@@ -125,9 +126,14 @@ def query_saliency():
limit = Validation.check_limit(limit, min_value=1, max_value=100) limit = Validation.check_limit(limit, min_value=1, max_value=100)
offset = data.get("offset", 0) offset = data.get("offset", 0)
offset = Validation.check_offset(offset=offset) offset = Validation.check_offset(offset=offset)
sorted_name = data.get("sorted_name")
sorted_name = data.get("sorted_name", "")
sorted_type = data.get("sorted_type", "descending") sorted_type = data.get("sorted_type", "descending")


if sorted_name not in ("", "confidence"):
raise ParamValueError("sorted_name")
if sorted_type not in ("ascending", "descending"):
raise ParamValueError("sorted_type")

encapsulator = SaliencyEncap( encapsulator = SaliencyEncap(
_image_url_formatter, _image_url_formatter,
ExplainManagerHolder.get_instance()) ExplainManagerHolder.get_instance())
@@ -175,6 +181,8 @@ def query_image():


encapsulator = ExplainJobEncap(ExplainManagerHolder.get_instance()) encapsulator = ExplainJobEncap(ExplainManagerHolder.get_instance())
image = encapsulator.query_image_binary(train_id, image_id, image_type) image = encapsulator.query_image_binary(train_id, image_id, image_type)
if image is None:
raise ImageNotExistError(f"image_id:{image_id}")


return image return image




Loading…
Cancel
Save