在Twisted客户端生成头部

Question

我正在使用Twisted的“HTTPClient”协议与一个HTTP服务器进行对话（用Twisted应用作为客户端），问题是我大部分请求都不接受我的请求头。有没有什么方法（可能是用某些库）可以生成请求头呢？我会贴一些代码让你看看。

class Local(HTTPClient):
def generateHeaders(self, type=None):
    self.sendHeader('User-Agent', 'Twisted Web Client')
    if type:
        self.sendHeader('Content-Type', 'x-www-form-urlencoded')
    self.sendHeader('Accept', '*/*')
    self.sendHeader('Host', 'localhost')
    self.endHeaders()

def connectionMade(self):
    ## GENERATE ID
    self.userid= self.generateID()
    self.sendCommand('POST', '/chat/action=join&user=pirosb3')
    self.generateHeaders(type=True)

def dataReceived(self, data):
    print data

如你所见，我创建了一个方法来为我生成请求头，但我必须说，这样做效率不高，我总是收到错误的响应。

有人知道解决办法吗？

谢谢

Answer 1

我强烈建议你使用Twisted的Agent类，配合Header类，这些都在twisted.web.http_headers模块里。

这里有一个很不错的指南，你可以点击查看。

如果你想要一个实际的例子，展示如何用Twisted的HTTP客户端添加一些请求头，并且能读取响应中的头信息，这里有一个示例：

from pprint import pformat

from twisted.internet import reactor
from twisted.internet.defer import Deferred
from twisted.internet.protocol import Protocol
from twisted.web.client import Agent
from twisted.web.http_headers import Headers

class BeginningPrinter(Protocol):
    def __init__(self, finished):
        self.finished = finished
        self.remaining = 1024 * 10

    def dataReceived(self, bytes):
        if self.remaining:
            display = bytes[:self.remaining]
            print 'Some data received:'
            print display
            self.remaining -= len(display)

    def connectionLost(self, reason):
        print 'Finished receiving body:', reason.getErrorMessage()
        self.finished.callback(None)

agent = Agent(reactor)
d = agent.request(
    'GET',
    'http://example.com/',
    Headers({'User-Agent': ['Twisted Web Client Example']}),
    None)

def cbRequest(response):
    print 'Response version:', response.version
    print 'Response code:', response.code
    print 'Response phrase:', response.phrase
    print 'Response headers:'
    print pformat(list(response.headers.getAllRawHeaders()))
    finished = Deferred()
    response.deliverBody(BeginningPrinter(finished))
    return finished
d.addCallback(cbRequest)

def cbShutdown(ignored):
    reactor.stop()
d.addBoth(cbShutdown)

reactor.run()