Falcon和falcon-multipart + POST请求上传文件实现

时间:2018-02-23 03:12:35

标签: python-3.x post binaryfiles falconframework

我正在尝试使用Falcon框架(python)实现上传文件的POST请求。

我使用了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 个答案:

答案 0 :(得分: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

答案 1 :(得分:0)

试试这个

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)`