mirror of
https://github.com/langgenius/dify.git
synced 2026-07-16 09:58:42 +08:00
ec67426ff6
Complete the request/response contract unification (GareArc review followup): the remaining handlers that returned a model via manual `.model_dump(mode="json")` now return the bare model behind `@returns`, so every openapi handler references its response model once and doc == runtime. Migrated: version, health, account info/session-revoke (x2), file upload, task stop, human-input form submit, and workspaces list/detail/switch/invite/remove/ role. Handlers returning a bare dict (account info, version, health) now go through `@returns` → `(body, status)`; HTTP behavior is unchanged. Health/task-stop/form-submit now construct their advertised model instead of a hand-built dict.
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
"""POST /openapi/v1/apps/<app_id>/files/upload — upload a file for use in app inputs."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from flask import request
|
|
from flask_restx import Resource
|
|
from flask_restx.api import HTTPStatus
|
|
from werkzeug.exceptions import BadRequest
|
|
|
|
import services
|
|
from controllers.common.errors import (
|
|
BlockedFileExtensionError,
|
|
FilenameNotExistsError,
|
|
FileTooLargeError,
|
|
NoFileUploadedError,
|
|
TooManyFilesError,
|
|
UnsupportedFileTypeError,
|
|
)
|
|
from controllers.openapi import openapi_ns
|
|
from controllers.openapi._contract import returns
|
|
from controllers.openapi.auth.composition import auth_router
|
|
from controllers.openapi.auth.data import AuthData
|
|
from extensions.ext_database import db
|
|
from fields.file_fields import FileResponse
|
|
from libs.oauth_bearer import Scope
|
|
from services.file_service import FileService
|
|
|
|
|
|
@openapi_ns.route("/apps/<string:app_id>/files/upload")
|
|
class AppFileUploadApi(Resource):
|
|
@openapi_ns.doc("upload_file_for_app_input")
|
|
@openapi_ns.doc(description="Upload a file to use as an input variable when running the app")
|
|
@openapi_ns.doc(
|
|
responses={
|
|
201: "File uploaded successfully",
|
|
400: "Bad request — no file or filename missing",
|
|
401: "Unauthorized — invalid or expired bearer token",
|
|
413: "File too large",
|
|
415: "Unsupported file type or blocked extension",
|
|
}
|
|
)
|
|
@auth_router.guard(scope=Scope.APPS_RUN)
|
|
@returns(HTTPStatus.CREATED, FileResponse, description="File uploaded")
|
|
def post(self, app_id: str, *, auth_data: AuthData):
|
|
app_model, caller, _ = auth_data.require_app_context()
|
|
if "file" not in request.files:
|
|
raise NoFileUploadedError()
|
|
if len(request.files) > 1:
|
|
raise TooManyFilesError()
|
|
|
|
file = request.files["file"]
|
|
if not file.mimetype:
|
|
raise UnsupportedFileTypeError()
|
|
if not file.filename:
|
|
raise FilenameNotExistsError()
|
|
|
|
try:
|
|
upload_file = FileService(db.engine).upload_file(
|
|
filename=file.filename,
|
|
content=file.stream.read(),
|
|
mimetype=file.mimetype,
|
|
user=caller,
|
|
)
|
|
except ValueError as exc:
|
|
raise BadRequest(str(exc))
|
|
except services.errors.file.FileTooLargeError as exc:
|
|
raise FileTooLargeError(exc.description)
|
|
except services.errors.file.UnsupportedFileTypeError:
|
|
raise UnsupportedFileTypeError()
|
|
except services.errors.file.BlockedFileExtensionError as exc:
|
|
raise BlockedFileExtensionError(exc.description)
|
|
|
|
return FileResponse.model_validate(upload_file, from_attributes=True)
|