Function scratch_upload

py-src/data_formulator/routes/agents.py:1150–1185

Upload a file to the workspace scratch/ folder. Accepts multipart/form-data with a 'file' field. Returns: { status: "success", data: { path, url } }

()

@agent_bp.route('/workspace/scratch/upload', methods=['POST'])
def scratch_upload():
    """Upload a file to the workspace scratch/ folder.

    Accepts multipart/form-data with a 'file' field.
    Returns: { status: "success", data: { path, url } }
    """
    import hashlib
    from werkzeug.utils import secure_filename as _werkzeug_secure_filename

    if 'file' not in request.files:
        raise AppError(ErrorCode.INVALID_REQUEST, "No file in request")

    file = request.files['file']
    if not file.filename:
        raise AppError(ErrorCode.INVALID_REQUEST, "No filename")

    identity_id = get_identity_id()
    workspace = get_workspace(identity_id)
    scratch_jail = workspace.confined_scratch

    raw = file.read()
    file_hash = hashlib.sha256(raw).hexdigest()[:8]
    safe_name = _werkzeug_secure_filename(file.filename)
    base, ext = os.path.splitext(safe_name)
    final_name = f"{base}_{file_hash}{ext}"

    try:
        dest = scratch_jail.resolve(final_name)
    except ValueError:
        raise AppError(ErrorCode.VALIDATION_ERROR, "Invalid filename")
    dest.write_bytes(raw)

    return json_ok({
        "path": f"scratch/{final_name}",
        "url": f"/api/workspace/scratch/{final_name}",
    })


@agent_bp.route('/workspace/scratch/<path:filename>', methods=['GET'])