2

这个简单的 Falcon API 将在参数中使用HTTP POSTwithenctype=multipart/form-data和文件上传，file并在控制台上打印文件的内容：

# simple_api.py
import cgi
import falcon

class SomeTestApi(object):
    def on_post(self, req, resp):
        upload = cgi.FieldStorage(fp=req.stream, environ=req.env)
        upload = upload['file'].file.read()
        print(upload)


app = falcon.API()
app.add_route('/', SomeTestApi())

也可以使用falcon-multipart中间件来实现相同的目标。

要试用它，请使用gunicorn( pip install gunicorn) 运行它，

gunicorn simple_api.py

然后使用 cUrl（或任何选择的 REST 客户端）上传文本文件：

# sample.txt
this is some sample text

curl -F "file=@sample.txt" localhost:8000

我现在想通过模拟文件上传使用 Falcon 的测试助手来测试这个 API。但是，我还不明白如何做到这一点（如果可能的话？）。模拟请求方法有一个file_wrapper可能有用的参数，但从文档中我不明白应该如何填充它。

有什么建议么？

4

1 回答 1

4

这就是我想出的，它试图模拟我的 Chrome 所做的事情。请注意，这模拟了仅上传一个文件的情况，但您可以简单地修改此函数以上传多个文件，每个文件由两个新行分隔。

def create_multipart(data, fieldname, filename, content_type):
    """
    Basic emulation of a browser's multipart file upload
    """
    boundry = '----WebKitFormBoundary' + random_string(16)
    buff = io.BytesIO()
    buff.write(b'--')
    buff.write(boundry.encode())
    buff.write(b'\r\n')
    buff.write(('Content-Disposition: form-data; name="%s"; filename="%s"' % \
               (fieldname, filename)).encode())
    buff.write(b'\r\n')
    buff.write(('Content-Type: %s' % content_type).encode())
    buff.write(b'\r\n')
    buff.write(b'\r\n')
    buff.write(data)
    buff.write(b'\r\n')
    buff.write(boundry.encode())
    buff.write(b'--\r\n')
    headers = {'Content-Type': 'multipart/form-data; boundary=%s' %boundry}
    headers['Content-Length'] = str(buff.tell())
    return buff.getvalue(), headers

然后你可以像下面这样使用这个函数：

with open('test/resources/foo.pdf', 'rb') as f:
    foodata = f.read()

# Create the multipart data
data, headers = create_multipart(foodata, fieldname='uploadFile',
                                 filename='foo.pdf',
                                 content_type='application/pdf')

# Post to endpoint
client.simulate_request(method='POST', path=url,
                        headers=headers, body=data)
于 2017-08-15T23:45:41.973 回答