我正在尝试用 python 编写一个网络代理。目标是访问一个 url,如:并像往常一样http://proxyurl/http://anothersite.com/
查看他的内容。http://anothersite.com
通过滥用 requests 库,我已经取得了不错的成绩,但这并不是 requests 框架的真正预期用途。我以前用twisted写过代理,但我不知道如何将它与我想做的事情联系起来。这就是我目前所处的位置......
import os
import urlparse
import requests
import tornado.ioloop
import tornado.web
from tornado import template
ROOT = os.path.dirname(os.path.abspath(__file__))
path = lambda *a: os.path.join(ROOT, *a)
loader = template.Loader(path(ROOT, 'templates'))
class ProxyHandler(tornado.web.RequestHandler):
def get(self, slug):
if slug.startswith("http://") or slug.startswith("https://"):
if self.get_argument("start", None) == "true":
parsed = urlparse.urlparse(slug)
self.set_cookie("scheme", value=parsed.scheme)
self.set_cookie("netloc", value=parsed.netloc)
self.set_cookie("urlpath", value=parsed.path)
#external resource
else:
response = requests.get(slug)
headers = response.headers
if 'content-type' in headers:
self.set_header('Content-type', headers['content-type'])
if 'length' in headers:
self.set_header('length', headers['length'])
for block in response.iter_content(1024):
self.write(block)
self.finish()
return
else:
#absolute
if slug.startswith('/'):
slug = "{scheme}://{netloc}{original_slug}".format(
scheme=self.get_cookie('scheme'),
netloc=self.get_cookie('netloc'),
original_slug=slug,
)
#relative
else:
slug = "{scheme}://{netloc}{path}{original_slug}".format(
scheme=self.get_cookie('scheme'),
netloc=self.get_cookie('netloc'),
path=self.get_cookie('urlpath'),
original_slug=slug,
)
response = requests.get(slug)
#get the headers
headers = response.headers
#get doctype
doctype = None
if '<!doctype' in response.content.lower()[:9]:
doctype = response.content[:response.content.find('>')+1]
if 'content-type' in headers:
self.set_header('Content-type', headers['content-type'])
if 'length' in headers:
self.set_header('length', headers['length'])
self.write(response.content)
application = tornado.web.Application([
(r"/(.+)", ProxyHandler),
])
if __name__ == "__main__":
application.listen(8888)
tornado.ioloop.IOLoop.instance().start()
请注意,如果查询字符串中有 start=true,我设置了一个 cookie 来保留方案、netloc 和 urlpath。这样,然后命中代理的任何相对或绝对链接都使用该 cookie 来解析完整的 url。
有了这个代码,如果你去http://localhost:8888/http://espn.com/?start=true
你会看到 ESPN 的内容。但是,在以下站点上它根本不起作用:http: //www.bottegaveneta.com/us/shop/。我的问题是,最好的方法是什么?我目前的实施方式是稳健的,还是这样做有一些可怕的陷阱?如果它是正确的,为什么像我指出的那样某些网站根本不起作用?
感谢您的任何帮助。