mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-04-23 06:09:43 +08:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from flask import Response
|
|
from flask_restful import Resource, reqparse
|
|
from werkzeug.exceptions import Forbidden, NotFound
|
|
|
|
from controllers.files import api
|
|
from controllers.files.error import UnsupportedFileTypeError
|
|
from core.tools.tool_file_manager import ToolFileManager
|
|
|
|
|
|
class ToolFilePreviewApi(Resource):
|
|
def get(self, file_id, extension):
|
|
file_id = str(file_id)
|
|
|
|
parser = reqparse.RequestParser()
|
|
|
|
parser.add_argument("timestamp", type=str, required=True, location="args")
|
|
parser.add_argument("nonce", type=str, required=True, location="args")
|
|
parser.add_argument("sign", type=str, required=True, location="args")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not ToolFileManager.verify_file(
|
|
file_id=file_id,
|
|
timestamp=args["timestamp"],
|
|
nonce=args["nonce"],
|
|
sign=args["sign"],
|
|
):
|
|
raise Forbidden("Invalid request.")
|
|
|
|
try:
|
|
result = ToolFileManager.get_file_generator_by_tool_file_id(
|
|
file_id,
|
|
)
|
|
|
|
if not result:
|
|
raise NotFound("file is not found")
|
|
|
|
generator, mimetype = result
|
|
except Exception:
|
|
raise UnsupportedFileTypeError()
|
|
|
|
return Response(generator, mimetype=mimetype)
|
|
|
|
|
|
api.add_resource(ToolFilePreviewApi, "/files/tools/<uuid:file_id>.<string:extension>")
|