mirror of
https://github.com/l1ving/youtube-dl
synced 2025-01-24 20:58:56 +08:00
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
"""
|
|
SocksiPy + urllib handler
|
|
|
|
version: 0.2
|
|
author: e<e@tr0ll.in>
|
|
|
|
This module provides a Handler which you can use with urllib2 to allow it to tunnel your connection through a socks.sockssocket socket, with out monkey patching the original socket...
|
|
"""
|
|
|
|
import urllib2
|
|
import httplib
|
|
import socks
|
|
|
|
class SocksiPyConnection(httplib.HTTPConnection):
|
|
|
|
def __init__(self, proxytype, proxyaddr, proxyport=None, rdns=True, username=None, password=None, *args, **kwargs):
|
|
self.proxyargs = (proxytype, proxyaddr, proxyport, rdns, username, password)
|
|
httplib.HTTPConnection.__init__(self, *args, **kwargs)
|
|
|
|
def connect(self):
|
|
self.sock = socks.socksocket()
|
|
self.sock.setproxy(*self.proxyargs)
|
|
if isinstance(self.timeout, float):
|
|
self.sock.settimeout(self.timeout)
|
|
self.sock.connect((self.host, self.port))
|
|
|
|
class SocksiPyHandler(urllib2.HTTPHandler):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.args = args
|
|
self.kw = kwargs
|
|
urllib2.HTTPHandler.__init__(self)
|
|
|
|
# make it look like a ProxyHandler
|
|
self.proxies = {
|
|
'socks': self.args[1] + ':' + str(self.args[2]),
|
|
}
|
|
|
|
def http_open(self, req):
|
|
def build(host, port=None, strict=None, timeout=0):
|
|
conn = SocksiPyConnection(*self.args, host=host, port=port, strict=strict, timeout=timeout, **self.kw)
|
|
return conn
|
|
return self.do_open(build, req)
|
|
|
|
if __name__ == "__main__":
|
|
opener = urllib2.build_opener(SocksiPyHandler(socks.PROXY_TYPE_SOCKS4, 'localhost', 9999))
|
|
print opener.open('http://www.whatismyip.com/automation/n09230945.asp').read()
|