处理curl文件上传请求时，Python HTTP服务器中的KeyError

Question

我在Python中有一个HTTP服务器，用于处理来自curl的文件上传请求

class HTTPRequestHandler(BaseHTTPRequestHandler):
    def do_POST(self):
            if re.search('/api/v1/addphoto', self.path):
                    form_data=cgi.FieldStorage()
                    file_data=form_data['photofile'].value 
                    # Next I will save this file 
                    # fp =open('some/file','wb')
                    # fp.write(file_data)
                    # fp.close()

现在我必须使用curl来发送请求，命令是

curl -i -F name=photofile -F filedata=@01.jpeg http://server_ip:port/api/v1/addphoto

但是服务器报告错误

File "./apiserver.py", line 21, in do_POST
    file_data=form_data['photofile'].value
File "/usr/lib/python2.7/cgi.py", line 541, in __getitem__
    raise KeyError, key
KeyError: 'photofile'

这里有什么问题？

Answer 1

在字典中找不到键时引发KeyError 。

我对cgi.FieldStorage不熟悉，但是我想您应该得到字典，而键'photofile'丢失了。 只需打印整个form_data并查看其内容即可。

您可以使用form_data.get('photofile', '')获得默认值并避免异常。

顺便说一句，要写入文件，您可以将with语句用作上下文管理器。

with open('some/file','wb') as fp:
    fp.write(file_data)

Answer 2

好的，我从这里找到原因以及如何做。

原因是“您不能像这样将CGI代码嵌入服务器代码中。CGIHTTPServer（通常是CGI）的全部要点是，它在指定的目录中执行单独的脚本并返回输出。 因为您已经覆盖了do_POST方法中，从未调用过解析表单数据的代码 。” （实际上，我打印了form_data，它是空的-“ None, None, [] ”）。

要解决此问题，请使用传入的参数（ 详细信息 ）

 form = cgi.FieldStorage(
        fp=self.rfile, 
        headers=self.headers,
        environ={'REQUEST_METHOD':'POST',
                 'CONTENT_TYPE':self.headers['Content-Type'],
                 })