blob: 500bf06544e15daca38c54b40175c01e0e020426 [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
Jeremy Hylton5f22af12007-08-16 17:55:18 +000025import httplib
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()
355 except httplib.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."""
372 return self._open_generic_http(httplib.HTTPConnection, url, data)
373
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):
398 return httplib.HTTPSConnection(host,
399 key_file=self.key_file,
400 cert_file=self.cert_file)
401
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
Guido van Rossume7ba4952007-06-06 23:52:48 +0000779 def __init__(self, user, passwd, host, port, dirs, timeout=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000780 self.user = user
781 self.passwd = passwd
782 self.host = host
783 self.port = port
784 self.dirs = dirs
Guido van Rossume7ba4952007-06-06 23:52:48 +0000785 self.timeout = timeout
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000786 self.init()
Guido van Rossume7b146f2000-02-04 15:28:42 +0000787
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000788 def init(self):
789 import ftplib
790 self.busy = 0
791 self.ftp = ftplib.FTP()
Guido van Rossume7ba4952007-06-06 23:52:48 +0000792 self.ftp.connect(self.host, self.port, self.timeout)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000793 self.ftp.login(self.user, self.passwd)
794 for dir in self.dirs:
795 self.ftp.cwd(dir)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000796
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000797 def retrfile(self, file, type):
798 import ftplib
799 self.endtransfer()
800 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
801 else: cmd = 'TYPE ' + type; isdir = 0
802 try:
803 self.ftp.voidcmd(cmd)
804 except ftplib.all_errors:
805 self.init()
806 self.ftp.voidcmd(cmd)
807 conn = None
808 if file and not isdir:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000809 # Try to retrieve as a file
810 try:
811 cmd = 'RETR ' + file
812 conn = self.ftp.ntransfercmd(cmd)
Guido van Rossumb940e112007-01-10 16:19:56 +0000813 except ftplib.error_perm as reason:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000814 if str(reason)[:3] != '550':
Collin Winter828f04a2007-08-31 00:04:24 +0000815 raise IOError('ftp error', reason).with_traceback(sys.exc_info()[2])
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000816 if not conn:
817 # Set transfer mode to ASCII!
818 self.ftp.voidcmd('TYPE A')
Christian Heimes9bd667a2008-01-20 15:14:11 +0000819 # Try a directory listing. Verify that directory exists.
820 if file:
821 pwd = self.ftp.pwd()
822 try:
823 try:
824 self.ftp.cwd(file)
825 except ftplib.error_perm as reason:
826 raise IOError('ftp error', reason) from reason
827 finally:
828 self.ftp.cwd(pwd)
829 cmd = 'LIST ' + file
830 else:
831 cmd = 'LIST'
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000832 conn = self.ftp.ntransfercmd(cmd)
833 self.busy = 1
834 # Pass back both a suitably decorated object and a retrieval length
835 return (addclosehook(conn[0].makefile('rb'),
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000836 self.endtransfer), conn[1])
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000837 def endtransfer(self):
838 if not self.busy:
839 return
840 self.busy = 0
841 try:
842 self.ftp.voidresp()
843 except ftperrors():
844 pass
Guido van Rossume7b146f2000-02-04 15:28:42 +0000845
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000846 def close(self):
847 self.endtransfer()
848 try:
849 self.ftp.close()
850 except ftperrors():
851 pass
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000852
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000853class addbase:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000854 """Base class for addinfo and addclosehook."""
855
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000856 # XXX Add a method to expose the timeout on the underlying socket?
857
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000858 def __init__(self, fp):
859 self.fp = fp
860 self.read = self.fp.read
861 self.readline = self.fp.readline
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000862 if hasattr(self.fp, "readlines"): self.readlines = self.fp.readlines
Georg Brandl1f663572005-11-26 16:50:44 +0000863 if hasattr(self.fp, "fileno"):
864 self.fileno = self.fp.fileno
865 else:
866 self.fileno = lambda: None
Raymond Hettinger42182eb2003-03-09 05:33:33 +0000867 if hasattr(self.fp, "__iter__"):
868 self.__iter__ = self.fp.__iter__
Georg Brandla18af4e2007-04-21 15:47:16 +0000869 if hasattr(self.fp, "__next__"):
870 self.__next__ = self.fp.__next__
Guido van Rossume7b146f2000-02-04 15:28:42 +0000871
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000872 def __repr__(self):
Walter Dörwald70a6b492004-02-12 17:35:32 +0000873 return '<%s at %r whose fp = %r>' % (self.__class__.__name__,
874 id(self), self.fp)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000875
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000876 def close(self):
877 self.read = None
878 self.readline = None
879 self.readlines = None
880 self.fileno = None
881 if self.fp: self.fp.close()
882 self.fp = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000883
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000884class addclosehook(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000885 """Class to add a close hook to an open file."""
886
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000887 def __init__(self, fp, closehook, *hookargs):
888 addbase.__init__(self, fp)
889 self.closehook = closehook
890 self.hookargs = hookargs
Guido van Rossume7b146f2000-02-04 15:28:42 +0000891
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000892 def close(self):
Guido van Rossumc580dae2000-05-24 13:21:46 +0000893 addbase.close(self)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000894 if self.closehook:
Guido van Rossum68468eb2003-02-27 20:14:51 +0000895 self.closehook(*self.hookargs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000896 self.closehook = None
897 self.hookargs = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000898
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000899class addinfo(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000900 """class to add an info() method to an open file."""
901
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000902 def __init__(self, fp, headers):
903 addbase.__init__(self, fp)
904 self.headers = headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000905
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000906 def info(self):
907 return self.headers
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000908
Guido van Rossume6ad8911996-09-10 17:02:56 +0000909class addinfourl(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000910 """class to add info() and geturl() methods to an open file."""
911
Christian Heimes9bd667a2008-01-20 15:14:11 +0000912 def __init__(self, fp, headers, url, code=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000913 addbase.__init__(self, fp)
914 self.headers = headers
915 self.url = url
Christian Heimes9bd667a2008-01-20 15:14:11 +0000916 self.code = code
Guido van Rossume7b146f2000-02-04 15:28:42 +0000917
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000918 def info(self):
919 return self.headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000920
Christian Heimes9bd667a2008-01-20 15:14:11 +0000921 def getcode(self):
922 return self.code
923
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000924 def geturl(self):
925 return self.url
Guido van Rossume6ad8911996-09-10 17:02:56 +0000926
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000927
Guido van Rossum7c395db1994-07-04 22:14:49 +0000928# Utilities to parse URLs (most of these return None for missing parts):
Sjoerd Mullendere0371b81995-11-10 10:36:07 +0000929# unwrap('<URL:type://host/path>') --> 'type://host/path'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000930# splittype('type:opaquestring') --> 'type', 'opaquestring'
931# splithost('//host[:port]/path') --> 'host[:port]', '/path'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000932# splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'
933# splitpasswd('user:passwd') -> 'user', 'passwd'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000934# splitport('host:port') --> 'host', 'port'
935# splitquery('/path?query') --> '/path', 'query'
936# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000937# splitattr('/path;attr1=value1;attr2=value2;...') ->
938# '/path', ['attr1=value1', 'attr2=value2', ...]
939# splitvalue('attr=value') --> 'attr', 'value'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000940# unquote('abc%20def') -> 'abc def'
941# quote('abc def') -> 'abc%20def')
942
Martin v. Löwis1d994332000-12-03 18:30:10 +0000943def toBytes(url):
944 """toBytes(u"URL") --> 'URL'."""
945 # Most URL schemes require ASCII. If that changes, the conversion
Guido van Rossum98297ee2007-11-06 21:34:58 +0000946 # can be relaxed.
947 # XXX get rid of toBytes()
948 if isinstance(url, str):
Martin v. Löwis1d994332000-12-03 18:30:10 +0000949 try:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000950 url = url.encode("ASCII").decode()
Martin v. Löwis1d994332000-12-03 18:30:10 +0000951 except UnicodeError:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000952 raise UnicodeError("URL " + repr(url) +
953 " contains non-ASCII characters")
Martin v. Löwis1d994332000-12-03 18:30:10 +0000954 return url
955
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000956def unwrap(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000957 """unwrap('<URL:type://host/path>') --> 'type://host/path'."""
Guido van Rossuma0982942007-07-10 08:30:03 +0000958 url = str(url).strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000959 if url[:1] == '<' and url[-1:] == '>':
Guido van Rossumb2493f82000-12-15 15:01:37 +0000960 url = url[1:-1].strip()
961 if url[:4] == 'URL:': url = url[4:].strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000962 return url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000963
Guido van Rossum332e1441997-09-29 23:23:46 +0000964_typeprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000965def splittype(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000966 """splittype('type:opaquestring') --> 'type', 'opaquestring'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000967 global _typeprog
968 if _typeprog is None:
969 import re
970 _typeprog = re.compile('^([^/:]+):')
Guido van Rossum332e1441997-09-29 23:23:46 +0000971
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000972 match = _typeprog.match(url)
973 if match:
974 scheme = match.group(1)
Fred Drake9e94afd2000-07-01 07:03:30 +0000975 return scheme.lower(), url[len(scheme) + 1:]
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000976 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000977
Guido van Rossum332e1441997-09-29 23:23:46 +0000978_hostprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000979def splithost(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000980 """splithost('//host[:port]/path') --> 'host[:port]', '/path'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000981 global _hostprog
982 if _hostprog is None:
983 import re
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000984 _hostprog = re.compile('^//([^/?]*)(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +0000985
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000986 match = _hostprog.match(url)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000987 if match: return match.group(1, 2)
988 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000989
Guido van Rossum332e1441997-09-29 23:23:46 +0000990_userprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +0000991def splituser(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000992 """splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000993 global _userprog
994 if _userprog is None:
995 import re
Raymond Hettingerf2e45dd2002-08-18 20:08:56 +0000996 _userprog = re.compile('^(.*)@(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +0000997
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000998 match = _userprog.match(host)
Fred Drake567ca8e2000-08-21 21:42:42 +0000999 if match: return map(unquote, match.group(1, 2))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001000 return None, host
Guido van Rossum7c395db1994-07-04 22:14:49 +00001001
Guido van Rossum332e1441997-09-29 23:23:46 +00001002_passwdprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001003def splitpasswd(user):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001004 """splitpasswd('user:passwd') -> 'user', 'passwd'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001005 global _passwdprog
1006 if _passwdprog is None:
1007 import re
1008 _passwdprog = re.compile('^([^:]*):(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001009
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001010 match = _passwdprog.match(user)
1011 if match: return match.group(1, 2)
1012 return user, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001013
Guido van Rossume7b146f2000-02-04 15:28:42 +00001014# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum332e1441997-09-29 23:23:46 +00001015_portprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001016def splitport(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001017 """splitport('host:port') --> 'host', 'port'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001018 global _portprog
1019 if _portprog is None:
1020 import re
1021 _portprog = re.compile('^(.*):([0-9]+)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001022
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001023 match = _portprog.match(host)
1024 if match: return match.group(1, 2)
1025 return host, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001026
Guido van Rossum332e1441997-09-29 23:23:46 +00001027_nportprog = None
Guido van Rossum53725a21996-06-13 19:12:35 +00001028def splitnport(host, defport=-1):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001029 """Split host and port, returning numeric port.
1030 Return given default port if no ':' found; defaults to -1.
1031 Return numerical port if a valid number are found after ':'.
1032 Return None if ':' but not a valid number."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001033 global _nportprog
1034 if _nportprog is None:
1035 import re
1036 _nportprog = re.compile('^(.*):(.*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001037
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001038 match = _nportprog.match(host)
1039 if match:
1040 host, port = match.group(1, 2)
1041 try:
Collin Winterce36ad82007-08-30 01:19:48 +00001042 if not port: raise ValueError("no digits")
Guido van Rossumb2493f82000-12-15 15:01:37 +00001043 nport = int(port)
1044 except ValueError:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001045 nport = None
1046 return host, nport
1047 return host, defport
Guido van Rossum53725a21996-06-13 19:12:35 +00001048
Guido van Rossum332e1441997-09-29 23:23:46 +00001049_queryprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001050def splitquery(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001051 """splitquery('/path?query') --> '/path', 'query'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001052 global _queryprog
1053 if _queryprog is None:
1054 import re
1055 _queryprog = re.compile('^(.*)\?([^?]*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001056
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001057 match = _queryprog.match(url)
1058 if match: return match.group(1, 2)
1059 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001060
Guido van Rossum332e1441997-09-29 23:23:46 +00001061_tagprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001062def splittag(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001063 """splittag('/path#tag') --> '/path', 'tag'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001064 global _tagprog
1065 if _tagprog is None:
1066 import re
1067 _tagprog = re.compile('^(.*)#([^#]*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001068
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001069 match = _tagprog.match(url)
1070 if match: return match.group(1, 2)
1071 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001072
Guido van Rossum7c395db1994-07-04 22:14:49 +00001073def splitattr(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001074 """splitattr('/path;attr1=value1;attr2=value2;...') ->
1075 '/path', ['attr1=value1', 'attr2=value2', ...]."""
Guido van Rossumb2493f82000-12-15 15:01:37 +00001076 words = url.split(';')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001077 return words[0], words[1:]
Guido van Rossum7c395db1994-07-04 22:14:49 +00001078
Guido van Rossum332e1441997-09-29 23:23:46 +00001079_valueprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001080def splitvalue(attr):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001081 """splitvalue('attr=value') --> 'attr', 'value'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001082 global _valueprog
1083 if _valueprog is None:
1084 import re
1085 _valueprog = re.compile('^([^=]*)=(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001086
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001087 match = _valueprog.match(attr)
1088 if match: return match.group(1, 2)
1089 return attr, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001090
Raymond Hettinger803ce802005-09-10 06:49:04 +00001091_hextochr = dict(('%02x' % i, chr(i)) for i in range(256))
1092_hextochr.update(('%02X' % i, chr(i)) for i in range(256))
1093
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001094def unquote(s):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001095 """unquote('abc%20def') -> 'abc def'."""
Raymond Hettinger803ce802005-09-10 06:49:04 +00001096 res = s.split('%')
Guido van Rossum805365e2007-05-07 22:24:25 +00001097 for i in range(1, len(res)):
Raymond Hettinger803ce802005-09-10 06:49:04 +00001098 item = res[i]
1099 try:
1100 res[i] = _hextochr[item[:2]] + item[2:]
1101 except KeyError:
1102 res[i] = '%' + item
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +00001103 except UnicodeDecodeError:
Guido van Rossum84fc66d2007-05-03 17:18:26 +00001104 res[i] = chr(int(item[:2], 16)) + item[2:]
Guido van Rossumb2493f82000-12-15 15:01:37 +00001105 return "".join(res)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001106
Guido van Rossum0564e121996-12-13 14:47:36 +00001107def unquote_plus(s):
Skip Montanaro79f1c172000-08-22 03:00:52 +00001108 """unquote('%7e/abc+def') -> '~/abc def'"""
Brett Cannonaaeffaf2004-03-23 23:50:17 +00001109 s = s.replace('+', ' ')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001110 return unquote(s)
Guido van Rossum0564e121996-12-13 14:47:36 +00001111
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001112always_safe = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
Jeremy Hylton6102e292000-08-31 15:48:10 +00001113 'abcdefghijklmnopqrstuvwxyz'
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001114 '0123456789' '_.-')
Guido van Rossumc0f2d2d2007-08-03 19:19:24 +00001115_safe_quoters= {}
1116
1117class Quoter:
1118 def __init__(self, safe):
1119 self.cache = {}
1120 self.safe = safe + always_safe
1121
1122 def __call__(self, c):
1123 try:
1124 return self.cache[c]
1125 except KeyError:
1126 if ord(c) < 256:
1127 res = (c in self.safe) and c or ('%%%02X' % ord(c))
1128 self.cache[c] = res
1129 return res
1130 else:
1131 return "".join(['%%%02X' % i for i in c.encode("utf-8")])
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001132
Guido van Rossum7c395db1994-07-04 22:14:49 +00001133def quote(s, safe = '/'):
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001134 """quote('abc def') -> 'abc%20def'
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001135
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001136 Each part of a URL, e.g. the path info, the query, etc., has a
1137 different set of reserved characters that must be quoted.
1138
1139 RFC 2396 Uniform Resource Identifiers (URI): Generic Syntax lists
1140 the following reserved characters.
1141
1142 reserved = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" |
1143 "$" | ","
1144
1145 Each of these characters is reserved in some component of a URL,
1146 but not necessarily in all of them.
1147
1148 By default, the quote function is intended for quoting the path
1149 section of a URL. Thus, it will not encode '/'. This character
1150 is reserved, but in typical usage the quote function is being
1151 called on a path where the existing slash characters are used as
1152 reserved characters.
1153 """
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001154 cachekey = (safe, always_safe)
1155 try:
Guido van Rossumc0f2d2d2007-08-03 19:19:24 +00001156 quoter = _safe_quoters[cachekey]
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001157 except KeyError:
Guido van Rossumc0f2d2d2007-08-03 19:19:24 +00001158 quoter = Quoter(safe)
1159 _safe_quoters[cachekey] = quoter
1160 res = map(quoter, s)
Guido van Rossumb2493f82000-12-15 15:01:37 +00001161 return ''.join(res)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001162
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001163def quote_plus(s, safe = ''):
1164 """Quote the query fragment of a URL; replacing ' ' with '+'"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001165 if ' ' in s:
Raymond Hettingercf6b6322005-09-10 18:17:54 +00001166 s = quote(s, safe + ' ')
1167 return s.replace(' ', '+')
1168 return quote(s, safe)
Guido van Rossum0564e121996-12-13 14:47:36 +00001169
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001170def urlencode(query,doseq=0):
1171 """Encode a sequence of two-element tuples or dictionary into a URL query string.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001172
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001173 If any values in the query arg are sequences and doseq is true, each
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001174 sequence element is converted to a separate parameter.
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001175
1176 If the query arg is a sequence of two-element tuples, the order of the
1177 parameters in the output will match the order of parameters in the
1178 input.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001179 """
Tim Peters658cba62001-02-09 20:06:00 +00001180
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001181 if hasattr(query,"items"):
1182 # mapping objects
1183 query = query.items()
1184 else:
1185 # it's a bother at times that strings and string-like objects are
1186 # sequences...
1187 try:
1188 # non-sequence items should not work with len()
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001189 # non-empty strings will fail this
Walter Dörwald65230a22002-06-03 15:58:32 +00001190 if len(query) and not isinstance(query[0], tuple):
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001191 raise TypeError
1192 # zero-length sequences of all types will get here and succeed,
1193 # but that's a minor nit - since the original implementation
1194 # allowed empty dicts that type of behavior probably should be
1195 # preserved for consistency
1196 except TypeError:
1197 ty,va,tb = sys.exc_info()
Collin Winter828f04a2007-08-31 00:04:24 +00001198 raise TypeError("not a valid non-string sequence or mapping object").with_traceback(tb)
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001199
Guido van Rossume7b146f2000-02-04 15:28:42 +00001200 l = []
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001201 if not doseq:
1202 # preserve old behavior
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001203 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001204 k = quote_plus(str(k))
1205 v = quote_plus(str(v))
1206 l.append(k + '=' + v)
1207 else:
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001208 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001209 k = quote_plus(str(k))
Walter Dörwald65230a22002-06-03 15:58:32 +00001210 if isinstance(v, str):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001211 v = quote_plus(v)
1212 l.append(k + '=' + v)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001213 elif isinstance(v, str):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001214 # is there a reasonable way to convert to ASCII?
1215 # encode generates a string, but "replace" or "ignore"
1216 # lose information and "strict" can raise UnicodeError
1217 v = quote_plus(v.encode("ASCII","replace"))
1218 l.append(k + '=' + v)
1219 else:
1220 try:
1221 # is this a sufficient test for sequence-ness?
1222 x = len(v)
1223 except TypeError:
1224 # not a sequence
1225 v = quote_plus(str(v))
1226 l.append(k + '=' + v)
1227 else:
1228 # loop over the sequence
1229 for elt in v:
1230 l.append(k + '=' + quote_plus(str(elt)))
Guido van Rossumb2493f82000-12-15 15:01:37 +00001231 return '&'.join(l)
Guido van Rossum810a3391998-07-22 21:33:23 +00001232
Guido van Rossum442e7201996-03-20 15:33:11 +00001233# Proxy handling
Mark Hammond4f570b92000-07-26 07:04:38 +00001234def getproxies_environment():
1235 """Return a dictionary of scheme -> proxy server URL mappings.
1236
1237 Scan the environment for variables named <scheme>_proxy;
1238 this seems to be the standard convention. If you need a
1239 different way, you can pass a proxies dictionary to the
1240 [Fancy]URLopener constructor.
1241
1242 """
1243 proxies = {}
1244 for name, value in os.environ.items():
Guido van Rossumb2493f82000-12-15 15:01:37 +00001245 name = name.lower()
Christian Heimes9bd667a2008-01-20 15:14:11 +00001246 if name == 'no_proxy':
1247 # handled in proxy_bypass_environment
1248 continue
Mark Hammond4f570b92000-07-26 07:04:38 +00001249 if value and name[-6:] == '_proxy':
1250 proxies[name[:-6]] = value
1251 return proxies
1252
Christian Heimes9bd667a2008-01-20 15:14:11 +00001253def proxy_bypass_environment(host):
1254 """Test if proxies should not be used for a particular host.
1255
1256 Checks the environment for a variable named no_proxy, which should
1257 be a list of DNS suffixes separated by commas, or '*' for all hosts.
1258 """
1259 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
1260 # '*' is special case for always bypass
1261 if no_proxy == '*':
1262 return 1
1263 # strip port off host
1264 hostonly, port = splitport(host)
1265 # check if the host ends with any of the DNS suffixes
1266 for name in no_proxy.split(','):
1267 if name and (hostonly.endswith(name) or host.endswith(name)):
1268 return 1
1269 # otherwise, don't bypass
1270 return 0
1271
1272
Jack Jansen11d9b062004-07-16 11:45:00 +00001273if sys.platform == 'darwin':
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001274
1275 def _CFSetup(sc):
1276 from ctypes import c_int32, c_void_p, c_char_p, c_int
1277 sc.CFStringCreateWithCString.argtypes = [ c_void_p, c_char_p, c_int32 ]
1278 sc.CFStringCreateWithCString.restype = c_void_p
1279 sc.SCDynamicStoreCopyProxies.argtypes = [ c_void_p ]
1280 sc.SCDynamicStoreCopyProxies.restype = c_void_p
1281 sc.CFDictionaryGetValue.argtypes = [ c_void_p, c_void_p ]
1282 sc.CFDictionaryGetValue.restype = c_void_p
1283 sc.CFStringGetLength.argtypes = [ c_void_p ]
1284 sc.CFStringGetLength.restype = c_int32
1285 sc.CFStringGetCString.argtypes = [ c_void_p, c_char_p, c_int32, c_int32 ]
1286 sc.CFStringGetCString.restype = c_int32
1287 sc.CFNumberGetValue.argtypes = [ c_void_p, c_int, c_void_p ]
1288 sc.CFNumberGetValue.restype = c_int32
1289 sc.CFRelease.argtypes = [ c_void_p ]
1290 sc.CFRelease.restype = None
1291
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001292 def _CStringFromCFString(sc, value):
1293 from ctypes import create_string_buffer
1294 length = sc.CFStringGetLength(value) + 1
1295 buff = create_string_buffer(length)
1296 sc.CFStringGetCString(value, buff, length, 0)
1297 return buff.value
1298
1299 def _CFNumberToInt32(sc, cfnum):
1300 from ctypes import byref, c_int
1301 val = c_int()
1302 kCFNumberSInt32Type = 3
1303 sc.CFNumberGetValue(cfnum, kCFNumberSInt32Type, byref(val))
1304 return val.value
1305
1306
1307 def proxy_bypass_macosx_sysconf(host):
1308 """
1309 Return True iff this host shouldn't be accessed using a proxy
1310
1311 This function uses the MacOSX framework SystemConfiguration
1312 to fetch the proxy information.
1313 """
1314 from ctypes import cdll
1315 from ctypes.util import find_library
1316 import re
1317 import socket
1318 from fnmatch import fnmatch
1319
1320 def ip2num(ipAddr):
1321 parts = ipAddr.split('.')
1322 parts = map(int, parts)
1323 if len(parts) != 4:
1324 parts = (parts + [0, 0, 0, 0])[:4]
1325 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
1326
1327 sc = cdll.LoadLibrary(find_library("SystemConfiguration"))
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001328 _CFSetup(sc)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001329
1330 hostIP = None
1331
1332 if not sc:
1333 return False
1334
1335 kSCPropNetProxiesExceptionsList = sc.CFStringCreateWithCString(0, "ExceptionsList", 0)
1336 kSCPropNetProxiesExcludeSimpleHostnames = sc.CFStringCreateWithCString(0,
1337 "ExcludeSimpleHostnames", 0)
1338
1339
1340 proxyDict = sc.SCDynamicStoreCopyProxies(None)
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001341 if proxyDict is None:
1342 return False
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001343
1344 try:
1345 # Check for simple host names:
1346 if '.' not in host:
1347 exclude_simple = sc.CFDictionaryGetValue(proxyDict,
1348 kSCPropNetProxiesExcludeSimpleHostnames)
1349 if exclude_simple and _CFNumberToInt32(sc, exclude_simple):
1350 return True
1351
1352
1353 # Check the exceptions list:
1354 exceptions = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesExceptionsList)
1355 if exceptions:
1356 # Items in the list are strings like these: *.local, 169.254/16
1357 for index in xrange(sc.CFArrayGetCount(exceptions)):
1358 value = sc.CFArrayGetValueAtIndex(exceptions, index)
1359 if not value: continue
1360 value = _CStringFromCFString(sc, value)
1361
1362 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
1363 if m is not None:
1364 if hostIP is None:
1365 hostIP = socket.gethostbyname(host)
1366 hostIP = ip2num(hostIP)
1367
1368 base = ip2num(m.group(1))
1369 mask = int(m.group(2)[1:])
1370 mask = 32 - mask
1371
1372 if (hostIP >> mask) == (base >> mask):
1373 return True
1374
1375 elif fnmatch(host, value):
1376 return True
1377
1378 return False
1379
1380 finally:
1381 sc.CFRelease(kSCPropNetProxiesExceptionsList)
1382 sc.CFRelease(kSCPropNetProxiesExcludeSimpleHostnames)
1383
1384
1385
1386 def getproxies_macosx_sysconf():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001387 """Return a dictionary of scheme -> proxy server URL mappings.
Guido van Rossum442e7201996-03-20 15:33:11 +00001388
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001389 This function uses the MacOSX framework SystemConfiguration
1390 to fetch the proxy information.
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001391 """
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001392 from ctypes import cdll
1393 from ctypes.util import find_library
1394
1395 sc = cdll.LoadLibrary(find_library("SystemConfiguration"))
Benjamin Petersona37cfc62008-05-26 13:48:34 +00001396 _CFSetup(sc)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001397
1398 if not sc:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001399 return {}
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001400
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001401 kSCPropNetProxiesHTTPEnable = sc.CFStringCreateWithCString(0, b"HTTPEnable", 0)
1402 kSCPropNetProxiesHTTPProxy = sc.CFStringCreateWithCString(0, b"HTTPProxy", 0)
1403 kSCPropNetProxiesHTTPPort = sc.CFStringCreateWithCString(0, b"HTTPPort", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001404
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001405 kSCPropNetProxiesHTTPSEnable = sc.CFStringCreateWithCString(0, b"HTTPSEnable", 0)
1406 kSCPropNetProxiesHTTPSProxy = sc.CFStringCreateWithCString(0, b"HTTPSProxy", 0)
1407 kSCPropNetProxiesHTTPSPort = sc.CFStringCreateWithCString(0, b"HTTPSPort", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001408
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001409 kSCPropNetProxiesFTPEnable = sc.CFStringCreateWithCString(0, b"FTPEnable", 0)
1410 kSCPropNetProxiesFTPPassive = sc.CFStringCreateWithCString(0, b"FTPPassive", 0)
1411 kSCPropNetProxiesFTPPort = sc.CFStringCreateWithCString(0, b"FTPPort", 0)
1412 kSCPropNetProxiesFTPProxy = sc.CFStringCreateWithCString(0, b"FTPProxy", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001413
Georg Brandlc4dc0d42008-05-18 11:12:40 +00001414 kSCPropNetProxiesGopherEnable = sc.CFStringCreateWithCString(0, b"GopherEnable", 0)
1415 kSCPropNetProxiesGopherPort = sc.CFStringCreateWithCString(0, b"GopherPort", 0)
1416 kSCPropNetProxiesGopherProxy = sc.CFStringCreateWithCString(0, b"GopherProxy", 0)
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001417
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001418 proxies = {}
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001419 proxyDict = sc.SCDynamicStoreCopyProxies(None)
1420
1421 try:
1422 # HTTP:
1423 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPEnable)
1424 if enabled and _CFNumberToInt32(sc, enabled):
1425 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPProxy)
1426 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPPort)
1427
1428 if proxy:
1429 proxy = _CStringFromCFString(sc, proxy)
1430 if port:
1431 port = _CFNumberToInt32(sc, port)
1432 proxies["http"] = "http://%s:%i" % (proxy, port)
1433 else:
1434 proxies["http"] = "http://%s" % (proxy, )
1435
1436 # HTTPS:
1437 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSEnable)
1438 if enabled and _CFNumberToInt32(sc, enabled):
1439 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSProxy)
1440 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesHTTPSPort)
1441
1442 if proxy:
1443 proxy = _CStringFromCFString(sc, proxy)
1444 if port:
1445 port = _CFNumberToInt32(sc, port)
1446 proxies["https"] = "http://%s:%i" % (proxy, port)
1447 else:
1448 proxies["https"] = "http://%s" % (proxy, )
1449
1450 # FTP:
1451 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPEnable)
1452 if enabled and _CFNumberToInt32(sc, enabled):
1453 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPProxy)
1454 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesFTPPort)
1455
1456 if proxy:
1457 proxy = _CStringFromCFString(sc, proxy)
1458 if port:
1459 port = _CFNumberToInt32(sc, port)
1460 proxies["ftp"] = "http://%s:%i" % (proxy, port)
1461 else:
1462 proxies["ftp"] = "http://%s" % (proxy, )
1463
1464 # Gopher:
1465 enabled = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherEnable)
1466 if enabled and _CFNumberToInt32(sc, enabled):
1467 proxy = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherProxy)
1468 port = sc.CFDictionaryGetValue(proxyDict, kSCPropNetProxiesGopherPort)
1469
1470 if proxy:
1471 proxy = _CStringFromCFString(sc, proxy)
1472 if port:
1473 port = _CFNumberToInt32(sc, port)
1474 proxies["gopher"] = "http://%s:%i" % (proxy, port)
1475 else:
1476 proxies["gopher"] = "http://%s" % (proxy, )
1477 finally:
1478 sc.CFRelease(proxyDict)
1479
1480 sc.CFRelease(kSCPropNetProxiesHTTPEnable)
1481 sc.CFRelease(kSCPropNetProxiesHTTPProxy)
1482 sc.CFRelease(kSCPropNetProxiesHTTPPort)
1483 sc.CFRelease(kSCPropNetProxiesFTPEnable)
1484 sc.CFRelease(kSCPropNetProxiesFTPPassive)
1485 sc.CFRelease(kSCPropNetProxiesFTPPort)
1486 sc.CFRelease(kSCPropNetProxiesFTPProxy)
1487 sc.CFRelease(kSCPropNetProxiesGopherEnable)
1488 sc.CFRelease(kSCPropNetProxiesGopherPort)
1489 sc.CFRelease(kSCPropNetProxiesGopherProxy)
1490
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001491 return proxies
Mark Hammond4f570b92000-07-26 07:04:38 +00001492
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001493
1494
Christian Heimes9bd667a2008-01-20 15:14:11 +00001495 def proxy_bypass(host):
1496 if getproxies_environment():
1497 return proxy_bypass_environment(host)
1498 else:
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001499 return proxy_bypass_macosx_sysconf(host)
Tim Peters55c12d42001-08-09 18:04:14 +00001500
Jack Jansen11d9b062004-07-16 11:45:00 +00001501 def getproxies():
Alexandre Vassalottieca20b62008-05-16 02:54:33 +00001502 return getproxies_environment() or getproxies_macosx_sysconf()
Tim Peters182b5ac2004-07-18 06:16:08 +00001503
Mark Hammond4f570b92000-07-26 07:04:38 +00001504elif os.name == 'nt':
1505 def getproxies_registry():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001506 """Return a dictionary of scheme -> proxy server URL mappings.
Mark Hammond4f570b92000-07-26 07:04:38 +00001507
1508 Win32 uses the registry to store proxies.
1509
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001510 """
1511 proxies = {}
Mark Hammond4f570b92000-07-26 07:04:38 +00001512 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001513 import winreg
Mark Hammond4f570b92000-07-26 07:04:38 +00001514 except ImportError:
1515 # Std module, so should be around - but you never know!
1516 return proxies
1517 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001518 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001519 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl38feaf02008-05-25 07:45:51 +00001520 proxyEnable = winreg.QueryValueEx(internetSettings,
1521 'ProxyEnable')[0]
Mark Hammond4f570b92000-07-26 07:04:38 +00001522 if proxyEnable:
1523 # Returned as Unicode but problems if not converted to ASCII
Georg Brandl38feaf02008-05-25 07:45:51 +00001524 proxyServer = str(winreg.QueryValueEx(internetSettings,
1525 'ProxyServer')[0])
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001526 if '=' in proxyServer:
1527 # Per-protocol settings
Mark Hammond4f570b92000-07-26 07:04:38 +00001528 for p in proxyServer.split(';'):
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001529 protocol, address = p.split('=', 1)
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001530 # See if address has a type:// prefix
Guido van Rossum64e5aa92002-04-02 14:38:16 +00001531 import re
1532 if not re.match('^([^/:]+)://', address):
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001533 address = '%s://%s' % (protocol, address)
1534 proxies[protocol] = address
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001535 else:
1536 # Use one setting for all protocols
1537 if proxyServer[:5] == 'http:':
1538 proxies['http'] = proxyServer
1539 else:
1540 proxies['http'] = 'http://%s' % proxyServer
1541 proxies['ftp'] = 'ftp://%s' % proxyServer
Mark Hammond4f570b92000-07-26 07:04:38 +00001542 internetSettings.Close()
1543 except (WindowsError, ValueError, TypeError):
1544 # Either registry key not found etc, or the value in an
1545 # unexpected format.
1546 # proxies already set up to be empty so nothing to do
1547 pass
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001548 return proxies
Guido van Rossum442e7201996-03-20 15:33:11 +00001549
Mark Hammond4f570b92000-07-26 07:04:38 +00001550 def getproxies():
1551 """Return a dictionary of scheme -> proxy server URL mappings.
1552
1553 Returns settings gathered from the environment, if specified,
1554 or the registry.
1555
1556 """
1557 return getproxies_environment() or getproxies_registry()
Tim Peters55c12d42001-08-09 18:04:14 +00001558
Christian Heimes9bd667a2008-01-20 15:14:11 +00001559 def proxy_bypass_registry(host):
Tim Peters55c12d42001-08-09 18:04:14 +00001560 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001561 import winreg
Tim Peters55c12d42001-08-09 18:04:14 +00001562 import re
Tim Peters55c12d42001-08-09 18:04:14 +00001563 except ImportError:
1564 # Std modules, so should be around - but you never know!
1565 return 0
1566 try:
Georg Brandl38feaf02008-05-25 07:45:51 +00001567 internetSettings = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
Tim Peters55c12d42001-08-09 18:04:14 +00001568 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Georg Brandl38feaf02008-05-25 07:45:51 +00001569 proxyEnable = winreg.QueryValueEx(internetSettings,
1570 'ProxyEnable')[0]
1571 proxyOverride = str(winreg.QueryValueEx(internetSettings,
1572 'ProxyOverride')[0])
Tim Peters55c12d42001-08-09 18:04:14 +00001573 # ^^^^ Returned as Unicode but problems if not converted to ASCII
1574 except WindowsError:
1575 return 0
1576 if not proxyEnable or not proxyOverride:
1577 return 0
1578 # try to make a host list from name and IP address.
Georg Brandl1f636702006-02-18 23:10:23 +00001579 rawHost, port = splitport(host)
1580 host = [rawHost]
Tim Peters55c12d42001-08-09 18:04:14 +00001581 try:
Georg Brandl1f636702006-02-18 23:10:23 +00001582 addr = socket.gethostbyname(rawHost)
1583 if addr != rawHost:
Tim Peters55c12d42001-08-09 18:04:14 +00001584 host.append(addr)
1585 except socket.error:
1586 pass
Georg Brandl1f636702006-02-18 23:10:23 +00001587 try:
1588 fqdn = socket.getfqdn(rawHost)
1589 if fqdn != rawHost:
1590 host.append(fqdn)
1591 except socket.error:
1592 pass
Tim Peters55c12d42001-08-09 18:04:14 +00001593 # make a check value list from the registry entry: replace the
1594 # '<local>' string by the localhost entry and the corresponding
1595 # canonical entry.
1596 proxyOverride = proxyOverride.split(';')
1597 i = 0
1598 while i < len(proxyOverride):
1599 if proxyOverride[i] == '<local>':
1600 proxyOverride[i:i+1] = ['localhost',
1601 '127.0.0.1',
1602 socket.gethostname(),
1603 socket.gethostbyname(
1604 socket.gethostname())]
1605 i += 1
1606 # print proxyOverride
1607 # now check if we match one of the registry values.
1608 for test in proxyOverride:
Tim Petersab9ba272001-08-09 21:40:30 +00001609 test = test.replace(".", r"\.") # mask dots
1610 test = test.replace("*", r".*") # change glob sequence
1611 test = test.replace("?", r".") # change glob char
Tim Peters55c12d42001-08-09 18:04:14 +00001612 for val in host:
1613 # print "%s <--> %s" %( test, val )
1614 if re.match(test, val, re.I):
1615 return 1
1616 return 0
1617
Christian Heimes9bd667a2008-01-20 15:14:11 +00001618 def proxy_bypass(host):
1619 """Return a dictionary of scheme -> proxy server URL mappings.
1620
1621 Returns settings gathered from the environment, if specified,
1622 or the registry.
1623
1624 """
1625 if getproxies_environment():
1626 return proxy_bypass_environment(host)
1627 else:
1628 return proxy_bypass_registry(host)
1629
Mark Hammond4f570b92000-07-26 07:04:38 +00001630else:
1631 # By default use environment variables
1632 getproxies = getproxies_environment
Christian Heimes9bd667a2008-01-20 15:14:11 +00001633 proxy_bypass = proxy_bypass_environment
Guido van Rossum442e7201996-03-20 15:33:11 +00001634
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001635# Test and time quote() and unquote()
1636def test1():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001637 s = ''
1638 for i in range(256): s = s + chr(i)
1639 s = s*4
1640 t0 = time.time()
1641 qs = quote(s)
1642 uqs = unquote(qs)
1643 t1 = time.time()
1644 if uqs != s:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001645 print('Wrong!')
1646 print(repr(s))
1647 print(repr(qs))
1648 print(repr(uqs))
1649 print(round(t1 - t0, 3), 'sec')
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001650
1651
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001652def reporthook(blocknum, blocksize, totalsize):
1653 # Report during remote transfers
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001654 print("Block number: %d, Block size: %d, Total size: %d" % (
1655 blocknum, blocksize, totalsize))
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001656
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001657# Test program
Guido van Rossum23490151998-06-25 02:39:00 +00001658def test(args=[]):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001659 if not args:
1660 args = [
1661 '/etc/passwd',
1662 'file:/etc/passwd',
1663 'file://localhost/etc/passwd',
Guido van Rossumd8faa362007-04-27 19:54:29 +00001664 'ftp://ftp.gnu.org/pub/README',
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001665 'http://www.python.org/index.html',
1666 ]
Guido van Rossum09c8b6c1999-12-07 21:37:17 +00001667 if hasattr(URLopener, "open_https"):
1668 args.append('https://synergy.as.cmu.edu/~geek/')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001669 try:
1670 for url in args:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001671 print('-'*10, url, '-'*10)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001672 fn, h = urlretrieve(url, None, reporthook)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001673 print(fn)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001674 if h:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001675 print('======')
1676 for k in h.keys(): print(k + ':', h[k])
1677 print('======')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001678 fp = open(fn, 'rb')
1679 data = fp.read()
1680 del fp
Georg Brandl7f13e6b2007-08-31 10:37:15 +00001681 data = data.replace("\r", "")
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001682 print(data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001683 fn, h = None, None
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001684 print('-'*40)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001685 finally:
1686 urlcleanup()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001687
Guido van Rossum23490151998-06-25 02:39:00 +00001688def main():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001689 import getopt, sys
1690 try:
1691 opts, args = getopt.getopt(sys.argv[1:], "th")
Guido van Rossumb940e112007-01-10 16:19:56 +00001692 except getopt.error as msg:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001693 print(msg)
1694 print("Use -h for help")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001695 return
1696 t = 0
1697 for o, a in opts:
1698 if o == '-t':
1699 t = t + 1
1700 if o == '-h':
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001701 print("Usage: python urllib.py [-t] [url ...]")
1702 print("-t runs self-test;", end=' ')
1703 print("otherwise, contents of urls are printed")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001704 return
1705 if t:
1706 if t > 1:
1707 test1()
1708 test(args)
1709 else:
1710 if not args:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001711 print("Use -h for help")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001712 for url in args:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001713 print(urlopen(url).read(), end=' ')
Guido van Rossum23490151998-06-25 02:39:00 +00001714
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001715# Run test program when run as a script
1716if __name__ == '__main__':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001717 main()