blob: ab22a958e4c1d7be16ca24725f0cd349901d9d22 [file] [log] [blame]
Guido van Rossume7b146f2000-02-04 15:28:42 +00001"""Open an arbitrary URL.
2
3See the following document for more info on URLs:
4"Names and Addresses, URIs, URLs, URNs, URCs", at
5http://www.w3.org/pub/WWW/Addressing/Overview.html
6
7See also the HTTP spec (from which the error codes are derived):
8"HTTP - Hypertext Transfer Protocol", at
9http://www.w3.org/pub/WWW/Protocols/
10
11Related standards and specs:
12- RFC1808: the "relative URL" spec. (authoritative status)
13- RFC1738 - the "URL standard". (authoritative status)
14- RFC1630 - the "URI spec". (informational status)
15
16The object returned by URLopener().open(file) will differ per
17protocol. All you know is that is has methods read(), readline(),
18readlines(), fileno(), close() and info(). The read*(), fileno()
Fredrik Lundhb49f88b2000-09-24 18:51:25 +000019and close() methods work like those of open files.
Guido van Rossume7b146f2000-02-04 15:28:42 +000020The info() method returns a mimetools.Message object which can be
21used to query various info about the object, if available.
22(mimetools.Message objects are queried with the getheader() method.)
23"""
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000024
Guido van Rossum7c395db1994-07-04 22:14:49 +000025import string
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000026import socket
Jack Jansendc3e3f61995-12-15 13:22:13 +000027import os
Guido van Rossumf0713d32001-08-09 17:43:35 +000028import time
Guido van Rossum3c8484e1996-11-20 22:02:24 +000029import sys
Brett Cannon69200fa2004-03-23 21:26:39 +000030from urlparse import urljoin as basejoin
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000031
Skip Montanaro40fc1602001-03-01 04:27:19 +000032__all__ = ["urlopen", "URLopener", "FancyURLopener", "urlretrieve",
33 "urlcleanup", "quote", "quote_plus", "unquote", "unquote_plus",
Skip Montanaro44d5e0c2001-03-13 19:47:16 +000034 "urlencode", "url2pathname", "pathname2url", "splittag",
35 "localhost", "thishost", "ftperrors", "basejoin", "unwrap",
36 "splittype", "splithost", "splituser", "splitpasswd", "splitport",
37 "splitnport", "splitquery", "splitattr", "splitvalue",
Brett Cannond75f0432007-05-16 22:42:29 +000038 "getproxies"]
Skip Montanaro40fc1602001-03-01 04:27:19 +000039
Martin v. Löwis3e865952006-01-24 15:51:21 +000040__version__ = '1.17' # XXX This version is not always updated :-(
Guido van Rossumf668d171997-06-06 21:11:11 +000041
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000042MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
Guido van Rossum6cb15a01995-06-22 19:00:13 +000043
Jack Jansendc3e3f61995-12-15 13:22:13 +000044# Helper for non-unix systems
45if os.name == 'mac':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000046 from macurl2path import url2pathname, pathname2url
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +000047elif os.name == 'nt':
Fredrik Lundhb49f88b2000-09-24 18:51:25 +000048 from nturl2path import url2pathname, pathname2url
Guido van Rossumd74fb6b2001-03-02 06:43:49 +000049elif os.name == 'riscos':
50 from rourl2path import url2pathname, pathname2url
Jack Jansendc3e3f61995-12-15 13:22:13 +000051else:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000052 def url2pathname(pathname):
Georg Brandlc0b24732005-12-26 22:53:56 +000053 """OS-specific conversion from a relative URL of the 'file' scheme
54 to a file system path; not recommended for general use."""
Guido van Rossum367ac801999-03-12 14:31:10 +000055 return unquote(pathname)
Georg Brandlc0b24732005-12-26 22:53:56 +000056
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000057 def pathname2url(pathname):
Georg Brandlc0b24732005-12-26 22:53:56 +000058 """OS-specific conversion from a file system path to a relative URL
59 of the 'file' scheme; not recommended for general use."""
Guido van Rossum367ac801999-03-12 14:31:10 +000060 return quote(pathname)
Guido van Rossum33add0a1998-12-18 15:25:22 +000061
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000062# This really consists of two pieces:
63# (1) a class which handles opening of all sorts of URLs
64# (plus assorted utilities etc.)
65# (2) a set of functions for parsing URLs
66# XXX Should these be separated out into different modules?
67
68
69# Shortcut for basic usage
70_urlopener = None
Fred Drakedf6eca72002-04-04 20:41:34 +000071def urlopen(url, data=None, proxies=None):
Skip Montanaro79f1c172000-08-22 03:00:52 +000072 """urlopen(url [, data]) -> open file-like object"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000073 global _urlopener
Fred Drakedf6eca72002-04-04 20:41:34 +000074 if proxies is not None:
75 opener = FancyURLopener(proxies=proxies)
76 elif not _urlopener:
77 opener = FancyURLopener()
78 _urlopener = opener
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000079 else:
Fred Drakedf6eca72002-04-04 20:41:34 +000080 opener = _urlopener
81 if data is None:
82 return opener.open(url)
83 else:
84 return opener.open(url, data)
Fred Drake316a7932000-08-24 01:01:26 +000085def urlretrieve(url, filename=None, reporthook=None, data=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000086 global _urlopener
87 if not _urlopener:
88 _urlopener = FancyURLopener()
Fred Drake316a7932000-08-24 01:01:26 +000089 return _urlopener.retrieve(url, filename, reporthook, data)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000090def urlcleanup():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000091 if _urlopener:
92 _urlopener.cleanup()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000093
Bill Janssen426ea0a2007-08-29 22:35:05 +000094# check for SSL
95try:
96 import ssl
97except:
98 _have_ssl = False
99else:
100 _have_ssl = True
101
Georg Brandlb9256022005-08-24 18:46:39 +0000102# exception raised when downloaded size does not match content-length
103class ContentTooShortError(IOError):
104 def __init__(self, message, content):
105 IOError.__init__(self, message)
106 self.content = content
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000107
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000108ftpcache = {}
109class URLopener:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000110 """Class to open URLs.
111 This is a class rather than just a subroutine because we may need
112 more than one set of global protocol-specific options.
113 Note -- this is a base class for those who don't want the
114 automatic handling of errors type 302 (relocated) and 401
115 (authorization needed)."""
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000116
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000117 __tempfiles = None
Guido van Rossum29e77811996-11-27 19:39:58 +0000118
Guido van Rossumba311382000-08-24 16:18:04 +0000119 version = "Python-urllib/%s" % __version__
120
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000121 # Constructor
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000122 def __init__(self, proxies=None, **x509):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000123 if proxies is None:
124 proxies = getproxies()
125 assert hasattr(proxies, 'has_key'), "proxies must be a mapping"
126 self.proxies = proxies
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000127 self.key_file = x509.get('key_file')
128 self.cert_file = x509.get('cert_file')
Georg Brandl0619a322006-07-26 07:40:17 +0000129 self.addheaders = [('User-Agent', self.version)]
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000130 self.__tempfiles = []
131 self.__unlink = os.unlink # See cleanup()
132 self.tempcache = None
133 # Undocumented feature: if you assign {} to tempcache,
134 # it is used to cache files retrieved with
135 # self.retrieve(). This is not enabled by default
136 # since it does not work for changing documents (and I
137 # haven't got the logic to check expiration headers
138 # yet).
139 self.ftpcache = ftpcache
140 # Undocumented feature: you can use a different
141 # ftp cache by assigning to the .ftpcache member;
142 # in case you want logically independent URL openers
143 # XXX This is not threadsafe. Bah.
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000144
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000145 def __del__(self):
146 self.close()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000147
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000148 def close(self):
149 self.cleanup()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000150
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000151 def cleanup(self):
152 # This code sometimes runs when the rest of this module
153 # has already been deleted, so it can't use any globals
154 # or import anything.
155 if self.__tempfiles:
156 for file in self.__tempfiles:
157 try:
158 self.__unlink(file)
Martin v. Löwis58682b72001-08-11 15:02:57 +0000159 except OSError:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000160 pass
161 del self.__tempfiles[:]
162 if self.tempcache:
163 self.tempcache.clear()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000164
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000165 def addheader(self, *args):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000166 """Add a header to be used by the HTTP interface only
167 e.g. u.addheader('Accept', 'sound/basic')"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000168 self.addheaders.append(args)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000169
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000170 # External interface
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000171 def open(self, fullurl, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000172 """Use URLopener().open(file) instead of open(file, 'r')."""
Martin v. Löwis1d994332000-12-03 18:30:10 +0000173 fullurl = unwrap(toBytes(fullurl))
Raymond Hettinger54f02222002-06-01 14:18:47 +0000174 if self.tempcache and fullurl in self.tempcache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000175 filename, headers = self.tempcache[fullurl]
176 fp = open(filename, 'rb')
177 return addinfourl(fp, headers, fullurl)
Martin v. Löwis1d994332000-12-03 18:30:10 +0000178 urltype, url = splittype(fullurl)
179 if not urltype:
180 urltype = 'file'
Raymond Hettinger54f02222002-06-01 14:18:47 +0000181 if urltype in self.proxies:
Martin v. Löwis1d994332000-12-03 18:30:10 +0000182 proxy = self.proxies[urltype]
183 urltype, proxyhost = splittype(proxy)
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000184 host, selector = splithost(proxyhost)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000185 url = (host, fullurl) # Signal special case to open_*()
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000186 else:
187 proxy = None
Martin v. Löwis1d994332000-12-03 18:30:10 +0000188 name = 'open_' + urltype
189 self.type = urltype
Brett Cannonaaeffaf2004-03-23 23:50:17 +0000190 name = name.replace('-', '_')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000191 if not hasattr(self, name):
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000192 if proxy:
193 return self.open_unknown_proxy(proxy, fullurl, data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000194 else:
195 return self.open_unknown(fullurl, data)
196 try:
197 if data is None:
198 return getattr(self, name)(url)
199 else:
200 return getattr(self, name)(url, data)
201 except socket.error, msg:
202 raise IOError, ('socket error', msg), sys.exc_info()[2]
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000203
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000204 def open_unknown(self, fullurl, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000205 """Overridable interface to open unknown URL type."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000206 type, url = splittype(fullurl)
207 raise IOError, ('url error', 'unknown url type', type)
Guido van Rossumca445401995-08-29 19:19:12 +0000208
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000209 def open_unknown_proxy(self, proxy, fullurl, data=None):
210 """Overridable interface to open unknown URL type."""
211 type, url = splittype(fullurl)
212 raise IOError, ('url error', 'invalid proxy for %s' % type, proxy)
213
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000214 # External interface
Sjoerd Mullenderd7b86f02000-08-25 11:23:36 +0000215 def retrieve(self, url, filename=None, reporthook=None, data=None):
Brett Cannon7d618c72003-04-24 02:43:20 +0000216 """retrieve(url) returns (filename, headers) for a local object
Guido van Rossume7b146f2000-02-04 15:28:42 +0000217 or (tempfilename, headers) for a remote object."""
Martin v. Löwis1d994332000-12-03 18:30:10 +0000218 url = unwrap(toBytes(url))
Raymond Hettinger54f02222002-06-01 14:18:47 +0000219 if self.tempcache and url in self.tempcache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000220 return self.tempcache[url]
221 type, url1 = splittype(url)
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000222 if filename is None and (not type or type == 'file'):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000223 try:
224 fp = self.open_local_file(url1)
225 hdrs = fp.info()
226 del fp
227 return url2pathname(splithost(url1)[1]), hdrs
228 except IOError, msg:
229 pass
Fred Drake316a7932000-08-24 01:01:26 +0000230 fp = self.open(url, data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000231 headers = fp.info()
Guido van Rossum3b0a3292002-08-09 16:38:32 +0000232 if filename:
233 tfp = open(filename, 'wb')
234 else:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000235 import tempfile
236 garbage, path = splittype(url)
237 garbage, path = splithost(path or "")
238 path, garbage = splitquery(path or "")
239 path, garbage = splitattr(path or "")
240 suffix = os.path.splitext(path)[1]
Guido van Rossum3b0a3292002-08-09 16:38:32 +0000241 (fd, filename) = tempfile.mkstemp(suffix)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000242 self.__tempfiles.append(filename)
Jeremy Hylton3bd6fde2002-10-11 14:36:24 +0000243 tfp = os.fdopen(fd, 'wb')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000244 result = filename, headers
245 if self.tempcache is not None:
246 self.tempcache[url] = result
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000247 bs = 1024*8
248 size = -1
Georg Brandlb9256022005-08-24 18:46:39 +0000249 read = 0
Georg Brandl5a650a22005-08-26 08:51:34 +0000250 blocknum = 0
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000251 if reporthook:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000252 if "content-length" in headers:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000253 size = int(headers["Content-Length"])
Georg Brandl5a650a22005-08-26 08:51:34 +0000254 reporthook(blocknum, bs, size)
255 while 1:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000256 block = fp.read(bs)
Georg Brandl5a650a22005-08-26 08:51:34 +0000257 if block == "":
258 break
Georg Brandlb9256022005-08-24 18:46:39 +0000259 read += len(block)
Georg Brandl5a650a22005-08-26 08:51:34 +0000260 tfp.write(block)
Georg Brandlb9256022005-08-24 18:46:39 +0000261 blocknum += 1
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000262 if reporthook:
263 reporthook(blocknum, bs, size)
264 fp.close()
265 tfp.close()
266 del fp
267 del tfp
Georg Brandlb9256022005-08-24 18:46:39 +0000268
269 # raise exception if actual size does not match content-length header
270 if size >= 0 and read < size:
271 raise ContentTooShortError("retrieval incomplete: got only %i out "
272 "of %i bytes" % (read, size), result)
273
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000274 return result
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000275
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000276 # Each method named open_<type> knows how to open that type of URL
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000277
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000278 def open_http(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000279 """Use HTTP protocol."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000280 import httplib
281 user_passwd = None
Martin v. Löwis3e865952006-01-24 15:51:21 +0000282 proxy_passwd= None
Walter Dörwald65230a22002-06-03 15:58:32 +0000283 if isinstance(url, str):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000284 host, selector = splithost(url)
285 if host:
286 user_passwd, host = splituser(host)
287 host = unquote(host)
288 realhost = host
289 else:
290 host, selector = url
Martin v. Löwis3e865952006-01-24 15:51:21 +0000291 # check whether the proxy contains authorization information
292 proxy_passwd, host = splituser(host)
293 # now we proceed with the url we want to obtain
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000294 urltype, rest = splittype(selector)
295 url = rest
296 user_passwd = None
Guido van Rossumb2493f82000-12-15 15:01:37 +0000297 if urltype.lower() != 'http':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000298 realhost = None
299 else:
300 realhost, rest = splithost(rest)
301 if realhost:
302 user_passwd, realhost = splituser(realhost)
303 if user_passwd:
304 selector = "%s://%s%s" % (urltype, realhost, rest)
Tim Peters55c12d42001-08-09 18:04:14 +0000305 if proxy_bypass(realhost):
306 host = realhost
307
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000308 #print "proxy via http:", host, selector
309 if not host: raise IOError, ('http error', 'no host given')
Tim Peters92037a12006-01-24 22:44:08 +0000310
Martin v. Löwis3e865952006-01-24 15:51:21 +0000311 if proxy_passwd:
312 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000313 proxy_auth = base64.b64encode(proxy_passwd).strip()
Martin v. Löwis3e865952006-01-24 15:51:21 +0000314 else:
315 proxy_auth = None
316
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000317 if user_passwd:
318 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000319 auth = base64.b64encode(user_passwd).strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000320 else:
321 auth = None
322 h = httplib.HTTP(host)
323 if data is not None:
324 h.putrequest('POST', selector)
Georg Brandl0619a322006-07-26 07:40:17 +0000325 h.putheader('Content-Type', 'application/x-www-form-urlencoded')
326 h.putheader('Content-Length', '%d' % len(data))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000327 else:
328 h.putrequest('GET', selector)
Martin v. Löwis3e865952006-01-24 15:51:21 +0000329 if proxy_auth: h.putheader('Proxy-Authorization', 'Basic %s' % proxy_auth)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000330 if auth: h.putheader('Authorization', 'Basic %s' % auth)
331 if realhost: h.putheader('Host', realhost)
Guido van Rossum68468eb2003-02-27 20:14:51 +0000332 for args in self.addheaders: h.putheader(*args)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000333 h.endheaders()
334 if data is not None:
Fred Drakeec3dfde2001-07-04 05:18:29 +0000335 h.send(data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000336 errcode, errmsg, headers = h.getreply()
Neal Norwitzce55e212007-03-20 08:14:57 +0000337 fp = h.getfile()
Georg Brandlf66b6032007-03-14 08:27:52 +0000338 if errcode == -1:
Neal Norwitzce55e212007-03-20 08:14:57 +0000339 if fp: fp.close()
Georg Brandlf66b6032007-03-14 08:27:52 +0000340 # something went wrong with the HTTP status line
341 raise IOError, ('http protocol error', 0,
342 'got a bad status line', None)
Sean Reifscheidera1afbf62007-09-19 07:52:56 +0000343 # According to RFC 2616, "2xx" code indicates that the client's
344 # request was successfully received, understood, and accepted.
Kurt B. Kaiser0f7c25d2008-01-02 04:11:28 +0000345 if (200 <= errcode < 300):
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000346 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000347 else:
348 if data is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000349 return self.http_error(url, fp, errcode, errmsg, headers)
Guido van Rossum29aab751999-03-09 19:31:21 +0000350 else:
351 return self.http_error(url, fp, errcode, errmsg, headers, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000352
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000353 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000354 """Handle http errors.
355 Derived class can override this, or provide specific handlers
356 named http_error_DDD where DDD is the 3-digit error code."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000357 # First check if there's a specific handler for this error
358 name = 'http_error_%d' % errcode
359 if hasattr(self, name):
360 method = getattr(self, name)
361 if data is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000362 result = method(url, fp, errcode, errmsg, headers)
Jeremy Hyltonb30f52a1999-02-25 16:14:58 +0000363 else:
364 result = method(url, fp, errcode, errmsg, headers, data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000365 if result: return result
Jeremy Hyltonb30f52a1999-02-25 16:14:58 +0000366 return self.http_error_default(url, fp, errcode, errmsg, headers)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000367
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000368 def http_error_default(self, url, fp, errcode, errmsg, headers):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000369 """Default error handler: close the connection and raise IOError."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000370 void = fp.read()
371 fp.close()
372 raise IOError, ('http error', errcode, errmsg, headers)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000373
Bill Janssen426ea0a2007-08-29 22:35:05 +0000374 if _have_ssl:
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000375 def open_https(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000376 """Use HTTPS protocol."""
Bill Janssen426ea0a2007-08-29 22:35:05 +0000377
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000378 import httplib
Fred Drake567ca8e2000-08-21 21:42:42 +0000379 user_passwd = None
Martin v. Löwis3e865952006-01-24 15:51:21 +0000380 proxy_passwd = None
Walter Dörwald65230a22002-06-03 15:58:32 +0000381 if isinstance(url, str):
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000382 host, selector = splithost(url)
Fred Drake567ca8e2000-08-21 21:42:42 +0000383 if host:
384 user_passwd, host = splituser(host)
385 host = unquote(host)
386 realhost = host
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000387 else:
388 host, selector = url
Martin v. Löwis3e865952006-01-24 15:51:21 +0000389 # here, we determine, whether the proxy contains authorization information
390 proxy_passwd, host = splituser(host)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000391 urltype, rest = splittype(selector)
Fred Drake567ca8e2000-08-21 21:42:42 +0000392 url = rest
393 user_passwd = None
Guido van Rossumb2493f82000-12-15 15:01:37 +0000394 if urltype.lower() != 'https':
Fred Drake567ca8e2000-08-21 21:42:42 +0000395 realhost = None
396 else:
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000397 realhost, rest = splithost(rest)
Fred Drake567ca8e2000-08-21 21:42:42 +0000398 if realhost:
399 user_passwd, realhost = splituser(realhost)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000400 if user_passwd:
401 selector = "%s://%s%s" % (urltype, realhost, rest)
Andrew M. Kuchling7ad47922000-06-10 01:41:48 +0000402 #print "proxy via https:", host, selector
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000403 if not host: raise IOError, ('https error', 'no host given')
Martin v. Löwis3e865952006-01-24 15:51:21 +0000404 if proxy_passwd:
405 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000406 proxy_auth = base64.b64encode(proxy_passwd).strip()
Martin v. Löwis3e865952006-01-24 15:51:21 +0000407 else:
408 proxy_auth = None
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000409 if user_passwd:
410 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000411 auth = base64.b64encode(user_passwd).strip()
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000412 else:
413 auth = None
414 h = httplib.HTTPS(host, 0,
415 key_file=self.key_file,
416 cert_file=self.cert_file)
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000417 if data is not None:
418 h.putrequest('POST', selector)
Georg Brandl0619a322006-07-26 07:40:17 +0000419 h.putheader('Content-Type',
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000420 'application/x-www-form-urlencoded')
Georg Brandl0619a322006-07-26 07:40:17 +0000421 h.putheader('Content-Length', '%d' % len(data))
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000422 else:
423 h.putrequest('GET', selector)
Andrew M. Kuchling52278572006-12-19 15:11:41 +0000424 if proxy_auth: h.putheader('Proxy-Authorization', 'Basic %s' % proxy_auth)
425 if auth: h.putheader('Authorization', 'Basic %s' % auth)
Fred Drake567ca8e2000-08-21 21:42:42 +0000426 if realhost: h.putheader('Host', realhost)
Guido van Rossum68468eb2003-02-27 20:14:51 +0000427 for args in self.addheaders: h.putheader(*args)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000428 h.endheaders()
Andrew M. Kuchling43c5af02000-04-24 14:17:06 +0000429 if data is not None:
Fred Drakeec3dfde2001-07-04 05:18:29 +0000430 h.send(data)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000431 errcode, errmsg, headers = h.getreply()
Neal Norwitzce55e212007-03-20 08:14:57 +0000432 fp = h.getfile()
Georg Brandlf66b6032007-03-14 08:27:52 +0000433 if errcode == -1:
Neal Norwitzce55e212007-03-20 08:14:57 +0000434 if fp: fp.close()
Georg Brandlf66b6032007-03-14 08:27:52 +0000435 # something went wrong with the HTTP status line
436 raise IOError, ('http protocol error', 0,
437 'got a bad status line', None)
Georg Brandl9b915672007-09-24 18:08:24 +0000438 # According to RFC 2616, "2xx" code indicates that the client's
439 # request was successfully received, understood, and accepted.
Kurt B. Kaiser0f7c25d2008-01-02 04:11:28 +0000440 if (200 <= errcode < 300):
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000441 return addinfourl(fp, headers, "https:" + url, errcode)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000442 else:
Fred Drake567ca8e2000-08-21 21:42:42 +0000443 if data is None:
444 return self.http_error(url, fp, errcode, errmsg, headers)
445 else:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000446 return self.http_error(url, fp, errcode, errmsg, headers,
447 data)
Fred Drake567ca8e2000-08-21 21:42:42 +0000448
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000449 def open_file(self, url):
Neal Norwitzc5d0dbd2006-04-09 04:00:49 +0000450 """Use local file or FTP depending on form of URL."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000451 if not isinstance(url, str):
452 raise IOError, ('file error', 'proxy support for file protocol currently not implemented')
Jack Jansen4ef11032002-09-12 20:14:04 +0000453 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000454 return self.open_ftp(url)
455 else:
456 return self.open_local_file(url)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000457
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000458 def open_local_file(self, url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000459 """Use local file."""
Georg Brandl5a096e12007-01-22 19:40:21 +0000460 import mimetypes, mimetools, email.utils
Raymond Hettingera6172712004-12-31 19:15:26 +0000461 try:
462 from cStringIO import StringIO
463 except ImportError:
464 from StringIO import StringIO
Guido van Rossumf0713d32001-08-09 17:43:35 +0000465 host, file = splithost(url)
466 localname = url2pathname(file)
Guido van Rossuma2da3052002-04-15 00:25:01 +0000467 try:
468 stats = os.stat(localname)
469 except OSError, e:
470 raise IOError(e.errno, e.strerror, e.filename)
Walter Dörwald92b48b72002-03-22 17:30:38 +0000471 size = stats.st_size
Georg Brandl5a096e12007-01-22 19:40:21 +0000472 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000473 mtype = mimetypes.guess_type(url)[0]
Raymond Hettingera6172712004-12-31 19:15:26 +0000474 headers = mimetools.Message(StringIO(
Guido van Rossumf0713d32001-08-09 17:43:35 +0000475 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
476 (mtype or 'text/plain', size, modified)))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000477 if not host:
Guido van Rossum336a2011999-06-24 15:27:36 +0000478 urlfile = file
479 if file[:1] == '/':
480 urlfile = 'file://' + file
Guido van Rossumf0713d32001-08-09 17:43:35 +0000481 return addinfourl(open(localname, 'rb'),
Guido van Rossum336a2011999-06-24 15:27:36 +0000482 headers, urlfile)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000483 host, port = splitport(host)
484 if not port \
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000485 and socket.gethostbyname(host) in (localhost(), thishost()):
Guido van Rossum336a2011999-06-24 15:27:36 +0000486 urlfile = file
487 if file[:1] == '/':
488 urlfile = 'file://' + file
Guido van Rossumf0713d32001-08-09 17:43:35 +0000489 return addinfourl(open(localname, 'rb'),
Guido van Rossum336a2011999-06-24 15:27:36 +0000490 headers, urlfile)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000491 raise IOError, ('local file error', 'not on local host')
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000492
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000493 def open_ftp(self, url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000494 """Use FTP protocol."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000495 if not isinstance(url, str):
496 raise IOError, ('ftp error', 'proxy support for ftp protocol currently not implemented')
Raymond Hettingera6172712004-12-31 19:15:26 +0000497 import mimetypes, mimetools
498 try:
499 from cStringIO import StringIO
500 except ImportError:
501 from StringIO import StringIO
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000502 host, path = splithost(url)
503 if not host: raise IOError, ('ftp error', 'no host given')
504 host, port = splitport(host)
505 user, host = splituser(host)
506 if user: user, passwd = splitpasswd(user)
507 else: passwd = None
508 host = unquote(host)
509 user = unquote(user or '')
510 passwd = unquote(passwd or '')
511 host = socket.gethostbyname(host)
512 if not port:
513 import ftplib
514 port = ftplib.FTP_PORT
515 else:
516 port = int(port)
517 path, attrs = splitattr(path)
518 path = unquote(path)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000519 dirs = path.split('/')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000520 dirs, file = dirs[:-1], dirs[-1]
521 if dirs and not dirs[0]: dirs = dirs[1:]
Guido van Rossum5e006a31999-08-18 17:40:33 +0000522 if dirs and not dirs[0]: dirs[0] = '/'
Guido van Rossumb2493f82000-12-15 15:01:37 +0000523 key = user, host, port, '/'.join(dirs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000524 # XXX thread unsafe!
525 if len(self.ftpcache) > MAXFTPCACHE:
526 # Prune the cache, rather arbitrarily
527 for k in self.ftpcache.keys():
528 if k != key:
529 v = self.ftpcache[k]
530 del self.ftpcache[k]
531 v.close()
532 try:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000533 if not key in self.ftpcache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000534 self.ftpcache[key] = \
535 ftpwrapper(user, passwd, host, port, dirs)
536 if not file: type = 'D'
537 else: type = 'I'
538 for attr in attrs:
539 attr, value = splitvalue(attr)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000540 if attr.lower() == 'type' and \
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000541 value in ('a', 'A', 'i', 'I', 'd', 'D'):
Guido van Rossumb2493f82000-12-15 15:01:37 +0000542 type = value.upper()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000543 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
Guido van Rossum88e0b5b2001-08-23 13:38:15 +0000544 mtype = mimetypes.guess_type("ftp:" + url)[0]
545 headers = ""
546 if mtype:
547 headers += "Content-Type: %s\n" % mtype
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000548 if retrlen is not None and retrlen >= 0:
Guido van Rossum88e0b5b2001-08-23 13:38:15 +0000549 headers += "Content-Length: %d\n" % retrlen
Raymond Hettingera6172712004-12-31 19:15:26 +0000550 headers = mimetools.Message(StringIO(headers))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000551 return addinfourl(fp, headers, "ftp:" + url)
552 except ftperrors(), msg:
553 raise IOError, ('ftp error', msg), sys.exc_info()[2]
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000554
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000555 def open_data(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000556 """Use "data" URL."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000557 if not isinstance(url, str):
558 raise IOError, ('data error', 'proxy support for data protocol currently not implemented')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000559 # ignore POSTed data
560 #
561 # syntax of data URLs:
562 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
563 # mediatype := [ type "/" subtype ] *( ";" parameter )
564 # data := *urlchar
565 # parameter := attribute "=" value
Raymond Hettingera6172712004-12-31 19:15:26 +0000566 import mimetools
567 try:
568 from cStringIO import StringIO
569 except ImportError:
570 from StringIO import StringIO
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000571 try:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000572 [type, data] = url.split(',', 1)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000573 except ValueError:
574 raise IOError, ('data error', 'bad data URL')
575 if not type:
576 type = 'text/plain;charset=US-ASCII'
Guido van Rossumb2493f82000-12-15 15:01:37 +0000577 semi = type.rfind(';')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000578 if semi >= 0 and '=' not in type[semi:]:
579 encoding = type[semi+1:]
580 type = type[:semi]
581 else:
582 encoding = ''
583 msg = []
584 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %T GMT',
585 time.gmtime(time.time())))
586 msg.append('Content-type: %s' % type)
587 if encoding == 'base64':
588 import base64
589 data = base64.decodestring(data)
590 else:
591 data = unquote(data)
Georg Brandl0619a322006-07-26 07:40:17 +0000592 msg.append('Content-Length: %d' % len(data))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000593 msg.append('')
594 msg.append(data)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000595 msg = '\n'.join(msg)
Raymond Hettingera6172712004-12-31 19:15:26 +0000596 f = StringIO(msg)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000597 headers = mimetools.Message(f, 0)
Georg Brandl1f663572005-11-26 16:50:44 +0000598 #f.fileno = None # needed for addinfourl
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000599 return addinfourl(f, headers, url)
Guido van Rossum6d4d1c21998-03-12 14:32:55 +0000600
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000601
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000602class FancyURLopener(URLopener):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000603 """Derived class with handlers for errors we can handle (perhaps)."""
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000604
Neal Norwitz60e04cd2002-06-11 13:38:51 +0000605 def __init__(self, *args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +0000606 URLopener.__init__(self, *args, **kwargs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000607 self.auth_cache = {}
Skip Montanaroc3e11d62001-02-15 16:56:36 +0000608 self.tries = 0
609 self.maxtries = 10
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000610
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000611 def http_error_default(self, url, fp, errcode, errmsg, headers):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000612 """Default error handling -- don't raise an exception."""
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000613 return addinfourl(fp, headers, "http:" + url, errcode)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000614
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000615 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000616 """Error 302 -- relocated (temporarily)."""
Skip Montanaroc3e11d62001-02-15 16:56:36 +0000617 self.tries += 1
618 if self.maxtries and self.tries >= self.maxtries:
619 if hasattr(self, "http_error_500"):
620 meth = self.http_error_500
621 else:
622 meth = self.http_error_default
623 self.tries = 0
624 return meth(url, fp, 500,
625 "Internal Server Error: Redirect Recursion", headers)
626 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
627 data)
628 self.tries = 0
629 return result
630
631 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
Raymond Hettinger54f02222002-06-01 14:18:47 +0000632 if 'location' in headers:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000633 newurl = headers['location']
Raymond Hettinger54f02222002-06-01 14:18:47 +0000634 elif 'uri' in headers:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000635 newurl = headers['uri']
636 else:
637 return
638 void = fp.read()
639 fp.close()
Guido van Rossum3527f591999-03-29 20:23:41 +0000640 # In case the server sent a relative URL, join with original:
Moshe Zadka5d87d472001-04-09 14:54:21 +0000641 newurl = basejoin(self.type + ":" + url, newurl)
Guido van Rossumfa19f7c2003-05-16 01:46:51 +0000642 return self.open(newurl)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000643
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000644 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000645 """Error 301 -- also relocated (permanently)."""
646 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
Guido van Rossume6ad8911996-09-10 17:02:56 +0000647
Raymond Hettinger024aaa12003-04-24 15:32:12 +0000648 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
649 """Error 303 -- also relocated (essentially identical to 302)."""
650 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
651
Guido van Rossumfa19f7c2003-05-16 01:46:51 +0000652 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
653 """Error 307 -- relocated, but turn POST into error."""
654 if data is None:
655 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
656 else:
657 return self.http_error_default(url, fp, errcode, errmsg, headers)
658
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000659 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000660 """Error 401 -- authentication required.
Martin v. Löwis3e865952006-01-24 15:51:21 +0000661 This function supports Basic authentication only."""
Raymond Hettinger54f02222002-06-01 14:18:47 +0000662 if not 'www-authenticate' in headers:
Tim Peters85ba6732001-02-28 08:26:44 +0000663 URLopener.http_error_default(self, url, fp,
Fred Drakec680ae82001-10-13 18:37:07 +0000664 errcode, errmsg, headers)
Moshe Zadkae99bd172001-02-27 06:27:04 +0000665 stuff = headers['www-authenticate']
666 import re
667 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
668 if not match:
Tim Peters85ba6732001-02-28 08:26:44 +0000669 URLopener.http_error_default(self, url, fp,
Moshe Zadkae99bd172001-02-27 06:27:04 +0000670 errcode, errmsg, headers)
671 scheme, realm = match.groups()
672 if scheme.lower() != 'basic':
Tim Peters85ba6732001-02-28 08:26:44 +0000673 URLopener.http_error_default(self, url, fp,
Moshe Zadkae99bd172001-02-27 06:27:04 +0000674 errcode, errmsg, headers)
675 name = 'retry_' + self.type + '_basic_auth'
676 if data is None:
677 return getattr(self,name)(url, realm)
678 else:
679 return getattr(self,name)(url, realm, data)
Tim Peters92037a12006-01-24 22:44:08 +0000680
Martin v. Löwis3e865952006-01-24 15:51:21 +0000681 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None):
682 """Error 407 -- proxy authentication required.
683 This function supports Basic authentication only."""
684 if not 'proxy-authenticate' in headers:
685 URLopener.http_error_default(self, url, fp,
686 errcode, errmsg, headers)
687 stuff = headers['proxy-authenticate']
688 import re
689 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
690 if not match:
691 URLopener.http_error_default(self, url, fp,
692 errcode, errmsg, headers)
693 scheme, realm = match.groups()
694 if scheme.lower() != 'basic':
695 URLopener.http_error_default(self, url, fp,
696 errcode, errmsg, headers)
697 name = 'retry_proxy_' + self.type + '_basic_auth'
698 if data is None:
699 return getattr(self,name)(url, realm)
700 else:
701 return getattr(self,name)(url, realm, data)
Tim Peters92037a12006-01-24 22:44:08 +0000702
Martin v. Löwis3e865952006-01-24 15:51:21 +0000703 def retry_proxy_http_basic_auth(self, url, realm, data=None):
704 host, selector = splithost(url)
705 newurl = 'http://' + host + selector
706 proxy = self.proxies['http']
707 urltype, proxyhost = splittype(proxy)
708 proxyhost, proxyselector = splithost(proxyhost)
709 i = proxyhost.find('@') + 1
710 proxyhost = proxyhost[i:]
711 user, passwd = self.get_user_passwd(proxyhost, realm, i)
712 if not (user or passwd): return None
713 proxyhost = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + proxyhost
714 self.proxies['http'] = 'http://' + proxyhost + proxyselector
715 if data is None:
716 return self.open(newurl)
717 else:
718 return self.open(newurl, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000719
Martin v. Löwis3e865952006-01-24 15:51:21 +0000720 def retry_proxy_https_basic_auth(self, url, realm, data=None):
721 host, selector = splithost(url)
722 newurl = 'https://' + host + selector
723 proxy = self.proxies['https']
724 urltype, proxyhost = splittype(proxy)
725 proxyhost, proxyselector = splithost(proxyhost)
726 i = proxyhost.find('@') + 1
727 proxyhost = proxyhost[i:]
728 user, passwd = self.get_user_passwd(proxyhost, realm, i)
729 if not (user or passwd): return None
730 proxyhost = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + proxyhost
731 self.proxies['https'] = 'https://' + proxyhost + proxyselector
732 if data is None:
733 return self.open(newurl)
734 else:
735 return self.open(newurl, data)
Tim Peters92037a12006-01-24 22:44:08 +0000736
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000737 def retry_http_basic_auth(self, url, realm, data=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000738 host, selector = splithost(url)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000739 i = host.find('@') + 1
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000740 host = host[i:]
741 user, passwd = self.get_user_passwd(host, realm, i)
742 if not (user or passwd): return None
Guido van Rossumafc4f042001-01-15 18:31:13 +0000743 host = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + host
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000744 newurl = 'http://' + host + selector
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000745 if data is None:
746 return self.open(newurl)
747 else:
748 return self.open(newurl, data)
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000749
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000750 def retry_https_basic_auth(self, url, realm, data=None):
Tim Peterse1190062001-01-15 03:34:38 +0000751 host, selector = splithost(url)
752 i = host.find('@') + 1
753 host = host[i:]
754 user, passwd = self.get_user_passwd(host, realm, i)
755 if not (user or passwd): return None
Guido van Rossumafc4f042001-01-15 18:31:13 +0000756 host = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + host
Martin v. Löwis3e865952006-01-24 15:51:21 +0000757 newurl = 'https://' + host + selector
758 if data is None:
759 return self.open(newurl)
760 else:
761 return self.open(newurl, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000762
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000763 def get_user_passwd(self, host, realm, clear_cache = 0):
Guido van Rossumb2493f82000-12-15 15:01:37 +0000764 key = realm + '@' + host.lower()
Raymond Hettinger54f02222002-06-01 14:18:47 +0000765 if key in self.auth_cache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000766 if clear_cache:
767 del self.auth_cache[key]
768 else:
769 return self.auth_cache[key]
770 user, passwd = self.prompt_user_passwd(host, realm)
771 if user or passwd: self.auth_cache[key] = (user, passwd)
772 return user, passwd
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000773
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000774 def prompt_user_passwd(self, host, realm):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000775 """Override this in a GUI environment!"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000776 import getpass
777 try:
778 user = raw_input("Enter username for %s at %s: " % (realm,
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000779 host))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000780 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
781 (user, realm, host))
782 return user, passwd
783 except KeyboardInterrupt:
784 print
785 return None, None
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000786
787
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000788# Utility functions
789
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000790_localhost = None
791def localhost():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000792 """Return the IP address of the magic hostname 'localhost'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000793 global _localhost
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000794 if _localhost is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000795 _localhost = socket.gethostbyname('localhost')
796 return _localhost
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000797
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000798_thishost = None
799def thishost():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000800 """Return the IP address of the current host."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000801 global _thishost
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000802 if _thishost is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000803 _thishost = socket.gethostbyname(socket.gethostname())
804 return _thishost
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000805
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000806_ftperrors = None
807def ftperrors():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000808 """Return the set of errors raised by the FTP class."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000809 global _ftperrors
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000810 if _ftperrors is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000811 import ftplib
812 _ftperrors = ftplib.all_errors
813 return _ftperrors
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000814
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000815_noheaders = None
816def noheaders():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000817 """Return an empty mimetools.Message object."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000818 global _noheaders
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000819 if _noheaders is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000820 import mimetools
Raymond Hettingera6172712004-12-31 19:15:26 +0000821 try:
822 from cStringIO import StringIO
823 except ImportError:
824 from StringIO import StringIO
825 _noheaders = mimetools.Message(StringIO(), 0)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000826 _noheaders.fp.close() # Recycle file descriptor
827 return _noheaders
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000828
829
830# Utility classes
831
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000832class ftpwrapper:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000833 """Class used by open_ftp() for cache of open FTP connections."""
834
Facundo Batista4f1b1ed2008-05-29 16:39:26 +0000835 def __init__(self, user, passwd, host, port, dirs,
836 timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000837 self.user = user
838 self.passwd = passwd
839 self.host = host
840 self.port = port
841 self.dirs = dirs
Facundo Batista711a54e2007-05-24 17:50:54 +0000842 self.timeout = timeout
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000843 self.init()
Guido van Rossume7b146f2000-02-04 15:28:42 +0000844
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000845 def init(self):
846 import ftplib
847 self.busy = 0
848 self.ftp = ftplib.FTP()
Facundo Batista711a54e2007-05-24 17:50:54 +0000849 self.ftp.connect(self.host, self.port, self.timeout)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000850 self.ftp.login(self.user, self.passwd)
851 for dir in self.dirs:
852 self.ftp.cwd(dir)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000853
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000854 def retrfile(self, file, type):
855 import ftplib
856 self.endtransfer()
857 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
858 else: cmd = 'TYPE ' + type; isdir = 0
859 try:
860 self.ftp.voidcmd(cmd)
861 except ftplib.all_errors:
862 self.init()
863 self.ftp.voidcmd(cmd)
864 conn = None
865 if file and not isdir:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000866 # Try to retrieve as a file
867 try:
868 cmd = 'RETR ' + file
869 conn = self.ftp.ntransfercmd(cmd)
870 except ftplib.error_perm, reason:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000871 if str(reason)[:3] != '550':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000872 raise IOError, ('ftp error', reason), sys.exc_info()[2]
873 if not conn:
874 # Set transfer mode to ASCII!
875 self.ftp.voidcmd('TYPE A')
Georg Brandld5e6cf22008-01-20 12:18:17 +0000876 # Try a directory listing. Verify that directory exists.
877 if file:
878 pwd = self.ftp.pwd()
879 try:
880 try:
881 self.ftp.cwd(file)
882 except ftplib.error_perm, reason:
883 raise IOError, ('ftp error', reason), sys.exc_info()[2]
884 finally:
885 self.ftp.cwd(pwd)
886 cmd = 'LIST ' + file
887 else:
888 cmd = 'LIST'
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000889 conn = self.ftp.ntransfercmd(cmd)
890 self.busy = 1
891 # Pass back both a suitably decorated object and a retrieval length
892 return (addclosehook(conn[0].makefile('rb'),
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000893 self.endtransfer), conn[1])
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000894 def endtransfer(self):
895 if not self.busy:
896 return
897 self.busy = 0
898 try:
899 self.ftp.voidresp()
900 except ftperrors():
901 pass
Guido van Rossume7b146f2000-02-04 15:28:42 +0000902
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000903 def close(self):
904 self.endtransfer()
905 try:
906 self.ftp.close()
907 except ftperrors():
908 pass
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000909
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000910class addbase:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000911 """Base class for addinfo and addclosehook."""
912
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000913 def __init__(self, fp):
914 self.fp = fp
915 self.read = self.fp.read
916 self.readline = self.fp.readline
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000917 if hasattr(self.fp, "readlines"): self.readlines = self.fp.readlines
Georg Brandl1f663572005-11-26 16:50:44 +0000918 if hasattr(self.fp, "fileno"):
919 self.fileno = self.fp.fileno
920 else:
921 self.fileno = lambda: None
Raymond Hettinger42182eb2003-03-09 05:33:33 +0000922 if hasattr(self.fp, "__iter__"):
923 self.__iter__ = self.fp.__iter__
924 if hasattr(self.fp, "next"):
925 self.next = self.fp.next
Guido van Rossume7b146f2000-02-04 15:28:42 +0000926
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000927 def __repr__(self):
Walter Dörwald70a6b492004-02-12 17:35:32 +0000928 return '<%s at %r whose fp = %r>' % (self.__class__.__name__,
929 id(self), self.fp)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000930
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000931 def close(self):
932 self.read = None
933 self.readline = None
934 self.readlines = None
935 self.fileno = None
936 if self.fp: self.fp.close()
937 self.fp = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000938
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000939class addclosehook(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000940 """Class to add a close hook to an open file."""
941
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000942 def __init__(self, fp, closehook, *hookargs):
943 addbase.__init__(self, fp)
944 self.closehook = closehook
945 self.hookargs = hookargs
Guido van Rossume7b146f2000-02-04 15:28:42 +0000946
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000947 def close(self):
Guido van Rossumc580dae2000-05-24 13:21:46 +0000948 addbase.close(self)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000949 if self.closehook:
Guido van Rossum68468eb2003-02-27 20:14:51 +0000950 self.closehook(*self.hookargs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000951 self.closehook = None
952 self.hookargs = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000953
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000954class addinfo(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000955 """class to add an info() method to an open file."""
956
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000957 def __init__(self, fp, headers):
958 addbase.__init__(self, fp)
959 self.headers = headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000960
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000961 def info(self):
962 return self.headers
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000963
Guido van Rossume6ad8911996-09-10 17:02:56 +0000964class addinfourl(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000965 """class to add info() and geturl() methods to an open file."""
966
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000967 def __init__(self, fp, headers, url, code=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000968 addbase.__init__(self, fp)
969 self.headers = headers
970 self.url = url
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000971 self.code = code
Guido van Rossume7b146f2000-02-04 15:28:42 +0000972
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000973 def info(self):
974 return self.headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000975
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000976 def getcode(self):
977 return self.code
978
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000979 def geturl(self):
980 return self.url
Guido van Rossume6ad8911996-09-10 17:02:56 +0000981
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000982
Guido van Rossum7c395db1994-07-04 22:14:49 +0000983# Utilities to parse URLs (most of these return None for missing parts):
Sjoerd Mullendere0371b81995-11-10 10:36:07 +0000984# unwrap('<URL:type://host/path>') --> 'type://host/path'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000985# splittype('type:opaquestring') --> 'type', 'opaquestring'
986# splithost('//host[:port]/path') --> 'host[:port]', '/path'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000987# splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'
988# splitpasswd('user:passwd') -> 'user', 'passwd'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000989# splitport('host:port') --> 'host', 'port'
990# splitquery('/path?query') --> '/path', 'query'
991# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000992# splitattr('/path;attr1=value1;attr2=value2;...') ->
993# '/path', ['attr1=value1', 'attr2=value2', ...]
994# splitvalue('attr=value') --> 'attr', 'value'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000995# unquote('abc%20def') -> 'abc def'
996# quote('abc def') -> 'abc%20def')
997
Walter Dörwald65230a22002-06-03 15:58:32 +0000998try:
999 unicode
1000except NameError:
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001001 def _is_unicode(x):
1002 return 0
Walter Dörwald65230a22002-06-03 15:58:32 +00001003else:
1004 def _is_unicode(x):
1005 return isinstance(x, unicode)
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001006
Martin v. Löwis1d994332000-12-03 18:30:10 +00001007def toBytes(url):
1008 """toBytes(u"URL") --> 'URL'."""
1009 # Most URL schemes require ASCII. If that changes, the conversion
1010 # can be relaxed
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001011 if _is_unicode(url):
Martin v. Löwis1d994332000-12-03 18:30:10 +00001012 try:
1013 url = url.encode("ASCII")
1014 except UnicodeError:
Guido van Rossumb2493f82000-12-15 15:01:37 +00001015 raise UnicodeError("URL " + repr(url) +
1016 " contains non-ASCII characters")
Martin v. Löwis1d994332000-12-03 18:30:10 +00001017 return url
1018
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001019def unwrap(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001020 """unwrap('<URL:type://host/path>') --> 'type://host/path'."""
Guido van Rossumb2493f82000-12-15 15:01:37 +00001021 url = url.strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001022 if url[:1] == '<' and url[-1:] == '>':
Guido van Rossumb2493f82000-12-15 15:01:37 +00001023 url = url[1:-1].strip()
1024 if url[:4] == 'URL:': url = url[4:].strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001025 return url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001026
Guido van Rossum332e1441997-09-29 23:23:46 +00001027_typeprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001028def splittype(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001029 """splittype('type:opaquestring') --> 'type', 'opaquestring'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001030 global _typeprog
1031 if _typeprog is None:
1032 import re
1033 _typeprog = re.compile('^([^/:]+):')
Guido van Rossum332e1441997-09-29 23:23:46 +00001034
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001035 match = _typeprog.match(url)
1036 if match:
1037 scheme = match.group(1)
Fred Drake9e94afd2000-07-01 07:03:30 +00001038 return scheme.lower(), url[len(scheme) + 1:]
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001039 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001040
Guido van Rossum332e1441997-09-29 23:23:46 +00001041_hostprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001042def splithost(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001043 """splithost('//host[:port]/path') --> 'host[:port]', '/path'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001044 global _hostprog
1045 if _hostprog is None:
1046 import re
Georg Brandl1c168d82006-03-26 20:59:38 +00001047 _hostprog = re.compile('^//([^/?]*)(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001048
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001049 match = _hostprog.match(url)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001050 if match: return match.group(1, 2)
1051 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001052
Guido van Rossum332e1441997-09-29 23:23:46 +00001053_userprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001054def splituser(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001055 """splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001056 global _userprog
1057 if _userprog is None:
1058 import re
Raymond Hettingerf2e45dd2002-08-18 20:08:56 +00001059 _userprog = re.compile('^(.*)@(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001060
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001061 match = _userprog.match(host)
Fred Drake567ca8e2000-08-21 21:42:42 +00001062 if match: return map(unquote, match.group(1, 2))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001063 return None, host
Guido van Rossum7c395db1994-07-04 22:14:49 +00001064
Guido van Rossum332e1441997-09-29 23:23:46 +00001065_passwdprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001066def splitpasswd(user):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001067 """splitpasswd('user:passwd') -> 'user', 'passwd'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001068 global _passwdprog
1069 if _passwdprog is None:
1070 import re
1071 _passwdprog = re.compile('^([^:]*):(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001072
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001073 match = _passwdprog.match(user)
1074 if match: return match.group(1, 2)
1075 return user, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001076
Guido van Rossume7b146f2000-02-04 15:28:42 +00001077# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum332e1441997-09-29 23:23:46 +00001078_portprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001079def splitport(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001080 """splitport('host:port') --> 'host', 'port'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001081 global _portprog
1082 if _portprog is None:
1083 import re
1084 _portprog = re.compile('^(.*):([0-9]+)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001085
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001086 match = _portprog.match(host)
1087 if match: return match.group(1, 2)
1088 return host, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001089
Guido van Rossum332e1441997-09-29 23:23:46 +00001090_nportprog = None
Guido van Rossum53725a21996-06-13 19:12:35 +00001091def splitnport(host, defport=-1):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001092 """Split host and port, returning numeric port.
1093 Return given default port if no ':' found; defaults to -1.
1094 Return numerical port if a valid number are found after ':'.
1095 Return None if ':' but not a valid number."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001096 global _nportprog
1097 if _nportprog is None:
1098 import re
1099 _nportprog = re.compile('^(.*):(.*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001100
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001101 match = _nportprog.match(host)
1102 if match:
1103 host, port = match.group(1, 2)
1104 try:
Guido van Rossumb2493f82000-12-15 15:01:37 +00001105 if not port: raise ValueError, "no digits"
1106 nport = int(port)
1107 except ValueError:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001108 nport = None
1109 return host, nport
1110 return host, defport
Guido van Rossum53725a21996-06-13 19:12:35 +00001111
Guido van Rossum332e1441997-09-29 23:23:46 +00001112_queryprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001113def splitquery(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001114 """splitquery('/path?query') --> '/path', 'query'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001115 global _queryprog
1116 if _queryprog is None:
1117 import re
1118 _queryprog = re.compile('^(.*)\?([^?]*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001119
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001120 match = _queryprog.match(url)
1121 if match: return match.group(1, 2)
1122 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001123
Guido van Rossum332e1441997-09-29 23:23:46 +00001124_tagprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001125def splittag(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001126 """splittag('/path#tag') --> '/path', 'tag'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001127 global _tagprog
1128 if _tagprog is None:
1129 import re
1130 _tagprog = re.compile('^(.*)#([^#]*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001131
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001132 match = _tagprog.match(url)
1133 if match: return match.group(1, 2)
1134 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001135
Guido van Rossum7c395db1994-07-04 22:14:49 +00001136def splitattr(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001137 """splitattr('/path;attr1=value1;attr2=value2;...') ->
1138 '/path', ['attr1=value1', 'attr2=value2', ...]."""
Guido van Rossumb2493f82000-12-15 15:01:37 +00001139 words = url.split(';')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001140 return words[0], words[1:]
Guido van Rossum7c395db1994-07-04 22:14:49 +00001141
Guido van Rossum332e1441997-09-29 23:23:46 +00001142_valueprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001143def splitvalue(attr):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001144 """splitvalue('attr=value') --> 'attr', 'value'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001145 global _valueprog
1146 if _valueprog is None:
1147 import re
1148 _valueprog = re.compile('^([^=]*)=(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001149
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001150 match = _valueprog.match(attr)
1151 if match: return match.group(1, 2)
1152 return attr, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001153
Raymond Hettinger803ce802005-09-10 06:49:04 +00001154_hextochr = dict(('%02x' % i, chr(i)) for i in range(256))
1155_hextochr.update(('%02X' % i, chr(i)) for i in range(256))
1156
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001157def unquote(s):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001158 """unquote('abc%20def') -> 'abc def'."""
Raymond Hettinger803ce802005-09-10 06:49:04 +00001159 res = s.split('%')
1160 for i in xrange(1, len(res)):
1161 item = res[i]
1162 try:
1163 res[i] = _hextochr[item[:2]] + item[2:]
1164 except KeyError:
1165 res[i] = '%' + item
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +00001166 except UnicodeDecodeError:
1167 res[i] = unichr(int(item[:2], 16)) + item[2:]
Guido van Rossumb2493f82000-12-15 15:01:37 +00001168 return "".join(res)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001169
Guido van Rossum0564e121996-12-13 14:47:36 +00001170def unquote_plus(s):
Skip Montanaro79f1c172000-08-22 03:00:52 +00001171 """unquote('%7e/abc+def') -> '~/abc def'"""
Brett Cannonaaeffaf2004-03-23 23:50:17 +00001172 s = s.replace('+', ' ')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001173 return unquote(s)
Guido van Rossum0564e121996-12-13 14:47:36 +00001174
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001175always_safe = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
Jeremy Hylton6102e292000-08-31 15:48:10 +00001176 'abcdefghijklmnopqrstuvwxyz'
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001177 '0123456789' '_.-')
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001178_safemaps = {}
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001179
Guido van Rossum7c395db1994-07-04 22:14:49 +00001180def quote(s, safe = '/'):
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001181 """quote('abc def') -> 'abc%20def'
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001182
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001183 Each part of a URL, e.g. the path info, the query, etc., has a
1184 different set of reserved characters that must be quoted.
1185
1186 RFC 2396 Uniform Resource Identifiers (URI): Generic Syntax lists
1187 the following reserved characters.
1188
1189 reserved = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" |
1190 "$" | ","
1191
1192 Each of these characters is reserved in some component of a URL,
1193 but not necessarily in all of them.
1194
1195 By default, the quote function is intended for quoting the path
1196 section of a URL. Thus, it will not encode '/'. This character
1197 is reserved, but in typical usage the quote function is being
1198 called on a path where the existing slash characters are used as
1199 reserved characters.
1200 """
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001201 cachekey = (safe, always_safe)
1202 try:
1203 safe_map = _safemaps[cachekey]
1204 except KeyError:
1205 safe += always_safe
1206 safe_map = {}
1207 for i in range(256):
1208 c = chr(i)
1209 safe_map[c] = (c in safe) and c or ('%%%02X' % i)
1210 _safemaps[cachekey] = safe_map
1211 res = map(safe_map.__getitem__, s)
Guido van Rossumb2493f82000-12-15 15:01:37 +00001212 return ''.join(res)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001213
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001214def quote_plus(s, safe = ''):
1215 """Quote the query fragment of a URL; replacing ' ' with '+'"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001216 if ' ' in s:
Raymond Hettingercf6b6322005-09-10 18:17:54 +00001217 s = quote(s, safe + ' ')
1218 return s.replace(' ', '+')
1219 return quote(s, safe)
Guido van Rossum0564e121996-12-13 14:47:36 +00001220
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001221def urlencode(query,doseq=0):
1222 """Encode a sequence of two-element tuples or dictionary into a URL query string.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001223
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001224 If any values in the query arg are sequences and doseq is true, each
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001225 sequence element is converted to a separate parameter.
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001226
1227 If the query arg is a sequence of two-element tuples, the order of the
1228 parameters in the output will match the order of parameters in the
1229 input.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001230 """
Tim Peters658cba62001-02-09 20:06:00 +00001231
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001232 if hasattr(query,"items"):
1233 # mapping objects
1234 query = query.items()
1235 else:
1236 # it's a bother at times that strings and string-like objects are
1237 # sequences...
1238 try:
1239 # non-sequence items should not work with len()
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001240 # non-empty strings will fail this
Walter Dörwald65230a22002-06-03 15:58:32 +00001241 if len(query) and not isinstance(query[0], tuple):
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001242 raise TypeError
1243 # zero-length sequences of all types will get here and succeed,
1244 # but that's a minor nit - since the original implementation
1245 # allowed empty dicts that type of behavior probably should be
1246 # preserved for consistency
1247 except TypeError:
1248 ty,va,tb = sys.exc_info()
1249 raise TypeError, "not a valid non-string sequence or mapping object", tb
1250
Guido van Rossume7b146f2000-02-04 15:28:42 +00001251 l = []
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001252 if not doseq:
1253 # preserve old behavior
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001254 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001255 k = quote_plus(str(k))
1256 v = quote_plus(str(v))
1257 l.append(k + '=' + v)
1258 else:
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001259 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001260 k = quote_plus(str(k))
Walter Dörwald65230a22002-06-03 15:58:32 +00001261 if isinstance(v, str):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001262 v = quote_plus(v)
1263 l.append(k + '=' + v)
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001264 elif _is_unicode(v):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001265 # is there a reasonable way to convert to ASCII?
1266 # encode generates a string, but "replace" or "ignore"
1267 # lose information and "strict" can raise UnicodeError
1268 v = quote_plus(v.encode("ASCII","replace"))
1269 l.append(k + '=' + v)
1270 else:
1271 try:
1272 # is this a sufficient test for sequence-ness?
1273 x = len(v)
1274 except TypeError:
1275 # not a sequence
1276 v = quote_plus(str(v))
1277 l.append(k + '=' + v)
1278 else:
1279 # loop over the sequence
1280 for elt in v:
1281 l.append(k + '=' + quote_plus(str(elt)))
Guido van Rossumb2493f82000-12-15 15:01:37 +00001282 return '&'.join(l)
Guido van Rossum810a3391998-07-22 21:33:23 +00001283
Guido van Rossum442e7201996-03-20 15:33:11 +00001284# Proxy handling
Mark Hammond4f570b92000-07-26 07:04:38 +00001285def getproxies_environment():
1286 """Return a dictionary of scheme -> proxy server URL mappings.
1287
1288 Scan the environment for variables named <scheme>_proxy;
1289 this seems to be the standard convention. If you need a
1290 different way, you can pass a proxies dictionary to the
1291 [Fancy]URLopener constructor.
1292
1293 """
1294 proxies = {}
1295 for name, value in os.environ.items():
Guido van Rossumb2493f82000-12-15 15:01:37 +00001296 name = name.lower()
Georg Brandl22350112008-01-20 12:05:43 +00001297 if name == 'no_proxy':
1298 # handled in proxy_bypass_environment
1299 continue
Mark Hammond4f570b92000-07-26 07:04:38 +00001300 if value and name[-6:] == '_proxy':
1301 proxies[name[:-6]] = value
1302 return proxies
1303
Georg Brandl22350112008-01-20 12:05:43 +00001304def proxy_bypass_environment(host):
1305 """Test if proxies should not be used for a particular host.
1306
1307 Checks the environment for a variable named no_proxy, which should
1308 be a list of DNS suffixes separated by commas, or '*' for all hosts.
1309 """
1310 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
1311 # '*' is special case for always bypass
1312 if no_proxy == '*':
1313 return 1
1314 # strip port off host
1315 hostonly, port = splitport(host)
1316 # check if the host ends with any of the DNS suffixes
1317 for name in no_proxy.split(','):
1318 if name and (hostonly.endswith(name) or host.endswith(name)):
1319 return 1
1320 # otherwise, don't bypass
1321 return 0
1322
1323
Jack Jansen11d9b062004-07-16 11:45:00 +00001324if sys.platform == 'darwin':
Ronald Oussoren099646f2008-05-18 20:09:54 +00001325
1326 def _CFSetup(sc):
1327 from ctypes import c_int32, c_void_p, c_char_p, c_int
1328 sc.CFStringCreateWithCString.argtypes = [ c_void_p, c_char_p, c_int32 ]
1329 sc.CFStringCreateWithCString.restype = c_void_p
1330 sc.SCDynamicStoreCopyProxies.argtypes = [ c_void_p ]
1331 sc.SCDynamicStoreCopyProxies.restype = c_void_p
1332 sc.CFDictionaryGetValue.argtypes = [ c_void_p, c_void_p ]
1333 sc.CFDictionaryGetValue.restype = c_void_p
1334 sc.CFStringGetLength.argtypes = [ c_void_p ]
1335 sc.CFStringGetLength.restype = c_int32
1336 sc.CFStringGetCString.argtypes = [ c_void_p, c_char_p, c_int32, c_int32 ]
1337 sc.CFStringGetCString.restype = c_int32
1338 sc.CFNumberGetValue.argtypes = [ c_void_p, c_int, c_void_p ]
1339 sc.CFNumberGetValue.restype = c_int32
1340 sc.CFRelease.argtypes = [ c_void_p ]
1341 sc.CFRelease.restype = None
1342
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001343 def _CStringFromCFString(sc, value):
1344 from ctypes import create_string_buffer
1345 length = sc.CFStringGetLength(value) + 1
1346 buff = create_string_buffer(length)
1347 sc.CFStringGetCString(value, buff, length, 0)
1348 return buff.value
1349
1350 def _CFNumberToInt32(sc, cfnum):
1351 from ctypes import byref, c_int
1352 val = c_int()
1353 kCFNumberSInt32Type = 3
1354 sc.CFNumberGetValue(cfnum, kCFNumberSInt32Type, byref(val))
1355 return val.value
1356
1357
1358 def proxy_bypass_macosx_sysconf(host):
1359 """
1360 Return True iff this host shouldn't be accessed using a proxy
1361
1362 This function uses the MacOSX framework SystemConfiguration
1363 to fetch the proxy information.
1364 """
1365 from ctypes import cdll
1366 from ctypes.util import find_library
1367 import re
1368 import socket
1369 from fnmatch import fnmatch
1370
1371 def ip2num(ipAddr):
1372 parts = ipAddr.split('.')
1373 parts = map(int, parts)
1374 if len(parts) != 4:
1375 parts = (parts + [0, 0, 0, 0])[:4]
1376 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
1377
1378 sc = cdll.LoadLibrary(find_library("SystemConfiguration"))
Ronald Oussoren099646f2008-05-18 20:09:54 +00001379 _CFSetup(sc)
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001380
1381 hostIP = None
1382
1383 if not sc:
1384 return False
1385
1386 kSCPropNetProxiesExceptionsList = sc.CFStringCreateWithCString(0, "ExceptionsList", 0)
1387 kSCPropNetProxiesExcludeSimpleHostnames = sc.CFStringCreateWithCString(0,
1388 "ExcludeSimpleHostnames", 0)
1389
1390
1391 proxyDict = sc.SCDynamicStoreCopyProxies(None)
Ronald Oussoren099646f2008-05-18 20:09:54 +00001392 if proxyDict is None:
1393 return False
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001394
1395 try:
1396 # Check for simple host names:
1397 if '.' not in host:
1398 exclude_simple = sc.CFDictionaryGetValue(proxyDict,
1399 kSCPropNetProxiesExcludeSimpleHostnames)
1400 if exclude_simple and _CFNumberToInt32(sc, exclude_simple):
1401 return True
1402
1403
1404 # Check the exceptions list:
1405 exceptions = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesExceptionsList)
1406 if exceptions:
1407 # Items in the list are strings like these: *.local, 169.254/16
1408 for index in xrange(sc.CFArrayGetCount(exceptions)):
1409 value = sc.CFArrayGetValueAtIndex(exceptions, index)
1410 if not value: continue
1411 value = _CStringFromCFString(sc, value)
1412
1413 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
1414 if m is not None:
1415 if hostIP is None:
1416 hostIP = socket.gethostbyname(host)
1417 hostIP = ip2num(hostIP)
1418
1419 base = ip2num(m.group(1))
1420 mask = int(m.group(2)[1:])
1421 mask = 32 - mask
1422
1423 if (hostIP >> mask) == (base >> mask):
1424 return True
1425
1426 elif fnmatch(host, value):
1427 return True
1428
1429 return False
1430
1431 finally:
1432 sc.CFRelease(kSCPropNetProxiesExceptionsList)
1433 sc.CFRelease(kSCPropNetProxiesExcludeSimpleHostnames)
1434
1435
1436
1437 def getproxies_macosx_sysconf():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001438 """Return a dictionary of scheme -> proxy server URL mappings.
Guido van Rossum442e7201996-03-20 15:33:11 +00001439
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001440 This function uses the MacOSX framework SystemConfiguration
1441 to fetch the proxy information.
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001442 """
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001443 from ctypes import cdll
1444 from ctypes.util import find_library
1445
1446 sc = cdll.LoadLibrary(find_library("SystemConfiguration"))
Ronald Oussoren099646f2008-05-18 20:09:54 +00001447 _CFSetup(sc)
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001448
1449 if not sc:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001450 return {}
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001451
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001452 kSCPropNetProxiesHTTPEnable = sc.CFStringCreateWithCString(0, "HTTPEnable", 0)
1453 kSCPropNetProxiesHTTPProxy = sc.CFStringCreateWithCString(0, "HTTPProxy", 0)
1454 kSCPropNetProxiesHTTPPort = sc.CFStringCreateWithCString(0, "HTTPPort", 0)
1455
1456 kSCPropNetProxiesHTTPSEnable = sc.CFStringCreateWithCString(0, "HTTPSEnable", 0)
1457 kSCPropNetProxiesHTTPSProxy = sc.CFStringCreateWithCString(0, "HTTPSProxy", 0)
1458 kSCPropNetProxiesHTTPSPort = sc.CFStringCreateWithCString(0, "HTTPSPort", 0)
1459
1460 kSCPropNetProxiesFTPEnable = sc.CFStringCreateWithCString(0, "FTPEnable", 0)
1461 kSCPropNetProxiesFTPPassive = sc.CFStringCreateWithCString(0, "FTPPassive", 0)
1462 kSCPropNetProxiesFTPPort = sc.CFStringCreateWithCString(0, "FTPPort", 0)
1463 kSCPropNetProxiesFTPProxy = sc.CFStringCreateWithCString(0, "FTPProxy", 0)
1464
1465 kSCPropNetProxiesGopherEnable = sc.CFStringCreateWithCString(0, "GopherEnable", 0)
1466 kSCPropNetProxiesGopherPort = sc.CFStringCreateWithCString(0, "GopherPort", 0)
1467 kSCPropNetProxiesGopherProxy = sc.CFStringCreateWithCString(0, "GopherProxy", 0)
1468
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001469 proxies = {}
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001470 proxyDict = sc.SCDynamicStoreCopyProxies(None)
1471
1472 try:
1473 # HTTP:
1474 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPEnable)
1475 if enabled and _CFNumberToInt32(sc, enabled):
1476 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPProxy)
1477 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPPort)
1478
1479 if proxy:
1480 proxy = _CStringFromCFString(sc, proxy)
1481 if port:
1482 port = _CFNumberToInt32(sc, port)
1483 proxies["http"] = "http://%s:%i" % (proxy, port)
1484 else:
1485 proxies["http"] = "http://%s" % (proxy, )
1486
1487 # HTTPS:
1488 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSEnable)
1489 if enabled and _CFNumberToInt32(sc, enabled):
1490 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSProxy)
1491 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSPort)
1492
1493 if proxy:
1494 proxy = _CStringFromCFString(sc, proxy)
1495 if port:
1496 port = _CFNumberToInt32(sc, port)
1497 proxies["https"] = "http://%s:%i" % (proxy, port)
1498 else:
1499 proxies["https"] = "http://%s" % (proxy, )
1500
1501 # FTP:
1502 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPEnable)
1503 if enabled and _CFNumberToInt32(sc, enabled):
1504 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPProxy)
1505 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPPort)
1506
1507 if proxy:
1508 proxy = _CStringFromCFString(sc, proxy)
1509 if port:
1510 port = _CFNumberToInt32(sc, port)
1511 proxies["ftp"] = "http://%s:%i" % (proxy, port)
1512 else:
1513 proxies["ftp"] = "http://%s" % (proxy, )
1514
1515 # Gopher:
1516 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherEnable)
1517 if enabled and _CFNumberToInt32(sc, enabled):
1518 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherProxy)
1519 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherPort)
1520
1521 if proxy:
1522 proxy = _CStringFromCFString(sc, proxy)
1523 if port:
1524 port = _CFNumberToInt32(sc, port)
1525 proxies["gopher"] = "http://%s:%i" % (proxy, port)
1526 else:
1527 proxies["gopher"] = "http://%s" % (proxy, )
1528 finally:
1529 sc.CFRelease(proxyDict)
1530
1531 sc.CFRelease(kSCPropNetProxiesHTTPEnable)
1532 sc.CFRelease(kSCPropNetProxiesHTTPProxy)
1533 sc.CFRelease(kSCPropNetProxiesHTTPPort)
1534 sc.CFRelease(kSCPropNetProxiesFTPEnable)
1535 sc.CFRelease(kSCPropNetProxiesFTPPassive)
1536 sc.CFRelease(kSCPropNetProxiesFTPPort)
1537 sc.CFRelease(kSCPropNetProxiesFTPProxy)
1538 sc.CFRelease(kSCPropNetProxiesGopherEnable)
1539 sc.CFRelease(kSCPropNetProxiesGopherPort)
1540 sc.CFRelease(kSCPropNetProxiesGopherProxy)
1541
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001542 return proxies
Mark Hammond4f570b92000-07-26 07:04:38 +00001543
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001544
1545
Georg Brandl22350112008-01-20 12:05:43 +00001546 def proxy_bypass(host):
1547 if getproxies_environment():
1548 return proxy_bypass_environment(host)
1549 else:
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001550 return proxy_bypass_macosx_sysconf(host)
Tim Peters55c12d42001-08-09 18:04:14 +00001551
Jack Jansen11d9b062004-07-16 11:45:00 +00001552 def getproxies():
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001553 return getproxies_environment() or getproxies_macosx_sysconf()
Tim Peters182b5ac2004-07-18 06:16:08 +00001554
Mark Hammond4f570b92000-07-26 07:04:38 +00001555elif os.name == 'nt':
1556 def getproxies_registry():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001557 """Return a dictionary of scheme -> proxy server URL mappings.
Mark Hammond4f570b92000-07-26 07:04:38 +00001558
1559 Win32 uses the registry to store proxies.
1560
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001561 """
1562 proxies = {}
Mark Hammond4f570b92000-07-26 07:04:38 +00001563 try:
1564 import _winreg
1565 except ImportError:
1566 # Std module, so should be around - but you never know!
1567 return proxies
1568 try:
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001569 internetSettings = _winreg.OpenKey(_winreg.HKEY_CURRENT_USER,
1570 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Mark Hammond4f570b92000-07-26 07:04:38 +00001571 proxyEnable = _winreg.QueryValueEx(internetSettings,
1572 'ProxyEnable')[0]
1573 if proxyEnable:
1574 # Returned as Unicode but problems if not converted to ASCII
1575 proxyServer = str(_winreg.QueryValueEx(internetSettings,
1576 'ProxyServer')[0])
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001577 if '=' in proxyServer:
1578 # Per-protocol settings
Mark Hammond4f570b92000-07-26 07:04:38 +00001579 for p in proxyServer.split(';'):
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001580 protocol, address = p.split('=', 1)
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001581 # See if address has a type:// prefix
Guido van Rossum64e5aa92002-04-02 14:38:16 +00001582 import re
1583 if not re.match('^([^/:]+)://', address):
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001584 address = '%s://%s' % (protocol, address)
1585 proxies[protocol] = address
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001586 else:
1587 # Use one setting for all protocols
1588 if proxyServer[:5] == 'http:':
1589 proxies['http'] = proxyServer
1590 else:
1591 proxies['http'] = 'http://%s' % proxyServer
1592 proxies['ftp'] = 'ftp://%s' % proxyServer
Mark Hammond4f570b92000-07-26 07:04:38 +00001593 internetSettings.Close()
1594 except (WindowsError, ValueError, TypeError):
1595 # Either registry key not found etc, or the value in an
1596 # unexpected format.
1597 # proxies already set up to be empty so nothing to do
1598 pass
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001599 return proxies
Guido van Rossum442e7201996-03-20 15:33:11 +00001600
Mark Hammond4f570b92000-07-26 07:04:38 +00001601 def getproxies():
1602 """Return a dictionary of scheme -> proxy server URL mappings.
1603
1604 Returns settings gathered from the environment, if specified,
1605 or the registry.
1606
1607 """
1608 return getproxies_environment() or getproxies_registry()
Tim Peters55c12d42001-08-09 18:04:14 +00001609
Georg Brandl22350112008-01-20 12:05:43 +00001610 def proxy_bypass_registry(host):
Tim Peters55c12d42001-08-09 18:04:14 +00001611 try:
1612 import _winreg
1613 import re
Tim Peters55c12d42001-08-09 18:04:14 +00001614 except ImportError:
1615 # Std modules, so should be around - but you never know!
1616 return 0
1617 try:
1618 internetSettings = _winreg.OpenKey(_winreg.HKEY_CURRENT_USER,
1619 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
1620 proxyEnable = _winreg.QueryValueEx(internetSettings,
1621 'ProxyEnable')[0]
1622 proxyOverride = str(_winreg.QueryValueEx(internetSettings,
1623 'ProxyOverride')[0])
1624 # ^^^^ Returned as Unicode but problems if not converted to ASCII
1625 except WindowsError:
1626 return 0
1627 if not proxyEnable or not proxyOverride:
1628 return 0
1629 # try to make a host list from name and IP address.
Georg Brandl1f636702006-02-18 23:10:23 +00001630 rawHost, port = splitport(host)
1631 host = [rawHost]
Tim Peters55c12d42001-08-09 18:04:14 +00001632 try:
Georg Brandl1f636702006-02-18 23:10:23 +00001633 addr = socket.gethostbyname(rawHost)
1634 if addr != rawHost:
Tim Peters55c12d42001-08-09 18:04:14 +00001635 host.append(addr)
1636 except socket.error:
1637 pass
Georg Brandl1f636702006-02-18 23:10:23 +00001638 try:
1639 fqdn = socket.getfqdn(rawHost)
1640 if fqdn != rawHost:
1641 host.append(fqdn)
1642 except socket.error:
1643 pass
Tim Peters55c12d42001-08-09 18:04:14 +00001644 # make a check value list from the registry entry: replace the
1645 # '<local>' string by the localhost entry and the corresponding
1646 # canonical entry.
1647 proxyOverride = proxyOverride.split(';')
1648 i = 0
1649 while i < len(proxyOverride):
1650 if proxyOverride[i] == '<local>':
1651 proxyOverride[i:i+1] = ['localhost',
1652 '127.0.0.1',
1653 socket.gethostname(),
1654 socket.gethostbyname(
1655 socket.gethostname())]
1656 i += 1
1657 # print proxyOverride
1658 # now check if we match one of the registry values.
1659 for test in proxyOverride:
Tim Petersab9ba272001-08-09 21:40:30 +00001660 test = test.replace(".", r"\.") # mask dots
1661 test = test.replace("*", r".*") # change glob sequence
1662 test = test.replace("?", r".") # change glob char
Tim Peters55c12d42001-08-09 18:04:14 +00001663 for val in host:
1664 # print "%s <--> %s" %( test, val )
1665 if re.match(test, val, re.I):
1666 return 1
1667 return 0
1668
Georg Brandl22350112008-01-20 12:05:43 +00001669 def proxy_bypass(host):
1670 """Return a dictionary of scheme -> proxy server URL mappings.
1671
1672 Returns settings gathered from the environment, if specified,
1673 or the registry.
1674
1675 """
1676 if getproxies_environment():
1677 return proxy_bypass_environment(host)
1678 else:
1679 return proxy_bypass_registry(host)
1680
Mark Hammond4f570b92000-07-26 07:04:38 +00001681else:
1682 # By default use environment variables
1683 getproxies = getproxies_environment
Georg Brandl22350112008-01-20 12:05:43 +00001684 proxy_bypass = proxy_bypass_environment
Guido van Rossum442e7201996-03-20 15:33:11 +00001685
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001686# Test and time quote() and unquote()
1687def test1():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001688 s = ''
1689 for i in range(256): s = s + chr(i)
1690 s = s*4
1691 t0 = time.time()
1692 qs = quote(s)
1693 uqs = unquote(qs)
1694 t1 = time.time()
1695 if uqs != s:
1696 print 'Wrong!'
Walter Dörwald70a6b492004-02-12 17:35:32 +00001697 print repr(s)
1698 print repr(qs)
1699 print repr(uqs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001700 print round(t1 - t0, 3), 'sec'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001701
1702
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001703def reporthook(blocknum, blocksize, totalsize):
1704 # Report during remote transfers
Guido van Rossumb2493f82000-12-15 15:01:37 +00001705 print "Block number: %d, Block size: %d, Total size: %d" % (
1706 blocknum, blocksize, totalsize)
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001707
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001708# Test program
Guido van Rossum23490151998-06-25 02:39:00 +00001709def test(args=[]):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001710 if not args:
1711 args = [
1712 '/etc/passwd',
1713 'file:/etc/passwd',
1714 'file://localhost/etc/passwd',
Collin Winter071d1ae2007-03-12 01:55:54 +00001715 'ftp://ftp.gnu.org/pub/README',
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001716 'http://www.python.org/index.html',
1717 ]
Guido van Rossum09c8b6c1999-12-07 21:37:17 +00001718 if hasattr(URLopener, "open_https"):
1719 args.append('https://synergy.as.cmu.edu/~geek/')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001720 try:
1721 for url in args:
1722 print '-'*10, url, '-'*10
1723 fn, h = urlretrieve(url, None, reporthook)
Guido van Rossumb2493f82000-12-15 15:01:37 +00001724 print fn
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001725 if h:
1726 print '======'
1727 for k in h.keys(): print k + ':', h[k]
1728 print '======'
1729 fp = open(fn, 'rb')
1730 data = fp.read()
1731 del fp
1732 if '\r' in data:
1733 table = string.maketrans("", "")
Guido van Rossumb2493f82000-12-15 15:01:37 +00001734 data = data.translate(table, "\r")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001735 print data
1736 fn, h = None, None
1737 print '-'*40
1738 finally:
1739 urlcleanup()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001740
Guido van Rossum23490151998-06-25 02:39:00 +00001741def main():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001742 import getopt, sys
1743 try:
1744 opts, args = getopt.getopt(sys.argv[1:], "th")
1745 except getopt.error, msg:
1746 print msg
1747 print "Use -h for help"
1748 return
1749 t = 0
1750 for o, a in opts:
1751 if o == '-t':
1752 t = t + 1
1753 if o == '-h':
1754 print "Usage: python urllib.py [-t] [url ...]"
1755 print "-t runs self-test;",
1756 print "otherwise, contents of urls are printed"
1757 return
1758 if t:
1759 if t > 1:
1760 test1()
1761 test(args)
1762 else:
1763 if not args:
1764 print "Use -h for help"
1765 for url in args:
1766 print urlopen(url).read(),
Guido van Rossum23490151998-06-25 02:39:00 +00001767
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001768# Run test program when run as a script
1769if __name__ == '__main__':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001770 main()