def GetModelStatus(self, request, context):
logger.debug("MODEL_STATUS, get request: {}".format(request))
model_name = request.model_spec.name
requested_version = request.model_spec.version.value
valid_model_status = check_availability_of_requested_status(
models=self.models,
requested_version=requested_version,
model_name=model_name)
if not valid_model_status:
context.set_code(StatusCode.NOT_FOUND)
context.set_details(
WRONG_MODEL_SPEC.format(model_name, requested_version))
logger.debug("MODEL_STATUS, invalid model spec from request")
return get_model_status_pb2.GetModelStatusResponse()
response = get_model_status_pb2.GetModelStatusResponse()
if requested_version:
version_status = self.models[model_name].versions_statuses[
requested_version]
add_status_to_response(version_status, response)
else:
for version_status in self.models[model_name].versions_statuses. \
values():
add_status_to_response(version_status, response)
logger.debug("MODEL_STATUS created a response for {} - {}".format(
model_name, requested_version))
return response
def on_get(self, req, resp, model_name, requested_version=0):
logger.debug("MODEL_STATUS, get request: {}, {}".format(
model_name, requested_version))
valid_model_status = check_availability_of_requested_status(
models=self.models,
requested_version=requested_version,
model_name=model_name)
if not valid_model_status:
resp.status = falcon.HTTP_NOT_FOUND
logger.debug("MODEL_STATUS, invalid model spec from request")
err_out_json = {
'error': WRONG_MODEL_SPEC.format(model_name, requested_version)
}
resp.body = json.dumps(err_out_json)
return
requested_version = int(requested_version)
response = get_model_status_pb2.GetModelStatusResponse()
if requested_version:
version_status = self.models[model_name].versions_statuses[
requested_version]
add_status_to_response(version_status, response)
else:
for version_status in self.models[model_name].versions_statuses. \
values():
add_status_to_response(version_status, response)
logger.debug("MODEL_STATUS created a response for {} - {}".format(
model_name, requested_version))
resp.status = falcon.HTTP_200
resp.body = MessageToJson(response,
including_default_value_fields=True)