Python - Twisted、代理和修改内容

所以我研究了一些涉及使用 python 和 Twisted 框架编写 HTTP 代理的事情。

本质上，像其他一些问题一样，我希望能够修改将发送回浏览器的数据。也就是说，浏览器请求资源，代理将获取它。在资源返回到浏览器之前，我希望能够修改任何（HTTP 标头和内容）内容。

This ( 需要帮助编写扭曲的代理 https://stackoverflow.com/questions/6491932/need-help-writing-a-twisted-proxy）是我最初发现的。我尝试过，但它对我不起作用。我还发现了这个（Python Twisted代理-如何拦截数据包 https://stackoverflow.com/questions/9063583/python-twisted-proxy-how-to-intercept-packets）我认为这可行，但是我只能看到来自浏览器的 HTTP 请求。

我正在寻找任何建议。我的一些想法是使用 ProxyClient 和 ProxyRequest 类并重写这些函数，但我读到 Proxy 类本身是两者的组合。

对于那些可能要求查看一些代码的人，应该注意的是，我只使用了上述两个示例。任何帮助都很棒。

Thanks.

创造ProxyFactory可以修改服务器响应标头，您可以覆盖的内容ProxyClient.handle*() methods http://twistedmatrix.com/trac/browser/trunk/twisted/web/proxy.py#L62:

from twisted.python import log
from twisted.web import http, proxy

class ProxyClient(proxy.ProxyClient):
    """Mangle returned header, content here.

    Use `self.father` methods to modify request directly.
    """
    def handleHeader(self, key, value):
        # change response header here
        log.msg("Header: %s: %s" % (key, value))
        proxy.ProxyClient.handleHeader(self, key, value)

    def handleResponsePart(self, buffer):
        # change response part here
        log.msg("Content: %s" % (buffer[:50],))
        # make all content upper case
        proxy.ProxyClient.handleResponsePart(self, buffer.upper())

class ProxyClientFactory(proxy.ProxyClientFactory):
    protocol = ProxyClient

class ProxyRequest(proxy.ProxyRequest):
    protocols = dict(http=ProxyClientFactory)

class Proxy(proxy.Proxy):
    requestFactory = ProxyRequest

class ProxyFactory(http.HTTPFactory):
    protocol = Proxy

我通过查看来源得到了这个解决方案twisted.web.proxy http://twistedmatrix.com/trac/browser/trunk/twisted/web/proxy.py。我不知道这有多么地道。

将其作为脚本运行或通过twistd，在最后添加：

portstr = "tcp:8080:interface=localhost" # serve on localhost:8080

if __name__ == '__main__': # $ python proxy_modify_request.py
    import sys
    from twisted.internet import endpoints, reactor

    def shutdown(reason, reactor, stopping=[]):
        """Stop the reactor."""
        if stopping: return
        stopping.append(True)
        if reason:
            log.msg(reason.value)
        reactor.callWhenRunning(reactor.stop)

    log.startLogging(sys.stdout)
    endpoint = endpoints.serverFromString(reactor, portstr)
    d = endpoint.listen(ProxyFactory())
    d.addErrback(shutdown, reactor)
    reactor.run()
else: # $ twistd -ny proxy_modify_request.py
    from twisted.application import service, strports

    application = service.Application("proxy_modify_request")
    strports.service(portstr, ProxyFactory()).setServiceParent(application)

Usage

$ twistd -ny proxy_modify_request.py

在另一个终端中：

$ curl -x localhost:8080 http://example.com