Falcon и falcon-multipart + POST-запрос для реализации загрузки файлов

Я пытаюсь реализовать запрос POST для загрузки файлов с помощью Falcon Framework (Python).

Я использовал falcon-multipart для multipart/form-data, это позволяет мне получать мой файл в cgi.FieldStorage(), в котором файл находится в двоичном формате, но теперь мне нужно записать этот файл в каталог с оригинальное расширение.

Это код, который я использую.

app.py:

import falcon

from .files import Resource

from falcon_multipart.middleware import MultipartMiddleware

api = application = falcon.API(middleware=[MultipartMiddleware()])

files = Resource()
api.add_route('/files', files)

files.py:

import io
import os
import shutil

import falcon
import json


class Resource(object):

   _storage_path = './uploaded_files'

   def on_post(self, req, resp):
        """
        POST METHOD
        """
        # Retrieve file extension
        ext = req.get_param('extension')

        # Retrieve input_file
        input_file = req.get_param('file')

        # Read file as binary
        raw = input_file.file.read()

        # Retrieve filename
        filename = input_file.filename

        # Define file_path
        file_path = os.path.join(self._storage_path, filename)

        # Write to a temporary file to prevent incomplete files from
        # being used.
        temp_file_path = file_path + '~'

        # Finally write the data to a temporary file
        with open(temp_file_path, 'wb') as output_file:
            shutil.copyfileobj(raw, output_file)

        # Now that we know the file has been fully saved to disk
        # move it into place.
        os.rename(temp_file_path, file_path)

        resp.status = falcon.HTTP_201

2 ответа

Я должен был изучить CGI

Это реализация, которую я использовал:

def on_post(self, req, resp):
        """
        POST METHOD
        """
        # Retrieve input_file
        input_file = req.get_param('file')

        # Test if the file was uploaded
        if input_file.filename:
            # Retrieve filename
            filename = input_file.filename

            # Define file_path
            file_path = os.path.join(self._storage_path, filename)

            # Write to a temporary file to prevent incomplete files
            # from being used.
            temp_file_path = file_path + '~'

            open(temp_file_path, 'wb').write(input_file.file.read())

            # Now that we know the file has been fully saved to disk
            # move it into place.
            os.rename(temp_file_path, file_path)

    resp.status = falcon.HTTP_201

Попробуй это

import io
import os
import uuid
import mimetypes
import falcon
import json




class Resource(object):

        _CHUNK_SIZE_BYTES = 4096

        def __init__(self, storage_path):
            self._storage_path = storage_path

        def on_post(self, req, resp):
            image = req.get_param("profilePic")
            # image_type = req.get_param("profilePic").type        
            ext = mimetypes.guess_extension(req.content_type)
            filename = "{uuid}{ext}".format(uuid=uuid.uuid4(), ext=ext)
            image_path = os.path.join(self._storage_path, filename)
            with open(image_path, "wb") as image_file:
                while True:
                    chunk = image.file.read(4096)
                    image_file.write(chunk)
                    if not chunk:
                        break
            resp.status = falcon.HTTP_200
            resp.location = filename
            resp.body = json.dumps("{name:" + image_path + "}")


import falcon
from falcon_multipart.middleware import MultipartMiddleware    

api = application = falcon.API(middleware=[MultipartMiddleware()])

images = Resource('images')
api.add_route('/images', images)`
Другие вопросы по тегам