54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8
|
|
|
|
import time
|
|
|
|
import tornado
|
|
import tornado.ioloop
|
|
import tornado.web
|
|
from tornado import gen, httpclient, ioloop
|
|
from tornado.options import define, options
|
|
|
|
|
|
define("port", default=8200, help="Run server on a specific port", type=int)
|
|
|
|
class MainHandler(tornado.web.RequestHandler):
|
|
def get(self):
|
|
self.write("Hello, proxy")
|
|
|
|
|
|
class ProxyHandler(tornado.web.RequestHandler):
|
|
http_client = httpclient.AsyncHTTPClient()
|
|
|
|
@gen.coroutine
|
|
def get(self, url):
|
|
print time.strftime('%Y-%m-%d %H:%M:%S'), 'PROXY http://' + url
|
|
response = yield self.http_client.fetch('http://'+url) #www.google.com')
|
|
# print response.body
|
|
if (response.error and not
|
|
isinstance(response.error, tornado.httpclient.HTTPError)):
|
|
self.set_status(500)
|
|
self.write('Internal server error:\n' + str(response.error))
|
|
else:
|
|
self.set_status(response.code, response.reason)
|
|
for header, v in response.headers.get_all():
|
|
if header not in ('Content-Length', 'Transfer-Encoding', 'Content-Encoding', 'Connection'):
|
|
self.set_header(header, v) # some header appear multiple times, eg 'Set-Cookie'
|
|
if response.body:
|
|
self.set_header('Content-Length', len(response.body))
|
|
self.write(response.body)
|
|
self.finish()
|
|
|
|
|
|
def make_app(debug=True):
|
|
return tornado.web.Application([
|
|
(r"/", MainHandler),
|
|
(r"/proxy/(.*)", ProxyHandler),
|
|
], debug=debug)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = make_app()
|
|
tornado.options.parse_command_line()
|
|
app.listen(options.port)
|
|
ioloop.IOLoop.current().start() |