blob: fb936049025d632b11be0859dc9a955083bbb923 [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
Georg Brandl24420152008-05-26 16:32:26 +000025import http.client
Jack Jansendc3e3f61995-12-15 13:22:13 +000026import os
Jeremy Hylton5f22af12007-08-16 17:55:18 +000027import socket
Guido van Rossum3c8484e1996-11-20 22:02:24 +000028import sys
Jeremy Hylton5f22af12007-08-16 17:55:18 +000029import time
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",
Guido van Rossumd59da4b2007-05-22 18:11:13 +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
Jack Jansendc3e3f61995-12-15 13:22:13 +000049else:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000050 def url2pathname(pathname):
Georg Brandlc0b24732005-12-26 22:53:56 +000051 """OS-specific conversion from a relative URL of the 'file' scheme
52 to a file system path; not recommended for general use."""
Guido van Rossum367ac801999-03-12 14:31:10 +000053 return unquote(pathname)
Georg Brandlc0b24732005-12-26 22:53:56 +000054
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000055 def pathname2url(pathname):
Georg Brandlc0b24732005-12-26 22:53:56 +000056 """OS-specific conversion from a file system path to a relative URL
57 of the 'file' scheme; not recommended for general use."""
Guido van Rossum367ac801999-03-12 14:31:10 +000058 return quote(pathname)
Guido van Rossum33add0a1998-12-18 15:25:22 +000059
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000060# This really consists of two pieces:
61# (1) a class which handles opening of all sorts of URLs
62# (plus assorted utilities etc.)
63# (2) a set of functions for parsing URLs
64# XXX Should these be separated out into different modules?
65
66
67# Shortcut for basic usage
68_urlopener = None
Fred Drakedf6eca72002-04-04 20:41:34 +000069def urlopen(url, data=None, proxies=None):
Skip Montanaro79f1c172000-08-22 03:00:52 +000070 """urlopen(url [, data]) -> open file-like object"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000071 global _urlopener
Fred Drakedf6eca72002-04-04 20:41:34 +000072 if proxies is not None:
73 opener = FancyURLopener(proxies=proxies)
74 elif not _urlopener:
75 opener = FancyURLopener()
76 _urlopener = opener
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000077 else:
Fred Drakedf6eca72002-04-04 20:41:34 +000078 opener = _urlopener
79 if data is None:
80 return opener.open(url)
81 else:
82 return opener.open(url, data)
Jeremy Hylton39b198d2007-08-04 19:22:00 +000083
Fred Drake316a7932000-08-24 01:01:26 +000084def urlretrieve(url, filename=None, reporthook=None, data=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000085 global _urlopener
86 if not _urlopener:
87 _urlopener = FancyURLopener()
Fred Drake316a7932000-08-24 01:01:26 +000088 return _urlopener.retrieve(url, filename, reporthook, data)
Jeremy Hylton39b198d2007-08-04 19:22:00 +000089
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
Thomas Wouters47b49bf2007-08-30 22:15:33 +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()
Guido van Rossume2b70bc2006-08-18 22:13:04 +0000125 assert hasattr(proxies, 'keys'), "proxies must be a mapping"
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000126 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')
Thomas Wouters0e3f5912006-08-11 14:57:12 +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)
Guido van Rossumb940e112007-01-10 16:19:56 +0000201 except socket.error as msg:
Collin Winter828f04a2007-08-31 00:04:24 +0000202 raise IOError('socket error', msg).with_traceback(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)
Collin Winterce36ad82007-08-30 01:19:48 +0000207 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)
Collin Winterce36ad82007-08-30 01:19:48 +0000212 raise IOError('url error', 'invalid proxy for %s' % type, proxy)
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000213
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
Guido van Rossumb940e112007-01-10 16:19:56 +0000228 except IOError as msg:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000229 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)
Guido van Rossuma0982942007-07-10 08:30:03 +0000257 if not block:
Georg Brandl5a650a22005-08-26 08:51:34 +0000258 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 Hylton5f22af12007-08-16 17:55:18 +0000278 def _open_generic_http(self, connection_factory, url, data):
279 """Make an HTTP connection using connection_class.
280
281 This is an internal method that should be called from
282 open_http() or open_https().
283
284 Arguments:
285 - connection_factory should take a host name and return an
286 HTTPConnection instance.
287 - url is the url to retrieval or a host, relative-path pair.
288 - data is payload for a POST request or None.
289 """
290
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000291 user_passwd = None
Martin v. Löwis3e865952006-01-24 15:51:21 +0000292 proxy_passwd= None
Walter Dörwald65230a22002-06-03 15:58:32 +0000293 if isinstance(url, str):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000294 host, selector = splithost(url)
295 if host:
296 user_passwd, host = splituser(host)
297 host = unquote(host)
298 realhost = host
299 else:
300 host, selector = url
Martin v. Löwis3e865952006-01-24 15:51:21 +0000301 # check whether the proxy contains authorization information
302 proxy_passwd, host = splituser(host)
303 # now we proceed with the url we want to obtain
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000304 urltype, rest = splittype(selector)
305 url = rest
306 user_passwd = None
Guido van Rossumb2493f82000-12-15 15:01:37 +0000307 if urltype.lower() != 'http':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000308 realhost = None
309 else:
310 realhost, rest = splithost(rest)
311 if realhost:
312 user_passwd, realhost = splituser(realhost)
313 if user_passwd:
314 selector = "%s://%s%s" % (urltype, realhost, rest)
Tim Peters55c12d42001-08-09 18:04:14 +0000315 if proxy_bypass(realhost):
316 host = realhost
317
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000318 #print "proxy via http:", host, selector
Collin Winterce36ad82007-08-30 01:19:48 +0000319 if not host: raise IOError('http error', 'no host given')
Tim Peters92037a12006-01-24 22:44:08 +0000320
Martin v. Löwis3e865952006-01-24 15:51:21 +0000321 if proxy_passwd:
322 import base64
Thomas Wouters89f507f2006-12-13 04:49:30 +0000323 proxy_auth = base64.b64encode(proxy_passwd).strip()
Martin v. Löwis3e865952006-01-24 15:51:21 +0000324 else:
325 proxy_auth = None
326
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000327 if user_passwd:
328 import base64
Thomas Wouters89f507f2006-12-13 04:49:30 +0000329 auth = base64.b64encode(user_passwd).strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000330 else:
331 auth = None
Jeremy Hylton5f22af12007-08-16 17:55:18 +0000332 http_conn = connection_factory(host)
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000333 # XXX We should fix urllib so that it works with HTTP/1.1.
334 http_conn._http_vsn = 10
335 http_conn._http_vsn_str = "HTTP/1.0"
336
337 headers = {}
338 if proxy_auth:
339 headers["Proxy-Authorization"] = "Basic %s" % proxy_auth
340 if auth:
341 headers["Authorization"] = "Basic %s" % auth
342 if realhost:
343 headers["Host"] = realhost
344 for header, value in self.addheaders:
345 headers[header] = value
346
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000347 if data is not None:
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000348 headers["Content-Type"] = "application/x-www-form-urlencoded"
349 http_conn.request("POST", selector, data, headers)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000350 else:
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000351 http_conn.request("GET", selector, headers=headers)
352
353 try:
354 response = http_conn.getresponse()
Georg Brandl24420152008-05-26 16:32:26 +0000355 except http.client.BadStatusLine:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000356 # something went wrong with the HTTP status line
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000357 raise IOError('http protocol error', 0,
358 'got a bad status line', None)
359
Thomas Woutersce272b62007-09-19 21:19:28 +0000360 # According to RFC 2616, "2xx" code indicates that the client's
361 # request was successfully received, understood, and accepted.
Christian Heimes57dddfb2008-01-02 18:30:52 +0000362 if (200 <= response.status < 300):
Georg Brandle3777b42008-01-26 11:23:13 +0000363 return addinfourl(response.fp, response.msg, "http:" + url,
364 response.status)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000365 else:
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000366 return self.http_error(
367 url, response.fp,
368 response.status, response.reason, response.msg, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000369
Jeremy Hylton5f22af12007-08-16 17:55:18 +0000370 def open_http(self, url, data=None):
371 """Use HTTP protocol."""
Georg Brandl24420152008-05-26 16:32:26 +0000372 return self._open_generic_http(http.client.HTTPConnection, url, data)
Jeremy Hylton5f22af12007-08-16 17:55:18 +0000373
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000374 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000375 """Handle http errors.
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000376
Guido van Rossume7b146f2000-02-04 15:28:42 +0000377 Derived class can override this, or provide specific handlers
378 named http_error_DDD where DDD is the 3-digit error code."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000379 # First check if there's a specific handler for this error
380 name = 'http_error_%d' % errcode
381 if hasattr(self, name):
382 method = getattr(self, name)
383 if data is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000384 result = method(url, fp, errcode, errmsg, headers)
Jeremy Hyltonb30f52a1999-02-25 16:14:58 +0000385 else:
386 result = method(url, fp, errcode, errmsg, headers, data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000387 if result: return result
Jeremy Hyltonb30f52a1999-02-25 16:14:58 +0000388 return self.http_error_default(url, fp, errcode, errmsg, headers)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000389
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000390 def http_error_default(self, url, fp, errcode, errmsg, headers):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000391 """Default error handler: close the connection and raise IOError."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000392 void = fp.read()
393 fp.close()
Collin Winterce36ad82007-08-30 01:19:48 +0000394 raise IOError('http error', errcode, errmsg, headers)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000395
Thomas Wouters47b49bf2007-08-30 22:15:33 +0000396 if _have_ssl:
Jeremy Hylton5f22af12007-08-16 17:55:18 +0000397 def _https_connection(self, host):
Georg Brandl24420152008-05-26 16:32:26 +0000398 return http.client.HTTPSConnection(host,
399 key_file=self.key_file,
400 cert_file=self.cert_file)
Jeremy Hylton5f22af12007-08-16 17:55:18 +0000401
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000402 def open_https(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000403 """Use HTTPS protocol."""
Jeremy Hylton5f22af12007-08-16 17:55:18 +0000404 return self._open_generic_http(self._https_connection, url, data)
Fred Drake567ca8e2000-08-21 21:42:42 +0000405
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000406 def open_file(self, url):
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000407 """Use local file or FTP depending on form of URL."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000408 if not isinstance(url, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000409 raise IOError('file error', 'proxy support for file protocol currently not implemented')
Jack Jansen4ef11032002-09-12 20:14:04 +0000410 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000411 return self.open_ftp(url)
412 else:
413 return self.open_local_file(url)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000414
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000415 def open_local_file(self, url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000416 """Use local file."""
Neal Norwitz315d8452007-08-30 03:06:59 +0000417 import mimetypes, mimetools, email.utils
Guido van Rossum68937b42007-05-18 00:51:22 +0000418 from io import StringIO
Guido van Rossumf0713d32001-08-09 17:43:35 +0000419 host, file = splithost(url)
420 localname = url2pathname(file)
Guido van Rossuma2da3052002-04-15 00:25:01 +0000421 try:
422 stats = os.stat(localname)
Guido van Rossumb940e112007-01-10 16:19:56 +0000423 except OSError as e:
Guido van Rossuma2da3052002-04-15 00:25:01 +0000424 raise IOError(e.errno, e.strerror, e.filename)
Walter Dörwald92b48b72002-03-22 17:30:38 +0000425 size = stats.st_size
Neal Norwitz315d8452007-08-30 03:06:59 +0000426 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000427 mtype = mimetypes.guess_type(url)[0]
Raymond Hettingera6172712004-12-31 19:15:26 +0000428 headers = mimetools.Message(StringIO(
Guido van Rossumf0713d32001-08-09 17:43:35 +0000429 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
430 (mtype or 'text/plain', size, modified)))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000431 if not host:
Guido van Rossum336a2011999-06-24 15:27:36 +0000432 urlfile = file
433 if file[:1] == '/':
434 urlfile = 'file://' + file
Guido van Rossumf0713d32001-08-09 17:43:35 +0000435 return addinfourl(open(localname, 'rb'),
Guido van Rossum336a2011999-06-24 15:27:36 +0000436 headers, urlfile)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000437 host, port = splitport(host)
438 if not port \
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000439 and socket.gethostbyname(host) in (localhost(), thishost()):
Guido van Rossum336a2011999-06-24 15:27:36 +0000440 urlfile = file
441 if file[:1] == '/':
442 urlfile = 'file://' + file
Guido van Rossumf0713d32001-08-09 17:43:35 +0000443 return addinfourl(open(localname, 'rb'),
Guido van Rossum336a2011999-06-24 15:27:36 +0000444 headers, urlfile)
Collin Winterce36ad82007-08-30 01:19:48 +0000445 raise IOError('local file error', 'not on local host')
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000446
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000447 def open_ftp(self, url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000448 """Use FTP protocol."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000449 if not isinstance(url, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000450 raise IOError('ftp error', 'proxy support for ftp protocol currently not implemented')
Raymond Hettingera6172712004-12-31 19:15:26 +0000451 import mimetypes, mimetools
Guido van Rossum68937b42007-05-18 00:51:22 +0000452 from io import StringIO
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000453 host, path = splithost(url)
Collin Winterce36ad82007-08-30 01:19:48 +0000454 if not host: raise IOError('ftp error', 'no host given')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000455 host, port = splitport(host)
456 user, host = splituser(host)
457 if user: user, passwd = splitpasswd(user)
458 else: passwd = None
459 host = unquote(host)
460 user = unquote(user or '')
461 passwd = unquote(passwd or '')
462 host = socket.gethostbyname(host)
463 if not port:
464 import ftplib
465 port = ftplib.FTP_PORT
466 else:
467 port = int(port)
468 path, attrs = splitattr(path)
469 path = unquote(path)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000470 dirs = path.split('/')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000471 dirs, file = dirs[:-1], dirs[-1]
472 if dirs and not dirs[0]: dirs = dirs[1:]
Guido van Rossum5e006a31999-08-18 17:40:33 +0000473 if dirs and not dirs[0]: dirs[0] = '/'
Guido van Rossumb2493f82000-12-15 15:01:37 +0000474 key = user, host, port, '/'.join(dirs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000475 # XXX thread unsafe!
476 if len(self.ftpcache) > MAXFTPCACHE:
477 # Prune the cache, rather arbitrarily
478 for k in self.ftpcache.keys():
479 if k != key:
480 v = self.ftpcache[k]
481 del self.ftpcache[k]
482 v.close()
483 try:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000484 if not key in self.ftpcache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000485 self.ftpcache[key] = \
486 ftpwrapper(user, passwd, host, port, dirs)
487 if not file: type = 'D'
488 else: type = 'I'
489 for attr in attrs:
490 attr, value = splitvalue(attr)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000491 if attr.lower() == 'type' and \
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000492 value in ('a', 'A', 'i', 'I', 'd', 'D'):
Guido van Rossumb2493f82000-12-15 15:01:37 +0000493 type = value.upper()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000494 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
Guido van Rossum88e0b5b2001-08-23 13:38:15 +0000495 mtype = mimetypes.guess_type("ftp:" + url)[0]
496 headers = ""
497 if mtype:
498 headers += "Content-Type: %s\n" % mtype
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000499 if retrlen is not None and retrlen >= 0:
Guido van Rossum88e0b5b2001-08-23 13:38:15 +0000500 headers += "Content-Length: %d\n" % retrlen
Raymond Hettingera6172712004-12-31 19:15:26 +0000501 headers = mimetools.Message(StringIO(headers))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000502 return addinfourl(fp, headers, "ftp:" + url)
Guido van Rossumb940e112007-01-10 16:19:56 +0000503 except ftperrors() as msg:
Collin Winter828f04a2007-08-31 00:04:24 +0000504 raise IOError('ftp error', msg).with_traceback(sys.exc_info()[2])
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000505
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000506 def open_data(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000507 """Use "data" URL."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000508 if not isinstance(url, str):
Collin Winterce36ad82007-08-30 01:19:48 +0000509 raise IOError('data error', 'proxy support for data protocol currently not implemented')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000510 # ignore POSTed data
511 #
512 # syntax of data URLs:
513 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
514 # mediatype := [ type "/" subtype ] *( ";" parameter )
515 # data := *urlchar
516 # parameter := attribute "=" value
Raymond Hettingera6172712004-12-31 19:15:26 +0000517 import mimetools
Guido van Rossum68937b42007-05-18 00:51:22 +0000518 from io import StringIO
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000519 try:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000520 [type, data] = url.split(',', 1)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000521 except ValueError:
Collin Winterce36ad82007-08-30 01:19:48 +0000522 raise IOError('data error', 'bad data URL')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000523 if not type:
524 type = 'text/plain;charset=US-ASCII'
Guido van Rossumb2493f82000-12-15 15:01:37 +0000525 semi = type.rfind(';')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000526 if semi >= 0 and '=' not in type[semi:]:
527 encoding = type[semi+1:]
528 type = type[:semi]
529 else:
530 encoding = ''
531 msg = []
532 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %T GMT',
533 time.gmtime(time.time())))
534 msg.append('Content-type: %s' % type)
535 if encoding == 'base64':
536 import base64
537 data = base64.decodestring(data)
538 else:
539 data = unquote(data)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000540 msg.append('Content-Length: %d' % len(data))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000541 msg.append('')
542 msg.append(data)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000543 msg = '\n'.join(msg)
Raymond Hettingera6172712004-12-31 19:15:26 +0000544 f = StringIO(msg)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000545 headers = mimetools.Message(f, 0)
Georg Brandl1f663572005-11-26 16:50:44 +0000546 #f.fileno = None # needed for addinfourl
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000547 return addinfourl(f, headers, url)
Guido van Rossum6d4d1c21998-03-12 14:32:55 +0000548
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000549
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000550class FancyURLopener(URLopener):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000551 """Derived class with handlers for errors we can handle (perhaps)."""
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000552
Neal Norwitz60e04cd2002-06-11 13:38:51 +0000553 def __init__(self, *args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +0000554 URLopener.__init__(self, *args, **kwargs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000555 self.auth_cache = {}
Skip Montanaroc3e11d62001-02-15 16:56:36 +0000556 self.tries = 0
557 self.maxtries = 10
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000558
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000559 def http_error_default(self, url, fp, errcode, errmsg, headers):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000560 """Default error handling -- don't raise an exception."""
Christian Heimes9bd667a2008-01-20 15:14:11 +0000561 return addinfourl(fp, headers, "http:" + url, errcode)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000562
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000563 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000564 """Error 302 -- relocated (temporarily)."""
Skip Montanaroc3e11d62001-02-15 16:56:36 +0000565 self.tries += 1
566 if self.maxtries and self.tries >= self.maxtries:
567 if hasattr(self, "http_error_500"):
568 meth = self.http_error_500
569 else:
570 meth = self.http_error_default
571 self.tries = 0
572 return meth(url, fp, 500,
573 "Internal Server Error: Redirect Recursion", headers)
574 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
575 data)
576 self.tries = 0
577 return result
578
579 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
Raymond Hettinger54f02222002-06-01 14:18:47 +0000580 if 'location' in headers:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000581 newurl = headers['location']
Raymond Hettinger54f02222002-06-01 14:18:47 +0000582 elif 'uri' in headers:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000583 newurl = headers['uri']
584 else:
585 return
586 void = fp.read()
587 fp.close()
Guido van Rossum3527f591999-03-29 20:23:41 +0000588 # In case the server sent a relative URL, join with original:
Moshe Zadka5d87d472001-04-09 14:54:21 +0000589 newurl = basejoin(self.type + ":" + url, newurl)
Guido van Rossumfa19f7c2003-05-16 01:46:51 +0000590 return self.open(newurl)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000591
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000592 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000593 """Error 301 -- also relocated (permanently)."""
594 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
Guido van Rossume6ad8911996-09-10 17:02:56 +0000595
Raymond Hettinger024aaa12003-04-24 15:32:12 +0000596 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
597 """Error 303 -- also relocated (essentially identical to 302)."""
598 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
599
Guido van Rossumfa19f7c2003-05-16 01:46:51 +0000600 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
601 """Error 307 -- relocated, but turn POST into error."""
602 if data is None:
603 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
604 else:
605 return self.http_error_default(url, fp, errcode, errmsg, headers)
606
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000607 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000608 """Error 401 -- authentication required.
Martin v. Löwis3e865952006-01-24 15:51:21 +0000609 This function supports Basic authentication only."""
Raymond Hettinger54f02222002-06-01 14:18:47 +0000610 if not 'www-authenticate' in headers:
Tim Peters85ba6732001-02-28 08:26:44 +0000611 URLopener.http_error_default(self, url, fp,
Fred Drakec680ae82001-10-13 18:37:07 +0000612 errcode, errmsg, headers)
Moshe Zadkae99bd172001-02-27 06:27:04 +0000613 stuff = headers['www-authenticate']
614 import re
615 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
616 if not match:
Tim Peters85ba6732001-02-28 08:26:44 +0000617 URLopener.http_error_default(self, url, fp,
Moshe Zadkae99bd172001-02-27 06:27:04 +0000618 errcode, errmsg, headers)
619 scheme, realm = match.groups()
620 if scheme.lower() != 'basic':
Tim Peters85ba6732001-02-28 08:26:44 +0000621 URLopener.http_error_default(self, url, fp,
Moshe Zadkae99bd172001-02-27 06:27:04 +0000622 errcode, errmsg, headers)
623 name = 'retry_' + self.type + '_basic_auth'
624 if data is None:
625 return getattr(self,name)(url, realm)
626 else:
627 return getattr(self,name)(url, realm, data)
Tim Peters92037a12006-01-24 22:44:08 +0000628
Martin v. Löwis3e865952006-01-24 15:51:21 +0000629 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None):
630 """Error 407 -- proxy authentication required.
631 This function supports Basic authentication only."""
632 if not 'proxy-authenticate' in headers:
633 URLopener.http_error_default(self, url, fp,
634 errcode, errmsg, headers)
635 stuff = headers['proxy-authenticate']
636 import re
637 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
638 if not match:
639 URLopener.http_error_default(self, url, fp,
640 errcode, errmsg, headers)
641 scheme, realm = match.groups()
642 if scheme.lower() != 'basic':
643 URLopener.http_error_default(self, url, fp,
644 errcode, errmsg, headers)
645 name = 'retry_proxy_' + self.type + '_basic_auth'
646 if data is None:
647 return getattr(self,name)(url, realm)
648 else:
649 return getattr(self,name)(url, realm, data)
Tim Peters92037a12006-01-24 22:44:08 +0000650
Martin v. Löwis3e865952006-01-24 15:51:21 +0000651 def retry_proxy_http_basic_auth(self, url, realm, data=None):
652 host, selector = splithost(url)
653 newurl = 'http://' + host + selector
654 proxy = self.proxies['http']
655 urltype, proxyhost = splittype(proxy)
656 proxyhost, proxyselector = splithost(proxyhost)
657 i = proxyhost.find('@') + 1
658 proxyhost = proxyhost[i:]
659 user, passwd = self.get_user_passwd(proxyhost, realm, i)
660 if not (user or passwd): return None
661 proxyhost = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + proxyhost
662 self.proxies['http'] = 'http://' + proxyhost + proxyselector
663 if data is None:
664 return self.open(newurl)
665 else:
666 return self.open(newurl, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000667
Martin v. Löwis3e865952006-01-24 15:51:21 +0000668 def retry_proxy_https_basic_auth(self, url, realm, data=None):
669 host, selector = splithost(url)
670 newurl = 'https://' + host + selector
671 proxy = self.proxies['https']
672 urltype, proxyhost = splittype(proxy)
673 proxyhost, proxyselector = splithost(proxyhost)
674 i = proxyhost.find('@') + 1
675 proxyhost = proxyhost[i:]
676 user, passwd = self.get_user_passwd(proxyhost, realm, i)
677 if not (user or passwd): return None
678 proxyhost = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + proxyhost
679 self.proxies['https'] = 'https://' + proxyhost + proxyselector
680 if data is None:
681 return self.open(newurl)
682 else:
683 return self.open(newurl, data)
Tim Peters92037a12006-01-24 22:44:08 +0000684
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000685 def retry_http_basic_auth(self, url, realm, data=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000686 host, selector = splithost(url)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000687 i = host.find('@') + 1
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000688 host = host[i:]
689 user, passwd = self.get_user_passwd(host, realm, i)
690 if not (user or passwd): return None
Guido van Rossumafc4f042001-01-15 18:31:13 +0000691 host = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + host
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000692 newurl = 'http://' + host + selector
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000693 if data is None:
694 return self.open(newurl)
695 else:
696 return self.open(newurl, data)
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000697
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000698 def retry_https_basic_auth(self, url, realm, data=None):
Tim Peterse1190062001-01-15 03:34:38 +0000699 host, selector = splithost(url)
700 i = host.find('@') + 1
701 host = host[i:]
702 user, passwd = self.get_user_passwd(host, realm, i)
703 if not (user or passwd): return None
Guido van Rossumafc4f042001-01-15 18:31:13 +0000704 host = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + host
Martin v. Löwis3e865952006-01-24 15:51:21 +0000705 newurl = 'https://' + host + selector
706 if data is None:
707 return self.open(newurl)
708 else:
709 return self.open(newurl, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000710
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000711 def get_user_passwd(self, host, realm, clear_cache = 0):
Guido van Rossumb2493f82000-12-15 15:01:37 +0000712 key = realm + '@' + host.lower()
Raymond Hettinger54f02222002-06-01 14:18:47 +0000713 if key in self.auth_cache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000714 if clear_cache:
715 del self.auth_cache[key]
716 else:
717 return self.auth_cache[key]
718 user, passwd = self.prompt_user_passwd(host, realm)
719 if user or passwd: self.auth_cache[key] = (user, passwd)
720 return user, passwd
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000721
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000722 def prompt_user_passwd(self, host, realm):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000723 """Override this in a GUI environment!"""
Guido van Rossum7cba8502007-03-19 22:23:59 +0000724 import getpass
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000725 try:
Guido van Rossum7cba8502007-03-19 22:23:59 +0000726 user = input("Enter username for %s at %s: " % (realm, host))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000727 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
728 (user, realm, host))
729 return user, passwd
730 except KeyboardInterrupt:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000731 print()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000732 return None, None
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000733
734
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000735# Utility functions
736
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000737_localhost = None
738def localhost():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000739 """Return the IP address of the magic hostname 'localhost'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000740 global _localhost
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000741 if _localhost is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000742 _localhost = socket.gethostbyname('localhost')
743 return _localhost
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000744
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000745_thishost = None
746def thishost():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000747 """Return the IP address of the current host."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000748 global _thishost
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000749 if _thishost is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000750 _thishost = socket.gethostbyname(socket.gethostname())
751 return _thishost
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000752
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000753_ftperrors = None
754def ftperrors():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000755 """Return the set of errors raised by the FTP class."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000756 global _ftperrors
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000757 if _ftperrors is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000758 import ftplib
759 _ftperrors = ftplib.all_errors
760 return _ftperrors
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000761
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000762_noheaders = None
763def noheaders():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000764 """Return an empty mimetools.Message object."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000765 global _noheaders
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000766 if _noheaders is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000767 import mimetools
Guido van Rossum68937b42007-05-18 00:51:22 +0000768 from io import StringIO
Raymond Hettingera6172712004-12-31 19:15:26 +0000769 _noheaders = mimetools.Message(StringIO(), 0)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000770 _noheaders.fp.close() # Recycle file descriptor
771 return _noheaders
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000772
773
774# Utility classes
775
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000776class ftpwrapper:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000777 """Class used by open_ftp() for cache of open FTP connections."""
778
Georg Brandlf78e02b2008-06-10 17:40:04 +0000779 def __init__(self, user, passwd, host, port, dirs,
780 timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000781 self.user = user
782 self.passwd = passwd
783 self.host = host
784 self.port = port
785 self.dirs = dirs
Guido van Rossume7ba4952007-06-06 23:52:48 +0000786 self.timeout = timeout
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000787 self.init()
Guido van Rossume7b146f2000-02-04 15:28:42 +0000788
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000789 def init(self):
790 import ftplib
791 self.busy = 0
792 self.ftp = ftplib.FTP()
Guido van Rossume7ba4952007-06-06 23:52:48 +0000793 self.ftp.connect(self.host, self.port, self.timeout)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000794 self.ftp.login(self.user, self.passwd)
795 for dir in self.dirs:
796 self.ftp.cwd(dir)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000797
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000798 def retrfile(self, file, type):
799 import ftplib
800 self.endtransfer()
801 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
802 else: cmd = 'TYPE ' + type; isdir = 0
803 try:
804 self.ftp.voidcmd(cmd)
805 except ftplib.all_errors:
806 self.init()
807 self.ftp.voidcmd(cmd)
808 conn = None
809 if file and not isdir:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000810 # Try to retrieve as a file
811 try:
812 cmd = 'RETR ' + file
813 conn = self.ftp.ntransfercmd(cmd)
Guido van Rossumb940e112007-01-10 16:19:56 +0000814 except ftplib.error_perm as reason:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000815 if str(reason)[:3] != '550':
Collin Winter828f04a2007-08-31 00:04:24 +0000816 raise IOError('ftp error', reason).with_traceback(sys.exc_info()[2])
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000817 if not conn:
818 # Set transfer mode to ASCII!
819 self.ftp.voidcmd('TYPE A')
Christian Heimes9bd667a2008-01-20 15:14:11 +0000820 # Try a directory listing. Verify that directory exists.
821 if file:
822 pwd = self.ftp.pwd()
823 try:
824 try:
825 self.ftp.cwd(file)
826 except ftplib.error_perm as reason:
827 raise IOError('ftp error', reason) from reason
828 finally:
829 self.ftp.cwd(pwd)
830 cmd = 'LIST ' + file
831 else:
832 cmd = 'LIST'
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000833 conn = self.ftp.ntransfercmd(cmd)
834 self.busy = 1
835 # Pass back both a suitably decorated object and a retrieval length
836 return (addclosehook(conn[0].makefile('rb'),
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000837 self.endtransfer), conn[1])
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000838 def endtransfer(self):
839 if not self.busy:
840 return
841 self.busy = 0
842 try:
843 self.ftp.voidresp()
844 except ftperrors():
845 pass
Guido van Rossume7b146f2000-02-04 15:28:42 +0000846
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000847 def close(self):
848 self.endtransfer()
849 try:
850 self.ftp.close()
851 except ftperrors():
852 pass
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000853
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000854class addbase:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000855 """Base class for addinfo and addclosehook."""
856
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000857 # XXX Add a method to expose the timeout on the underlying socket?
858
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000859 def __init__(self, fp):
860 self.fp = fp
861 self.read = self.fp.read
862 self.readline = self.fp.readline
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000863 if hasattr(self.fp, "readlines"): self.readlines = self.fp.readlines
Georg Brandl1f663572005-11-26 16:50:44 +0000864 if hasattr(self.fp, "fileno"):
865 self.fileno = self.fp.fileno
866 else:
867 self.fileno = lambda: None
Raymond Hettinger42182eb2003-03-09 05:33:33 +0000868 if hasattr(self.fp, "__iter__"):
869 self.__iter__ = self.fp.__iter__
Georg Brandla18af4e2007-04-21 15:47:16 +0000870 if hasattr(self.fp, "__next__"):
871 self.__next__ = self.fp.__next__
Guido van Rossume7b146f2000-02-04 15:28:42 +0000872
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000873 def __repr__(self):
Walter Dörwald70a6b492004-02-12 17:35:32 +0000874 return '<%s at %r whose fp = %r>' % (self.__class__.__name__,
875 id(self), self.fp)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000876
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000877 def close(self):
878 self.read = None
879 self.readline = None
880 self.readlines = None
881 self.fileno = None
882 if self.fp: self.fp.close()
883 self.fp = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000884
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000885class addclosehook(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000886 """Class to add a close hook to an open file."""
887
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000888 def __init__(self, fp, closehook, *hookargs):
889 addbase.__init__(self, fp)
890 self.closehook = closehook
891 self.hookargs = hookargs
Guido van Rossume7b146f2000-02-04 15:28:42 +0000892
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000893 def close(self):
Guido van Rossumc580dae2000-05-24 13:21:46 +0000894 addbase.close(self)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000895 if self.closehook:
Guido van Rossum68468eb2003-02-27 20:14:51 +0000896 self.closehook(*self.hookargs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000897 self.closehook = None
898 self.hookargs = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000899
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000900class addinfo(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000901 """class to add an info() method to an open file."""
902
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000903 def __init__(self, fp, headers):
904 addbase.__init__(self, fp)
905 self.headers = headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000906
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000907 def info(self):
908 return self.headers
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000909
Guido van Rossume6ad8911996-09-10 17:02:56 +0000910class addinfourl(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000911 """class to add info() and geturl() methods to an open file."""
912
Christian Heimes9bd667a2008-01-20 15:14:11 +0000913 def __init__(self, fp, headers, url, code=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000914 addbase.__init__(self, fp)
915 self.headers = headers
916 self.url = url
Christian Heimes9bd667a2008-01-20 15:14:11 +0000917 self.code = code
Guido van Rossume7b146f2000-02-04 15:28:42 +0000918
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000919 def info(self):
920 return self.headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000921
Christian Heimes9bd667a2008-01-20 15:14:11 +0000922 def getcode(self):
923 return self.code
924
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000925 def geturl(self):
926 return self.url
Guido van Rossume6ad8911996-09-10 17:02:56 +0000927
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000928
Guido van Rossum7c395db1994-07-04 22:14:49 +0000929# Utilities to parse URLs (most of these return None for missing parts):
Sjoerd Mullendere0371b81995-11-10 10:36:07 +0000930# unwrap('<URL:type://host/path>') --> 'type://host/path'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000931# splittype('type:opaquestring') --> 'type', 'opaquestring'
932# splithost('//host[:port]/path') --> 'host[:port]', '/path'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000933# splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'
934# splitpasswd('user:passwd') -> 'user', 'passwd'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000935# splitport('host:port') --> 'host', 'port'
936# splitquery('/path?query') --> '/path', 'query'
937# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000938# splitattr('/path;attr1=value1;attr2=value2;...') ->
939# '/path', ['attr1=value1', 'attr2=value2', ...]
940# splitvalue('attr=value') --> 'attr', 'value'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000941# unquote('abc%20def') -> 'abc def'
942# quote('abc def') -> 'abc%20def')
943
Martin v. Löwis1d994332000-12-03 18:30:10 +0000944def toBytes(url):
945 """toBytes(u"URL") --> 'URL'."""
946 # Most URL schemes require ASCII. If that changes, the conversion
Guido van Rossum98297ee2007-11-06 21:34:58 +0000947 # can be relaxed.
948 # XXX get rid of toBytes()
949 if isinstance(url, str):
Martin v. Löwis1d994332000-12-03 18:30:10 +0000950 try:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000951 url = url.encode("ASCII").decode()
Martin v. Löwis1d994332000-12-03 18:30:10 +0000952 except UnicodeError:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000953 raise UnicodeError("URL " + repr(url) +
954 " contains non-ASCII characters")
Martin v. Löwis1d994332000-12-03 18:30:10 +0000955 return url
956
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000957def unwrap(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000958 """unwrap('<URL:type://host/path>') --> 'type://host/path'."""
Guido van Rossuma0982942007-07-10 08:30:03 +0000959 url = str(url).strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000960 if url[:1] == '<' and url[-1:] == '>':
Guido van Rossumb2493f82000-12-15 15:01:37 +0000961 url = url[1:-1].strip()
962 if url[:4] == 'URL:': url = url[4:].strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000963 return url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000964
Guido van Rossum332e1441997-09-29 23:23:46 +0000965_typeprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000966def splittype(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000967 """splittype('type:opaquestring') --> 'type', 'opaquestring'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000968 global _typeprog
969 if _typeprog is None:
970 import re
971 _typeprog = re.compile('^([^/:]+):')
Guido van Rossum332e1441997-09-29 23:23:46 +0000972
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000973 match = _typeprog.match(url)
974 if match:
975 scheme = match.group(1)
Fred Drake9e94afd2000-07-01 07:03:30 +0000976 return scheme.lower(), url[len(scheme) + 1:]
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000977 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000978
Guido van Rossum332e1441997-09-29 23:23:46 +0000979_hostprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000980def splithost(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000981 """splithost('//host[:port]/path') --> 'host[:port]', '/path'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000982 global _hostprog
983 if _hostprog is None:
984 import re
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000985 _hostprog = re.compile('^//([^/?]*)(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +0000986
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000987 match = _hostprog.match(url)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000988 if match: return match.group(1, 2)
989 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000990
Guido van Rossum332e1441997-09-29 23:23:46 +0000991_userprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +0000992def splituser(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000993 """splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000994 global _userprog
995 if _userprog is None:
996 import re
Raymond Hettingerf2e45dd2002-08-18 20:08:56 +0000997 _userprog = re.compile('^(.*)@(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +0000998
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000999 match = _userprog.match(host)
Fred Drake567ca8e2000-08-21 21:42:42 +00001000 if match: return map(unquote, match.group(1, 2))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001001 return None, host
Guido van Rossum7c395db1994-07-04 22:14:49 +00001002
Guido van Rossum332e1441997-09-29 23:23:46 +00001003_passwdprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001004def splitpasswd(user):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001005 """splitpasswd('user:passwd') -> 'user', 'passwd'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001006 global _passwdprog
1007 if _passwdprog is None:
1008 import re
1009 _passwdprog = re.compile('^([^:]*):(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001010
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001011 match = _passwdprog.match(user)
1012 if match: return match.group(1, 2)
1013 return user, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001014
Guido van Rossume7b146f2000-02-04 15:28:42 +00001015# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum332e1441997-09-29 23:23:46 +00001016_portprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001017def splitport(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001018 """splitport('host:port') --> 'host', 'port'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001019 global _portprog
1020 if _portprog is None:
1021 import re
1022 _portprog = re.compile('^(.*):([0-9]+)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001023
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001024 match = _portprog.match(host)
1025 if match: return match.group(1, 2)
1026 return host, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001027
Guido van Rossum332e1441997-09-29 23:23:46 +00001028_nportprog = None
Guido van Rossum53725a21996-06-13 19:12:35 +00001029def splitnport(host, defport=-1):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001030 """Split host and port, returning numeric port.
1031 Return given default port if no ':' found; defaults to -1.
1032 Return numerical port if a valid number are found after ':'.
1033 Return None if ':' but not a valid number."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001034 global _nportprog
1035 if _nportprog is None:
1036 import re
1037 _nportprog = re.compile('^(.*):(.*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001038
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001039 match = _nportprog.match(host)
1040 if match:
1041 host, port = match.group(1, 2)
1042 try:
Collin Winterce36ad82007-08-30 01:19:48 +00001043 if not port: raise ValueError("no digits")
Guido van Rossumb2493f82000-12-15 15:01:37 +00001044 nport = int(port)
1045 except ValueError:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001046 nport = None
1047 return host, nport
1048 return host, defport
Guido van Rossum53725a21996-06-13 19:12:35 +00001049
Guido van Rossum332e1441997-09-29 23:23:46 +00001050_queryprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001051def splitquery(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001052 """splitquery('/path?query') --> '/path', 'query'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001053 global _queryprog
1054 if _queryprog is None:
1055 import re
1056 _queryprog = re.compile('^(.*)\?([^?]*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001057
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001058 match = _queryprog.match(url)
1059 if match: return match.group(1, 2)
1060 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001061
Guido van Rossum332e1441997-09-29 23:23:46 +00001062_tagprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001063def splittag(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001064 """splittag('/path#tag') --> '/path', 'tag'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001065 global _tagprog
1066 if _tagprog is None:
1067 import re
1068 _tagprog = re.compile('^(.*)#([^#]*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001069
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001070 match = _tagprog.match(url)
1071 if match: return match.group(1, 2)
1072 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001073
Guido van Rossum7c395db1994-07-04 22:14:49 +00001074def splitattr(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001075 """splitattr('/path;attr1=value1;attr2=value2;...') ->
1076 '/path', ['attr1=value1', 'attr2=value2', ...]."""
Guido van Rossumb2493f82000-12-15 15:01:37 +00001077 words = url.split(';')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001078 return words[0], words[1:]
Guido van Rossum7c395db1994-07-04 22:14:49 +00001079
Guido van Rossum332e1441997-09-29 23:23:46 +00001080_valueprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001081def splitvalue(attr):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001082 """splitvalue('attr=value') --> 'attr', 'value'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001083 global _valueprog
1084 if _valueprog is None:
1085 import re
1086 _valueprog = re.compile('^([^=]*)=(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001087
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001088 match = _valueprog.match(attr)
1089 if match: return match.group(1, 2)
1090 return attr, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001091
Raymond Hettinger803ce802005-09-10 06:49:04 +00001092_hextochr = dict(('%02x' % i, chr(i)) for i in range(256))
1093_hextochr.update(('%02X' % i, chr(i)) for i in range(256))
1094
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001095def unquote(s):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001096 """unquote('abc%20def') -> 'abc def'."""
Raymond Hettinger803ce802005-09-10 06:49:04 +00001097 res = s.split('%')
Guido van Rossum805365e2007-05-07 22:24:25 +00001098 for i in range(1, len(res)):
Raymond Hettinger803ce802005-09-10 06:49:04 +00001099 item = res[i]
1100 try:
1101 res[i] = _hextochr[item[:2]] + item[2:]
1102 except KeyError:
1103 res[i] = '%' + item
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +00001104 except UnicodeDecodeError:
Guido van Rossum84fc66d2007-05-03 17:18:26 +00001105 res[i] = chr(int(item[:2], 16)) + item[2:]
Guido van Rossumb2493f82000-12-15 15:01:37 +00001106 return "".join(res)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001107
Guido van Rossum0564e121996-12-13 14:47:36 +00001108def unquote_plus(s):
Skip Montanaro79f1c172000-08-22 03:00:52 +00001109 """unquote('%7e/abc+def') -> '~/abc def'"""
Brett Cannonaaeffaf2004-03-23 23:50:17 +00001110 s = s.replace('+', ' ')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001111 return unquote(s)
Guido van Rossum0564e121996-12-13 14:47:36 +00001112
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001113always_safe = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
Jeremy Hylton6102e292000-08-31 15:48:10 +00001114 'abcdefghijklmnopqrstuvwxyz'
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001115 '0123456789' '_.-')
Guido van Rossumc0f2d2d2007-08-03 19:19:24 +00001116_safe_quoters= {}
1117
1118class Quoter:
1119 def __init__(self, safe):
1120 self.cache = {}
1121 self.safe = safe + always_safe
1122
1123 def __call__(self, c):
1124 try:
1125 return self.cache[c]
1126 except KeyError:
1127 if ord(c) < 256:
1128 res = (c in self.safe) and c or ('%%%02X' % ord(c))
1129 self.cache[c] = res
1130 return res
1131 else:
1132 return "".join(['%%%02X' % i for i in c.encode("utf-8")])
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001133
Guido van Rossum7c395db1994-07-04 22:14:49 +00001134def quote(s, safe = '/'):
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001135 """quote('abc def') -> 'abc%20def'
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001136
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001137 Each part of a URL, e.g. the path info, the query, etc., has a
1138 different set of reserved characters that must be quoted.
1139
1140 RFC 2396 Uniform Resource Identifiers (URI): Generic Syntax lists
1141 the following reserved characters.
1142
1143 reserved = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" |
1144 "$" | ","
1145
1146 Each of these characters is reserved in some component of a URL,
1147 but not necessarily in all of them.
1148
1149 By default, the quote function is intended for quoting the path
1150 section of a URL. Thus, it will not encode '/'. This character
1151 is reserved, but in typical usage the quote function is being
1152 called on a path where the existing slash characters are used as
1153 reserved characters.
1154 """
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001155 cachekey = (safe, always_safe)
1156 try:
Guido van Rossumc0f2d2d2007-08-03 19:19:24 +00001157 quoter = _safe_quoters[cachekey]
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001158 except KeyError:
Guido van Rossumc0f2d2d2007-08-03 19:19:24 +00001159 quoter = Quoter(safe)
1160 _safe_quoters[cachekey] = quoter
1161 res = map(quoter, s)
Guido van Rossumb2493f82000-12-15 15:01:37 +00001162 return ''.join(res)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001163
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001164def quote_plus(s, safe = ''):
1165 """Quote the query fragment of a URL; replacing ' ' with '+'"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001166 if ' ' in s:
Raymond Hettingercf6b6322005-09-10 18:17:54 +00001167 s = quote(s, safe + ' ')
1168 return s.replace(' ', '+')
1169 return quote(s, safe)
Guido van Rossum0564e121996-12-13 14:47:36 +00001170
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001171def urlencode(query,doseq=0):
1172 """Encode a sequence of two-element tuples or dictionary into a URL query string.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001173
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001174 If any values in the query arg are sequences and doseq is true, each
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001175 sequence element is converted to a separate parameter.
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001176
1177 If the query arg is a sequence of two-element tuples, the order of the
1178 parameters in the output will match the order of parameters in the
1179 input.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001180 """
Tim Peters658cba62001-02-09 20:06:00 +00001181
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001182 if hasattr(query,"items"):
1183 # mapping objects
1184 query = query.items()
1185 else:
1186 # it's a bother at times that strings and string-like objects are
1187 # sequences...
1188 try:
1189 # non-sequence items should not work with len()
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001190 # non-empty strings will fail this
Walter Dörwald65230a22002-06-03 15:58:32 +00001191 if len(query) and not isinstance(query[0], tuple):
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001192 raise TypeError
1193 # zero-length sequences of all types will get here and succeed,
1194 # but that's a minor nit - since the original implementation
1195 # allowed empty dicts that type of behavior probably should be
1196 # preserved for consistency
1197 except TypeError:
1198 ty,va,tb = sys.exc_info()
Collin Winter828f04a2007-08-31 00:04:24 +00001199 raise TypeError("not a valid non-string sequence or mapping object").with_traceback(tb)
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001200
Guido van Rossume7b146f2000-02-04 15:28:42 +00001201 l = []
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001202 if not doseq:
1203 # preserve old behavior
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001204 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001205 k = quote_plus(str(k))
1206 v = quote_plus(str(v))
1207 l.append(k + '=' + v)
1208 else:
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001209 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001210 k = quote_plus(str(k))
Walter Dörwald65230a22002-06-03 15:58:32 +00001211 if isinstance(v, str):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001212 v = quote_plus(v)
1213 l.append(k + '=' + v)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001214 elif isinstance(v, str):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001215 # is there a reasonable way to convert to ASCII?
1216 # encode generates a string, but "replace" or "ignore"
1217 # lose information and "strict" can raise UnicodeError
1218 v = quote_plus(v.encode("ASCII","replace"))
1219 l.append(k + '=' + v)
1220 else:
1221 try:
1222 # is this a sufficient test for sequence-ness?
1223 x = len(v)
1224 except TypeError:
1225 # not a sequence
1226 v = quote_plus(str(v))
1227 l.append(k + '=' + v)
1228 else:
1229 # loop over the sequence
1230 for elt in v:
1231 l.append(k + '=' + quote_plus(str(elt)))
Guido van Rossumb2493f82000-12-15 15:01:37 +00001232 return '&'.join(l)
Guido van Rossum810a3391998-07-22 21:33:23 +00001233
Guido van Rossum442e7201996-03-20 15:33:11 +00001234# Proxy handling
Mark Hammond4f570b92000-07-26 07:04:38 +00001235def getproxies_environment():
1236 """Return a dictionary of scheme -> proxy server URL mappings.
1237
1238 Scan the environment for variables named <scheme>_proxy;
1239 this seems to be the standard convention. If you need a
1240 different way, you can pass a proxies dictionary to the
1241 [Fancy]URLopener constructor.
1242
1243 """
1244 proxies = {}
1245 for name, value in os.environ.items():
Guido van Rossumb2493f82000-12-15 15:01:37 +00001246 name = name.lower()
Christian Heimes9bd667a2008-01-20 15:14:11 +00001247 if name == 'no_proxy':
1248 # handled in proxy_bypass_environment
1249 continue
Mark Hammond4f570b92000-07-26 07:04:38 +00001250 if value and name[-6:] == '_proxy':
1251 proxies[name[:-6]] = value
1252 return proxies
1253
Christian Heimes9bd667a2008-01-20 15:14:11 +00001254def proxy_bypass_environment(host):
1255 """Test if proxies should not be used for a particular host.
1256
1257 Checks the environment for a variable named no_proxy, which should
1258 be a list of DNS suffixes separated by commas, or '*' for all hosts.
1259 """
1260 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
1261 # '*' is special case for always bypass
1262 if no_proxy == '*':
1263 return 1
1264 # strip port off host
1265 hostonly, port = splitport(host)
1266 # check if the host ends with any of the DNS suffixes
1267 for name in no_proxy.split(','):
1268 if name and (hostonly.endswith(name) or host.endswith(name)):
1269 return 1
1270 # otherwise, don't bypass
1271 return 0
1272
1273
Jack Jansen11d9b062004-07-16 11:45:00 +00001274if sys.platform == 'darwin':
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001275
1276 def _CFSetup(sc):
1277 from ctypes import c_int32, c_void_p, c_char_p, c_int
1278 sc.CFStringCreateWithCString.argtypes = [ c_void_p, c_char_p, c_int32 ]
1279 sc.CFStringCreateWithCString.restype = c_void_p
1280 sc.SCDynamicStoreCopyProxies.argtypes = [ c_void_p ]
1281 sc.SCDynamicStoreCopyProxies.restype = c_void_p
1282 sc.CFDictionaryGetValue.argtypes = [ c_void_p, c_void_p ]
1283 sc.CFDictionaryGetValue.restype = c_void_p
1284 sc.CFStringGetLength.argtypes = [ c_void_p ]
1285 sc.CFStringGetLength.restype = c_int32
1286 sc.CFStringGetCString.argtypes = [ c_void_p, c_char_p, c_int32, c_int32 ]
1287 sc.CFStringGetCString.restype = c_int32
1288 sc.CFNumberGetValue.argtypes = [ c_void_p, c_int, c_void_p ]
1289 sc.CFNumberGetValue.restype = c_int32
1290 sc.CFRelease.argtypes = [ c_void_p ]
1291 sc.CFRelease.restype = None
1292
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001293 def _CStringFromCFString(sc, value):
1294 from ctypes import create_string_buffer
1295 length = sc.CFStringGetLength(value) + 1
1296 buff = create_string_buffer(length)
1297 sc.CFStringGetCString(value, buff, length, 0)
1298 return buff.value
1299
1300 def _CFNumberToInt32(sc, cfnum):
1301 from ctypes import byref, c_int
1302 val = c_int()
1303 kCFNumberSInt32Type = 3
1304 sc.CFNumberGetValue(cfnum, kCFNumberSInt32Type, byref(val))
1305 return val.value
1306
1307
1308 def proxy_bypass_macosx_sysconf(host):
1309 """
1310 Return True iff this host shouldn't be accessed using a proxy
1311
1312 This function uses the MacOSX framework SystemConfiguration
1313 to fetch the proxy information.
1314 """
1315 from ctypes import cdll
1316 from ctypes.util import find_library
1317 import re
1318 import socket
1319 from fnmatch import fnmatch
1320
1321 def ip2num(ipAddr):
1322 parts = ipAddr.split('.')
1323 parts = map(int, parts)
1324 if len(parts) != 4:
1325 parts = (parts + [0, 0, 0, 0])[:4]
1326 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
1327
1328 sc = cdll.LoadLibrary(find_library("SystemConfiguration"))
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001329 _CFSetup(sc)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001330
1331 hostIP = None
1332
1333 if not sc:
1334 return False
1335
1336 kSCPropNetProxiesExceptionsList = sc.CFStringCreateWithCString(0, "ExceptionsList", 0)
1337 kSCPropNetProxiesExcludeSimpleHostnames = sc.CFStringCreateWithCString(0,
1338 "ExcludeSimpleHostnames", 0)
1339
1340
1341 proxyDict = sc.SCDynamicStoreCopyProxies(None)
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001342 if proxyDict is None:
1343 return False
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001344
1345 try:
1346 # Check for simple host names:
1347 if '.' not in host:
1348 exclude_simple = sc.CFDictionaryGetValue(proxyDict,
1349 kSCPropNetProxiesExcludeSimpleHostnames)
1350 if exclude_simple and _CFNumberToInt32(sc, exclude_simple):
1351 return True
1352
1353
1354 # Check the exceptions list:
1355 exceptions = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesExceptionsList)
1356 if exceptions:
1357 # Items in the list are strings like these: *.local, 169.254/16
1358 for index in xrange(sc.CFArrayGetCount(exceptions)):
1359 value = sc.CFArrayGetValueAtIndex(exceptions, index)
1360 if not value: continue
1361 value = _CStringFromCFString(sc, value)
1362
1363 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
1364 if m is not None:
1365 if hostIP is None:
1366 hostIP = socket.gethostbyname(host)
1367 hostIP = ip2num(hostIP)
1368
1369 base = ip2num(m.group(1))
1370 mask = int(m.group(2)[1:])
1371 mask = 32 - mask
1372
1373 if (hostIP >> mask) == (base >> mask):
1374 return True
1375
1376 elif fnmatch(host, value):
1377 return True
1378
1379 return False
1380
1381 finally:
1382 sc.CFRelease(kSCPropNetProxiesExceptionsList)
1383 sc.CFRelease(kSCPropNetProxiesExcludeSimpleHostnames)
1384
1385
1386
1387 def getproxies_macosx_sysconf():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001388 """Return a dictionary of scheme -> proxy server URL mappings.
Guido van Rossum442e7201996-03-20 15:33:11 +00001389
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001390 This function uses the MacOSX framework SystemConfiguration
1391 to fetch the proxy information.
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001392 """
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001393 from ctypes import cdll
1394 from ctypes.util import find_library
1395
1396 sc = cdll.LoadLibrary(find_library("SystemConfiguration"))
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001397 _CFSetup(sc)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001398
1399 if not sc:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001400 return {}
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001401
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001402 kSCPropNetProxiesHTTPEnable = sc.CFStringCreateWithCString(0, b"HTTPEnable", 0)
1403 kSCPropNetProxiesHTTPProxy = sc.CFStringCreateWithCString(0, b"HTTPProxy", 0)
1404 kSCPropNetProxiesHTTPPort = sc.CFStringCreateWithCString(0, b"HTTPPort", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001405
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001406 kSCPropNetProxiesHTTPSEnable = sc.CFStringCreateWithCString(0, b"HTTPSEnable", 0)
1407 kSCPropNetProxiesHTTPSProxy = sc.CFStringCreateWithCString(0, b"HTTPSProxy", 0)
1408 kSCPropNetProxiesHTTPSPort = sc.CFStringCreateWithCString(0, b"HTTPSPort", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001409
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001410 kSCPropNetProxiesFTPEnable = sc.CFStringCreateWithCString(0, b"FTPEnable", 0)
1411 kSCPropNetProxiesFTPPassive = sc.CFStringCreateWithCString(0, b"FTPPassive", 0)
1412 kSCPropNetProxiesFTPPort = sc.CFStringCreateWithCString(0, b"FTPPort", 0)
1413 kSCPropNetProxiesFTPProxy = sc.CFStringCreateWithCString(0, b"FTPProxy", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001414
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001415 kSCPropNetProxiesGopherEnable = sc.CFStringCreateWithCString(0, b"GopherEnable", 0)
1416 kSCPropNetProxiesGopherPort = sc.CFStringCreateWithCString(0, b"GopherPort", 0)
1417 kSCPropNetProxiesGopherProxy = sc.CFStringCreateWithCString(0, b"GopherProxy", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001418
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001419 proxies = {}
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001420 proxyDict = sc.SCDynamicStoreCopyProxies(None)
1421
1422 try:
1423 # HTTP:
1424 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPEnable)
1425 if enabled and _CFNumberToInt32(sc, enabled):
1426 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPProxy)
1427 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPPort)
1428
1429 if proxy:
1430 proxy = _CStringFromCFString(sc, proxy)
1431 if port:
1432 port = _CFNumberToInt32(sc, port)
1433 proxies["http"] = "http://%s:%i" % (proxy, port)
1434 else:
1435 proxies["http"] = "http://%s" % (proxy, )
1436
1437 # HTTPS:
1438 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSEnable)
1439 if enabled and _CFNumberToInt32(sc, enabled):
1440 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSProxy)
1441 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSPort)
1442
1443 if proxy:
1444 proxy = _CStringFromCFString(sc, proxy)
1445 if port:
1446 port = _CFNumberToInt32(sc, port)
1447 proxies["https"] = "http://%s:%i" % (proxy, port)
1448 else:
1449 proxies["https"] = "http://%s" % (proxy, )
1450
1451 # FTP:
1452 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPEnable)
1453 if enabled and _CFNumberToInt32(sc, enabled):
1454 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPProxy)
1455 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPPort)
1456
1457 if proxy:
1458 proxy = _CStringFromCFString(sc, proxy)
1459 if port:
1460 port = _CFNumberToInt32(sc, port)
1461 proxies["ftp"] = "http://%s:%i" % (proxy, port)
1462 else:
1463 proxies["ftp"] = "http://%s" % (proxy, )
1464
1465 # Gopher:
1466 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherEnable)
1467 if enabled and _CFNumberToInt32(sc, enabled):
1468 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherProxy)
1469 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherPort)
1470
1471 if proxy:
1472 proxy = _CStringFromCFString(sc, proxy)
1473 if port:
1474 port = _CFNumberToInt32(sc, port)
1475 proxies["gopher"] = "http://%s:%i" % (proxy, port)
1476 else:
1477 proxies["gopher"] = "http://%s" % (proxy, )
1478 finally:
1479 sc.CFRelease(proxyDict)
1480
1481 sc.CFRelease(kSCPropNetProxiesHTTPEnable)
1482 sc.CFRelease(kSCPropNetProxiesHTTPProxy)
1483 sc.CFRelease(kSCPropNetProxiesHTTPPort)
1484 sc.CFRelease(kSCPropNetProxiesFTPEnable)
1485 sc.CFRelease(kSCPropNetProxiesFTPPassive)
1486 sc.CFRelease(kSCPropNetProxiesFTPPort)
1487 sc.CFRelease(kSCPropNetProxiesFTPProxy)
1488 sc.CFRelease(kSCPropNetProxiesGopherEnable)
1489 sc.CFRelease(kSCPropNetProxiesGopherPort)
1490 sc.CFRelease(kSCPropNetProxiesGopherProxy)
1491
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001492 return proxies
Mark Hammond4f570b92000-07-26 07:04:38 +00001493
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001494
1495
Christian Heimes9bd667a2008-01-20 15:14:11 +00001496 def proxy_bypass(host):
1497 if getproxies_environment():
1498 return proxy_bypass_environment(host)
1499 else:
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001500 return proxy_bypass_macosx_sysconf(host)
Tim Peters55c12d42001-08-09 18:04:14 +00001501
Jack Jansen11d9b062004-07-16 11:45:00 +00001502 def getproxies():
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001503 return getproxies_environment() or getproxies_macosx_sysconf()
Tim Peters182b5ac2004-07-18 06:16:08 +00001504
Mark Hammond4f570b92000-07-26 07:04:38 +00001505elif os.name == 'nt':
1506 def getproxies_registry():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001507 """Return a dictionary of scheme -> proxy server URL mappings.
Mark Hammond4f570b92000-07-26 07:04:38 +00001508
1509 Win32 uses the registry to store proxies.
1510
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001511 """
1512 proxies = {}
Mark Hammond4f570b92000-07-26 07:04:38 +00001513 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001514 import winreg
Mark Hammond4f570b92000-07-26 07:04:38 +00001515 except ImportError:
1516 # Std module, so should be around - but you never know!
1517 return proxies
1518 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001519 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001520 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl38feaf02008-05-25 07:45:51 +00001521 proxyEnable = winreg.QueryValueEx(internetSettings,
1522 'ProxyEnable')[0]
Mark Hammond4f570b92000-07-26 07:04:38 +00001523 if proxyEnable:
1524 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl38feaf02008-05-25 07:45:51 +00001525 proxyServer = str(winreg.QueryValueEx(internetSettings,
1526 'ProxyServer')[0])
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001527 if '=' in proxyServer:
1528 # Per-protocol settings
Mark Hammond4f570b92000-07-26 07:04:38 +00001529 for p in proxyServer.split(';'):
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001530 protocol, address = p.split('=', 1)
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001531 # See if address has a type:// prefix
Guido van Rossum64e5aa92002-04-02 14:38:16 +00001532 import re
1533 if not re.match('^([^/:]+)://', address):
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001534 address = '%s://%s' % (protocol, address)
1535 proxies[protocol] = address
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001536 else:
1537 # Use one setting for all protocols
1538 if proxyServer[:5] == 'http:':
1539 proxies['http'] = proxyServer
1540 else:
1541 proxies['http'] = 'http://%s' % proxyServer
1542 proxies['ftp'] = 'ftp://%s' % proxyServer
Mark Hammond4f570b92000-07-26 07:04:38 +00001543 internetSettings.Close()
1544 except (WindowsError, ValueError, TypeError):
1545 # Either registry key not found etc, or the value in an
1546 # unexpected format.
1547 # proxies already set up to be empty so nothing to do
1548 pass
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001549 return proxies
Guido van Rossum442e7201996-03-20 15:33:11 +00001550
Mark Hammond4f570b92000-07-26 07:04:38 +00001551 def getproxies():
1552 """Return a dictionary of scheme -> proxy server URL mappings.
1553
1554 Returns settings gathered from the environment, if specified,
1555 or the registry.
1556
1557 """
1558 return getproxies_environment() or getproxies_registry()
Tim Peters55c12d42001-08-09 18:04:14 +00001559
Christian Heimes9bd667a2008-01-20 15:14:11 +00001560 def proxy_bypass_registry(host):
Tim Peters55c12d42001-08-09 18:04:14 +00001561 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001562 import winreg
Tim Peters55c12d42001-08-09 18:04:14 +00001563 import re
Tim Peters55c12d42001-08-09 18:04:14 +00001564 except ImportError:
1565 # Std modules, so should be around - but you never know!
1566 return 0
1567 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001568 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Tim Peters55c12d42001-08-09 18:04:14 +00001569 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl38feaf02008-05-25 07:45:51 +00001570 proxyEnable = winreg.QueryValueEx(internetSettings,
1571 'ProxyEnable')[0]
1572 proxyOverride = str(winreg.QueryValueEx(internetSettings,
1573 'ProxyOverride')[0])
Tim Peters55c12d42001-08-09 18:04:14 +00001574 # ^^^^ Returned as Unicode but problems if not converted to ASCII
1575 except WindowsError:
1576 return 0
1577 if not proxyEnable or not proxyOverride:
1578 return 0
1579 # try to make a host list from name and IP address.
Georg Brandl1f636702006-02-18 23:10:23 +00001580 rawHost, port = splitport(host)
1581 host = [rawHost]
Tim Peters55c12d42001-08-09 18:04:14 +00001582 try:
Georg Brandl1f636702006-02-18 23:10:23 +00001583 addr = socket.gethostbyname(rawHost)
1584 if addr != rawHost:
Tim Peters55c12d42001-08-09 18:04:14 +00001585 host.append(addr)
1586 except socket.error:
1587 pass
Georg Brandl1f636702006-02-18 23:10:23 +00001588 try:
1589 fqdn = socket.getfqdn(rawHost)
1590 if fqdn != rawHost:
1591 host.append(fqdn)
1592 except socket.error:
1593 pass
Tim Peters55c12d42001-08-09 18:04:14 +00001594 # make a check value list from the registry entry: replace the
1595 # '<local>' string by the localhost entry and the corresponding
1596 # canonical entry.
1597 proxyOverride = proxyOverride.split(';')
1598 i = 0
1599 while i < len(proxyOverride):
1600 if proxyOverride[i] == '<local>':
1601 proxyOverride[i:i+1] = ['localhost',
1602 '127.0.0.1',
1603 socket.gethostname(),
1604 socket.gethostbyname(
1605 socket.gethostname())]
1606 i += 1
1607 # print proxyOverride
1608 # now check if we match one of the registry values.
1609 for test in proxyOverride:
Tim Petersab9ba272001-08-09 21:40:30 +00001610 test = test.replace(".", r"\.") # mask dots
1611 test = test.replace("*", r".*") # change glob sequence
1612 test = test.replace("?", r".") # change glob char
Tim Peters55c12d42001-08-09 18:04:14 +00001613 for val in host:
1614 # print "%s <--> %s" %( test, val )
1615 if re.match(test, val, re.I):
1616 return 1
1617 return 0
1618
Christian Heimes9bd667a2008-01-20 15:14:11 +00001619 def proxy_bypass(host):
1620 """Return a dictionary of scheme -> proxy server URL mappings.
1621
1622 Returns settings gathered from the environment, if specified,
1623 or the registry.
1624
1625 """
1626 if getproxies_environment():
1627 return proxy_bypass_environment(host)
1628 else:
1629 return proxy_bypass_registry(host)
1630
Mark Hammond4f570b92000-07-26 07:04:38 +00001631else:
1632 # By default use environment variables
1633 getproxies = getproxies_environment
Christian Heimes9bd667a2008-01-20 15:14:11 +00001634 proxy_bypass = proxy_bypass_environment
Guido van Rossum442e7201996-03-20 15:33:11 +00001635
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001636# Test and time quote() and unquote()
1637def test1():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001638 s = ''
1639 for i in range(256): s = s + chr(i)
1640 s = s*4
1641 t0 = time.time()
1642 qs = quote(s)
1643 uqs = unquote(qs)
1644 t1 = time.time()
1645 if uqs != s:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001646 print('Wrong!')
1647 print(repr(s))
1648 print(repr(qs))
1649 print(repr(uqs))
1650 print(round(t1 - t0, 3), 'sec')
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001651
1652
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001653def reporthook(blocknum, blocksize, totalsize):
1654 # Report during remote transfers
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001655 print("Block number: %d, Block size: %d, Total size: %d" % (
1656 blocknum, blocksize, totalsize))
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001657
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001658# Test program
Guido van Rossum23490151998-06-25 02:39:00 +00001659def test(args=[]):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001660 if not args:
1661 args = [
1662 '/etc/passwd',
1663 'file:/etc/passwd',
1664 'file://localhost/etc/passwd',
Guido van Rossumd8faa362007-04-27 19:54:29 +00001665 'ftp://ftp.gnu.org/pub/README',
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001666 'http://www.python.org/index.html',
1667 ]
Guido van Rossum09c8b6c1999-12-07 21:37:17 +00001668 if hasattr(URLopener, "open_https"):
1669 args.append('https://synergy.as.cmu.edu/~geek/')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001670 try:
1671 for url in args:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001672 print('-'*10, url, '-'*10)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001673 fn, h = urlretrieve(url, None, reporthook)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001674 print(fn)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001675 if h:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001676 print('======')
1677 for k in h.keys(): print(k + ':', h[k])
1678 print('======')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001679 fp = open(fn, 'rb')
1680 data = fp.read()
1681 del fp
Georg Brandl7f13e6b2007-08-31 10:37:15 +00001682 data = data.replace("\r", "")
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001683 print(data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001684 fn, h = None, None
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001685 print('-'*40)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001686 finally:
1687 urlcleanup()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001688
Guido van Rossum23490151998-06-25 02:39:00 +00001689def main():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001690 import getopt, sys
1691 try:
1692 opts, args = getopt.getopt(sys.argv[1:], "th")
Guido van Rossumb940e112007-01-10 16:19:56 +00001693 except getopt.error as msg:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001694 print(msg)
1695 print("Use -h for help")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001696 return
1697 t = 0
1698 for o, a in opts:
1699 if o == '-t':
1700 t = t + 1
1701 if o == '-h':
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001702 print("Usage: python urllib.py [-t] [url ...]")
1703 print("-t runs self-test;", end=' ')
1704 print("otherwise, contents of urls are printed")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001705 return
1706 if t:
1707 if t > 1:
1708 test1()
1709 test(args)
1710 else:
1711 if not args:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001712 print("Use -h for help")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001713 for url in args:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001714 print(urlopen(url).read(), end=' ')
Guido van Rossum23490151998-06-25 02:39:00 +00001715
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001716# Run test program when run as a script
1717if __name__ == '__main__':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001718 main()