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

时间:2014-09-02 06:31:58

标签: python curl file-upload keyerror

我在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'

这里有什么问题?

2 个答案:

答案 0 :(得分:1)

如果在字典中找不到密钥,则会引发{p> KeyError

我不熟悉cgi.FieldStorage,但我想你应该得到一本字典并且缺少密钥'photofile'。 只需打印整个form_data并查看其内容。

您可以使用form_data.get('photofile', '')获取默认值并避免例外。

BTW,对于写入文件,您可以使用with语句作为上下文管理器。

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

答案 1 :(得分:0)

好的,我从here找到原因和方法。

原因是"您不能将CGI代码嵌入到服务器代码中。 CGIHTTPServer(以及一般的CGI)的重点是它在指定的目录中执行单独的脚本并返回输出。 因为您已经覆盖了do_POST方法,所以从不调用解析表单数据的代码。" (实际上我打印了form_data,它是空的 - " None, None, []")。

要修复它,请使用传入的参数(details

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