Esempio n. 1
0
File: api.py Progetto: atbrox/dpxdt
def download():
    """Downloads an artifact by it's content hash."""
    # Allow access to a resource hash for API keys with access to the given
    # build_id. Falls back to standard user login for the frontend (which
    # may require redirecting the user to the login form).
    try:
        _, build = auth.can_api_key_access_build('build_id')
    except HTTPException:
        build = auth.can_user_access_build('build_id')

    sha1sum = request.args.get('sha1sum', type=str)
    artifact = models.Artifact.query.get(sha1sum)
    if not artifact:
        logging.debug('Artifact sha1sum=%r does not exist', sha1sum)
        abort(404)

    build_id = request.args.get('build_id', type=int)
    if not build_id:
        logging.debug('build_id missing for artifact sha1sum=%r', sha1sum)
        abort(404)

    is_owned = artifact.owners.filter_by(id=build_id).first()
    if not is_owned:
        logging.debug('build_id=%r not owner of artifact sha1sum=%r',
                      build_id, sha1sum)
        abort(403)

    # Make sure there are no Set-Cookie headers on the response so this
    # request is cachable by all HTTP frontends.
    @utils.after_this_request
    def no_session(response):
        if 'Set-Cookie' in response.headers:
            del response.headers['Set-Cookie']

    if request.if_none_match and request.if_none_match.contains(sha1sum):
        response = flask.Response(status=304)
        return response

    return _get_artifact_response(artifact)
Esempio n. 2
0
def download():
    """Downloads an artifact by it's content hash."""
    # Allow access to a resource hash for API keys with access to the given
    # build_id. Falls back to standard user login for the frontend (which
    # may require redirecting the user to the login form).
    try:
        build = auth.can_api_key_access_build('build_id')
    except HTTPException:
        build = auth.can_user_access_build('build_id')

    sha1sum = request.args.get('sha1sum', type=str)
    artifact = models.Artifact.query.get(sha1sum)
    if not artifact:
        logging.debug('Artifact sha1sum=%r does not exist', sha1sum)
        abort(404)

    build_id = request.args.get('build_id', type=int)
    if not build_id:
        logging.debug('build_id missing for artifact sha1sum=%r', sha1sum)
        abort(404)

    is_owned = artifact.owners.filter_by(id=build_id).first()
    if not is_owned:
        logging.debug('build_id=%r not owner of artifact sha1sum=%r',
                      build_id, sha1sum)
        abort(403)

    if request.if_none_match and request.if_none_match.contains(sha1sum):
        return flask.Response(status=304)

    response = flask.Response(
        artifact.data,
        mimetype=artifact.content_type)
    response.cache_control.private = True
    response.cache_control.max_age = 8640000
    response.set_etag(sha1sum)
    return response