def parse_request()

in docker/build_artifacts/sagemaker/tfs_utils.py [0:0]

• 17 lines of code
• 4 McCabe index (conditional complexity)

def parse_request(req, rest_port, grpc_port, default_model_name, model_name=None, channel=None):
    tfs_attributes = parse_tfs_custom_attributes(req)
    tfs_uri = make_tfs_uri(rest_port, tfs_attributes, default_model_name, model_name)

    if not model_name:
        model_name = tfs_attributes.get("tfs-model-name")

    context = Context(model_name,
                      tfs_attributes.get("tfs-model-version"),
                      tfs_attributes.get("tfs-method"),
                      tfs_uri,
                      grpc_port,
                      channel,
                      req.get_header(CUSTOM_ATTRIBUTES_HEADER),
                      req.get_header("Content-Type") or DEFAULT_CONTENT_TYPE,
                      req.get_header("Accept") or DEFAULT_ACCEPT_HEADER,
                      req.content_length)

    data = req.stream
    return data, context