Python - BaseHTTPServer do_GET() - wfile.write(filedata) 损坏的管道

Question

我需要设置一个返回几个 3MB 文件的 Python 网络服务器。它使用 baseHTTPServer 来处理 GET 请求。如何使用 wfile.write() 发送 3MB 的文件？

from SocketServer import ThreadingMixIn
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
import BaseHTTPServer


class StoreHandler(BaseHTTPServer.BaseHTTPRequestHandler):
    request_queue_size = 100

    def do_GET(self):
        try:

            filepath = os.path.join(os.path.join(os.path.dirname(__file__), "tools"), "tools.zip")
            if not os.path.exists:
                print 'Tool doesnt exist'

            f = open(filepath, 'rb')
            file_data = f.read()
            f.close()

            self.send_header("Content-type", "application/octet-stream")
            self.end_headers()
            self.wfile.write(file_data) 
            self.send_response(200)
        except Exception,e:
            print e
            self.send_response(400)

错误:

----------------------------------------
Exception happened during processing of request from ('192.168.0.6', 41025)
Traceback (most recent call last):
  File "/usr/lib/python2.7/SocketServer.py", line 593, in process_request_thread
    self.finish_request(request, client_address)
  File "/usr/lib/python2.7/SocketServer.py", line 334, in finish_request
    self.RequestHandlerClass(request, client_address, self)
  File "/usr/lib/python2.7/SocketServer.py", line 651, in __init__
    self.finish()
  File "/usr/lib/python2.7/SocketServer.py", line 710, in finish
    self.wfile.close()
  File "/usr/lib/python2.7/socket.py", line 279, in close
    self.flush()
  File "/usr/lib/python2.7/socket.py", line 303, in flush
    self._sock.sendall(view[write_offset:write_offset+buffer_size])
error: [Errno 32] Broken pipe

编辑:

客户端代码:

import requests

headers = {'user-agent': 'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; FSL 7.0.5.01003)'}
r = requests.get(url, headers=headers, timeout=60)

Answer 1

您离正确的服务器不远...

您根本不遵守 HTTP 协议(protocol)的命令顺序:第一个命令必须是send_response(或send_error)，接着是其他最终 header ，然后是 end_header 和数据。

此外，当不需要时，您还将整个文件加载到内存中。您的 do_GET 方法可以是:

def do_GET(self):
    try:

        filepath = os.path.join(os.path.join(os.path.dirname(__file__), "tools"), "tools.zip")
        if not os.path.exists:
            print 'Tool doesnt exist'

        f = open(filepath, 'rb')

        self.send_response(200)
        self.send_header("Content-type", "application/octet-stream")
        self.end_headers()
        while True:
            file_data = f.read(32768) # use an appropriate chunk size
            if file_data is None or len(file_data) == 0:
                break
            self.wfile.write(file_data) 
        f.close()
    except Exception,e:
        print e
        self.send_response(400)