def _verify_full_response(self, request, expected_response):
request_proto = api_pb2.ListSessionGroupsRequest()
text_format.Merge(request, request_proto)
handler = list_session_groups.Handler(
backend_context.Context(self._mock_tb_context), request_proto)
response = handler.run()
self.assertProtoEquals(expected_response, response)
def download_data_route(self, request):
ctx = plugin_util.context(request.environ)
experiment_id = plugin_util.experiment_id(request.environ)
try:
response_format = request.args.get("format")
columns_visibility = json.loads(
request.args.get("columnsVisibility")
)
request_proto = _parse_request_argument(
request, api_pb2.ListSessionGroupsRequest
)
session_groups = list_session_groups.Handler(
ctx, self._context, experiment_id, request_proto
).run()
experiment = get_experiment.Handler(
ctx, self._context, experiment_id
).run()
body, mime_type = download_data.Handler(
self._context,
experiment,
session_groups,
response_format,
columns_visibility,
).run()
return http_util.Respond(request, body, mime_type)
except error.HParamsError as e:
logger.error("HParams error: %s" % e)
raise werkzeug.exceptions.BadRequest(description=str(e))
def _verify_handler(self, request, expected_session_group_names,
expected_total_size):
request_proto = api_pb2.ListSessionGroupsRequest()
text_format.Merge(request, request_proto)
handler = list_session_groups.Handler(
backend_context.Context(self._mock_tb_context), request_proto)
response = handler.run()
self.assertEqual(expected_session_group_names,
[sg.name for sg in response.session_groups])
self.assertEqual(expected_total_size, response.total_size)
def _run_handler(self, request):
request_proto = api_pb2.ListSessionGroupsRequest()
text_format.Merge(request, request_proto)
handler = list_session_groups.Handler(
backend_context.Context(self._mock_tb_context), request_proto)
response = handler.run()
# Sort the metric values repeated field in each session group to
# canonicalize the response.
for group in response.session_groups:
group.metric_values.sort(key=operator.attrgetter("name.tag"))
return response
def list_session_groups_route(self, request):
try:
request_proto = _parse_request_argument(
request, api_pb2.ListSessionGroupsRequest)
return http_util.Respond(
request,
json_format.MessageToJson(
list_session_groups.Handler(self._context,
request_proto).run(),
including_default_value_fields=True,
), 'application/json')
except error.HParamsError as e:
logger.error('HParams error: %s' % e)
raise werkzeug.exceptions.BadRequest(description=str(e))
def list_session_groups_route(self, request):
try:
# args.get() returns the request unquoted.
request_proto = request.args.get('request')
if request_proto is None:
raise error.HParamsError(
'/session_groups must have a \'request\' arg.')
request_proto = json_format.Parse(
request_proto, api_pb2.ListSessionGroupsRequest())
return http_util.Respond(
request,
json_format.MessageToJson(
list_session_groups.Handler(self._context,
request_proto).run()),
'application/json')
except error.HParamsError as e:
raise werkzeug.exceptions.BadRequest(description=str(e))
def list_session_groups_route(self, request):
ctx = plugin_util.context(request.environ)
experiment_id = plugin_util.experiment_id(request.environ)
try:
request_proto = _parse_request_argument(
request, api_pb2.ListSessionGroupsRequest)
return http_util.Respond(
request,
json_format.MessageToJson(
list_session_groups.Handler(ctx, self._context,
experiment_id,
request_proto).run(),
including_default_value_fields=True,
),
"application/json",
)
except error.HParamsError as e:
logger.error("HParams error: %s" % e)
raise werkzeug.exceptions.BadRequest(description=str(e))
