blob: 707e2f0c65f07345efb0680006574de6c135a4c5 [file] [log] [blame]
Guido van Rossume7b146f2000-02-04 15:28:42 +00001"""Open an arbitrary URL.
2
3See the following document for more info on URLs:
4"Names and Addresses, URIs, URLs, URNs, URCs", at
5http://www.w3.org/pub/WWW/Addressing/Overview.html
6
7See also the HTTP spec (from which the error codes are derived):
8"HTTP - Hypertext Transfer Protocol", at
9http://www.w3.org/pub/WWW/Protocols/
10
11Related standards and specs:
12- RFC1808: the "relative URL" spec. (authoritative status)
13- RFC1738 - the "URL standard". (authoritative status)
14- RFC1630 - the "URI spec". (informational status)
15
16The object returned by URLopener().open(file) will differ per
17protocol. All you know is that is has methods read(), readline(),
18readlines(), fileno(), close() and info(). The read*(), fileno()
Fredrik Lundhb49f88b2000-09-24 18:51:25 +000019and close() methods work like those of open files.
Guido van Rossume7b146f2000-02-04 15:28:42 +000020The info() method returns a mimetools.Message object which can be
21used to query various info about the object, if available.
22(mimetools.Message objects are queried with the getheader() method.)
23"""
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000024
Guido van Rossum7c395db1994-07-04 22:14:49 +000025import string
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000026import socket
Jack Jansendc3e3f61995-12-15 13:22:13 +000027import os
Guido van Rossumf0713d32001-08-09 17:43:35 +000028import time
Guido van Rossum3c8484e1996-11-20 22:02:24 +000029import sys
Brett Cannon69200fa2004-03-23 21:26:39 +000030from urlparse import urljoin as basejoin
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000031
Skip Montanaro40fc1602001-03-01 04:27:19 +000032__all__ = ["urlopen", "URLopener", "FancyURLopener", "urlretrieve",
33 "urlcleanup", "quote", "quote_plus", "unquote", "unquote_plus",
Skip Montanaro44d5e0c2001-03-13 19:47:16 +000034 "urlencode", "url2pathname", "pathname2url", "splittag",
35 "localhost", "thishost", "ftperrors", "basejoin", "unwrap",
36 "splittype", "splithost", "splituser", "splitpasswd", "splitport",
37 "splitnport", "splitquery", "splitattr", "splitvalue",
Brett Cannond75f0432007-05-16 22:42:29 +000038 "getproxies"]
Skip Montanaro40fc1602001-03-01 04:27:19 +000039
Martin v. Löwis3e865952006-01-24 15:51:21 +000040__version__ = '1.17' # XXX This version is not always updated :-(
Guido van Rossumf668d171997-06-06 21:11:11 +000041
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000042MAXFTPCACHE = 10 # Trim the ftp cache beyond this size
Guido van Rossum6cb15a01995-06-22 19:00:13 +000043
Jack Jansendc3e3f61995-12-15 13:22:13 +000044# Helper for non-unix systems
Ronald Oussoren9545a232010-05-05 19:09:31 +000045if os.name == 'nt':
Fredrik Lundhb49f88b2000-09-24 18:51:25 +000046 from nturl2path import url2pathname, pathname2url
Guido van Rossumd74fb6b2001-03-02 06:43:49 +000047elif os.name == 'riscos':
48 from rourl2path 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):
Brett Cannon8bb8fa52008-07-02 01:57:08 +000070 """Create a file-like object for the specified URL to read from."""
71 from warnings import warnpy3k
Georg Brandl48e65f52010-02-06 22:44:17 +000072 warnpy3k("urllib.urlopen() has been removed in Python 3.0 in "
73 "favor of urllib2.urlopen()", stacklevel=2)
Brett Cannon8bb8fa52008-07-02 01:57:08 +000074
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000075 global _urlopener
Fred Drakedf6eca72002-04-04 20:41:34 +000076 if proxies is not None:
77 opener = FancyURLopener(proxies=proxies)
78 elif not _urlopener:
79 opener = FancyURLopener()
80 _urlopener = opener
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000081 else:
Fred Drakedf6eca72002-04-04 20:41:34 +000082 opener = _urlopener
83 if data is None:
84 return opener.open(url)
85 else:
86 return opener.open(url, data)
Fred Drake316a7932000-08-24 01:01:26 +000087def urlretrieve(url, filename=None, reporthook=None, data=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000088 global _urlopener
89 if not _urlopener:
90 _urlopener = FancyURLopener()
Fred Drake316a7932000-08-24 01:01:26 +000091 return _urlopener.retrieve(url, filename, reporthook, data)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000092def urlcleanup():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +000093 if _urlopener:
94 _urlopener.cleanup()
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +000095 _safe_quoters.clear()
Antoine Pitrouca173e22009-12-08 19:35:12 +000096 ftpcache.clear()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +000097
Bill Janssen426ea0a2007-08-29 22:35:05 +000098# check for SSL
99try:
100 import ssl
101except:
102 _have_ssl = False
103else:
104 _have_ssl = True
105
Georg Brandlb9256022005-08-24 18:46:39 +0000106# exception raised when downloaded size does not match content-length
107class ContentTooShortError(IOError):
108 def __init__(self, message, content):
109 IOError.__init__(self, message)
110 self.content = content
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000111
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000112ftpcache = {}
113class URLopener:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000114 """Class to open URLs.
115 This is a class rather than just a subroutine because we may need
116 more than one set of global protocol-specific options.
117 Note -- this is a base class for those who don't want the
118 automatic handling of errors type 302 (relocated) and 401
119 (authorization needed)."""
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000120
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000121 __tempfiles = None
Guido van Rossum29e77811996-11-27 19:39:58 +0000122
Guido van Rossumba311382000-08-24 16:18:04 +0000123 version = "Python-urllib/%s" % __version__
124
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000125 # Constructor
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000126 def __init__(self, proxies=None, **x509):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000127 if proxies is None:
128 proxies = getproxies()
129 assert hasattr(proxies, 'has_key'), "proxies must be a mapping"
130 self.proxies = proxies
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000131 self.key_file = x509.get('key_file')
132 self.cert_file = x509.get('cert_file')
Georg Brandl0619a322006-07-26 07:40:17 +0000133 self.addheaders = [('User-Agent', self.version)]
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000134 self.__tempfiles = []
135 self.__unlink = os.unlink # See cleanup()
136 self.tempcache = None
137 # Undocumented feature: if you assign {} to tempcache,
138 # it is used to cache files retrieved with
139 # self.retrieve(). This is not enabled by default
140 # since it does not work for changing documents (and I
141 # haven't got the logic to check expiration headers
142 # yet).
143 self.ftpcache = ftpcache
144 # Undocumented feature: you can use a different
145 # ftp cache by assigning to the .ftpcache member;
146 # in case you want logically independent URL openers
147 # XXX This is not threadsafe. Bah.
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000148
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000149 def __del__(self):
150 self.close()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000151
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000152 def close(self):
153 self.cleanup()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000154
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000155 def cleanup(self):
156 # This code sometimes runs when the rest of this module
157 # has already been deleted, so it can't use any globals
158 # or import anything.
159 if self.__tempfiles:
160 for file in self.__tempfiles:
161 try:
162 self.__unlink(file)
Martin v. Löwis58682b72001-08-11 15:02:57 +0000163 except OSError:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000164 pass
165 del self.__tempfiles[:]
166 if self.tempcache:
167 self.tempcache.clear()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000168
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000169 def addheader(self, *args):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000170 """Add a header to be used by the HTTP interface only
171 e.g. u.addheader('Accept', 'sound/basic')"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000172 self.addheaders.append(args)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000173
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000174 # External interface
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000175 def open(self, fullurl, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000176 """Use URLopener().open(file) instead of open(file, 'r')."""
Martin v. Löwis1d994332000-12-03 18:30:10 +0000177 fullurl = unwrap(toBytes(fullurl))
Senthil Kumaran7c2867f2009-04-21 03:24:19 +0000178 # percent encode url, fixing lame server errors for e.g, like space
179 # within url paths.
Senthil Kumaran18d5a692010-02-20 22:05:34 +0000180 fullurl = quote(fullurl, safe="%/:=&?~#+!$,;'@()*[]|")
Raymond Hettinger54f02222002-06-01 14:18:47 +0000181 if self.tempcache and fullurl in self.tempcache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000182 filename, headers = self.tempcache[fullurl]
183 fp = open(filename, 'rb')
184 return addinfourl(fp, headers, fullurl)
Martin v. Löwis1d994332000-12-03 18:30:10 +0000185 urltype, url = splittype(fullurl)
186 if not urltype:
187 urltype = 'file'
Raymond Hettinger54f02222002-06-01 14:18:47 +0000188 if urltype in self.proxies:
Martin v. Löwis1d994332000-12-03 18:30:10 +0000189 proxy = self.proxies[urltype]
190 urltype, proxyhost = splittype(proxy)
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000191 host, selector = splithost(proxyhost)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000192 url = (host, fullurl) # Signal special case to open_*()
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000193 else:
194 proxy = None
Martin v. Löwis1d994332000-12-03 18:30:10 +0000195 name = 'open_' + urltype
196 self.type = urltype
Brett Cannonaaeffaf2004-03-23 23:50:17 +0000197 name = name.replace('-', '_')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000198 if not hasattr(self, name):
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000199 if proxy:
200 return self.open_unknown_proxy(proxy, fullurl, data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000201 else:
202 return self.open_unknown(fullurl, data)
203 try:
204 if data is None:
205 return getattr(self, name)(url)
206 else:
207 return getattr(self, name)(url, data)
208 except socket.error, msg:
209 raise IOError, ('socket error', msg), sys.exc_info()[2]
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000210
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000211 def open_unknown(self, fullurl, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000212 """Overridable interface to open unknown URL type."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000213 type, url = splittype(fullurl)
214 raise IOError, ('url error', 'unknown url type', type)
Guido van Rossumca445401995-08-29 19:19:12 +0000215
Jeremy Hyltond52755f2000-10-02 23:04:02 +0000216 def open_unknown_proxy(self, proxy, fullurl, data=None):
217 """Overridable interface to open unknown URL type."""
218 type, url = splittype(fullurl)
219 raise IOError, ('url error', 'invalid proxy for %s' % type, proxy)
220
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000221 # External interface
Sjoerd Mullenderd7b86f02000-08-25 11:23:36 +0000222 def retrieve(self, url, filename=None, reporthook=None, data=None):
Brett Cannon7d618c72003-04-24 02:43:20 +0000223 """retrieve(url) returns (filename, headers) for a local object
Guido van Rossume7b146f2000-02-04 15:28:42 +0000224 or (tempfilename, headers) for a remote object."""
Martin v. Löwis1d994332000-12-03 18:30:10 +0000225 url = unwrap(toBytes(url))
Raymond Hettinger54f02222002-06-01 14:18:47 +0000226 if self.tempcache and url in self.tempcache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000227 return self.tempcache[url]
228 type, url1 = splittype(url)
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000229 if filename is None and (not type or type == 'file'):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000230 try:
231 fp = self.open_local_file(url1)
232 hdrs = fp.info()
Philip Jenvey0299d0d2009-12-03 02:40:13 +0000233 fp.close()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000234 return url2pathname(splithost(url1)[1]), hdrs
Georg Brandl84fedf72010-02-06 22:59:15 +0000235 except IOError:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000236 pass
Fred Drake316a7932000-08-24 01:01:26 +0000237 fp = self.open(url, data)
Benjamin Petersonb364bfe2009-03-22 17:45:11 +0000238 try:
239 headers = fp.info()
240 if filename:
241 tfp = open(filename, 'wb')
242 else:
243 import tempfile
244 garbage, path = splittype(url)
245 garbage, path = splithost(path or "")
246 path, garbage = splitquery(path or "")
247 path, garbage = splitattr(path or "")
248 suffix = os.path.splitext(path)[1]
249 (fd, filename) = tempfile.mkstemp(suffix)
250 self.__tempfiles.append(filename)
251 tfp = os.fdopen(fd, 'wb')
252 try:
253 result = filename, headers
254 if self.tempcache is not None:
255 self.tempcache[url] = result
256 bs = 1024*8
257 size = -1
258 read = 0
259 blocknum = 0
260 if reporthook:
261 if "content-length" in headers:
262 size = int(headers["Content-Length"])
263 reporthook(blocknum, bs, size)
264 while 1:
265 block = fp.read(bs)
266 if block == "":
267 break
268 read += len(block)
269 tfp.write(block)
270 blocknum += 1
271 if reporthook:
272 reporthook(blocknum, bs, size)
273 finally:
274 tfp.close()
275 finally:
276 fp.close()
Georg Brandlb9256022005-08-24 18:46:39 +0000277
278 # raise exception if actual size does not match content-length header
279 if size >= 0 and read < size:
280 raise ContentTooShortError("retrieval incomplete: got only %i out "
281 "of %i bytes" % (read, size), result)
282
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000283 return result
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000284
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000285 # Each method named open_<type> knows how to open that type of URL
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000286
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000287 def open_http(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000288 """Use HTTP protocol."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000289 import httplib
290 user_passwd = None
Martin v. Löwis3e865952006-01-24 15:51:21 +0000291 proxy_passwd= None
Walter Dörwald65230a22002-06-03 15:58:32 +0000292 if isinstance(url, str):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000293 host, selector = splithost(url)
294 if host:
295 user_passwd, host = splituser(host)
296 host = unquote(host)
297 realhost = host
298 else:
299 host, selector = url
Martin v. Löwis3e865952006-01-24 15:51:21 +0000300 # check whether the proxy contains authorization information
301 proxy_passwd, host = splituser(host)
302 # now we proceed with the url we want to obtain
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000303 urltype, rest = splittype(selector)
304 url = rest
305 user_passwd = None
Guido van Rossumb2493f82000-12-15 15:01:37 +0000306 if urltype.lower() != 'http':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000307 realhost = None
308 else:
309 realhost, rest = splithost(rest)
310 if realhost:
311 user_passwd, realhost = splituser(realhost)
312 if user_passwd:
313 selector = "%s://%s%s" % (urltype, realhost, rest)
Tim Peters55c12d42001-08-09 18:04:14 +0000314 if proxy_bypass(realhost):
315 host = realhost
316
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000317 #print "proxy via http:", host, selector
318 if not host: raise IOError, ('http error', 'no host given')
Tim Peters92037a12006-01-24 22:44:08 +0000319
Martin v. Löwis3e865952006-01-24 15:51:21 +0000320 if proxy_passwd:
321 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000322 proxy_auth = base64.b64encode(proxy_passwd).strip()
Martin v. Löwis3e865952006-01-24 15:51:21 +0000323 else:
324 proxy_auth = None
325
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000326 if user_passwd:
327 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000328 auth = base64.b64encode(user_passwd).strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000329 else:
330 auth = None
331 h = httplib.HTTP(host)
332 if data is not None:
333 h.putrequest('POST', selector)
Georg Brandl0619a322006-07-26 07:40:17 +0000334 h.putheader('Content-Type', 'application/x-www-form-urlencoded')
335 h.putheader('Content-Length', '%d' % len(data))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000336 else:
337 h.putrequest('GET', selector)
Martin v. Löwis3e865952006-01-24 15:51:21 +0000338 if proxy_auth: h.putheader('Proxy-Authorization', 'Basic %s' % proxy_auth)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000339 if auth: h.putheader('Authorization', 'Basic %s' % auth)
340 if realhost: h.putheader('Host', realhost)
Guido van Rossum68468eb2003-02-27 20:14:51 +0000341 for args in self.addheaders: h.putheader(*args)
Kristján Valur Jónsson84040db2009-01-09 20:27:16 +0000342 h.endheaders(data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000343 errcode, errmsg, headers = h.getreply()
Neal Norwitzce55e212007-03-20 08:14:57 +0000344 fp = h.getfile()
Georg Brandlf66b6032007-03-14 08:27:52 +0000345 if errcode == -1:
Neal Norwitzce55e212007-03-20 08:14:57 +0000346 if fp: fp.close()
Georg Brandlf66b6032007-03-14 08:27:52 +0000347 # something went wrong with the HTTP status line
348 raise IOError, ('http protocol error', 0,
349 'got a bad status line', None)
Sean Reifscheidera1afbf62007-09-19 07:52:56 +0000350 # According to RFC 2616, "2xx" code indicates that the client's
351 # request was successfully received, understood, and accepted.
Kurt B. Kaiser0f7c25d2008-01-02 04:11:28 +0000352 if (200 <= errcode < 300):
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000353 return addinfourl(fp, headers, "http:" + url, errcode)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000354 else:
355 if data is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000356 return self.http_error(url, fp, errcode, errmsg, headers)
Guido van Rossum29aab751999-03-09 19:31:21 +0000357 else:
358 return self.http_error(url, fp, errcode, errmsg, headers, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000359
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000360 def http_error(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000361 """Handle http errors.
362 Derived class can override this, or provide specific handlers
363 named http_error_DDD where DDD is the 3-digit error code."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000364 # First check if there's a specific handler for this error
365 name = 'http_error_%d' % errcode
366 if hasattr(self, name):
367 method = getattr(self, name)
368 if data is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000369 result = method(url, fp, errcode, errmsg, headers)
Jeremy Hyltonb30f52a1999-02-25 16:14:58 +0000370 else:
371 result = method(url, fp, errcode, errmsg, headers, data)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000372 if result: return result
Jeremy Hyltonb30f52a1999-02-25 16:14:58 +0000373 return self.http_error_default(url, fp, errcode, errmsg, headers)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000374
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000375 def http_error_default(self, url, fp, errcode, errmsg, headers):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000376 """Default error handler: close the connection and raise IOError."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000377 void = fp.read()
378 fp.close()
379 raise IOError, ('http error', errcode, errmsg, headers)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000380
Bill Janssen426ea0a2007-08-29 22:35:05 +0000381 if _have_ssl:
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000382 def open_https(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000383 """Use HTTPS protocol."""
Bill Janssen426ea0a2007-08-29 22:35:05 +0000384
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000385 import httplib
Fred Drake567ca8e2000-08-21 21:42:42 +0000386 user_passwd = None
Martin v. Löwis3e865952006-01-24 15:51:21 +0000387 proxy_passwd = None
Walter Dörwald65230a22002-06-03 15:58:32 +0000388 if isinstance(url, str):
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000389 host, selector = splithost(url)
Fred Drake567ca8e2000-08-21 21:42:42 +0000390 if host:
391 user_passwd, host = splituser(host)
392 host = unquote(host)
393 realhost = host
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000394 else:
395 host, selector = url
Martin v. Löwis3e865952006-01-24 15:51:21 +0000396 # here, we determine, whether the proxy contains authorization information
397 proxy_passwd, host = splituser(host)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000398 urltype, rest = splittype(selector)
Fred Drake567ca8e2000-08-21 21:42:42 +0000399 url = rest
400 user_passwd = None
Guido van Rossumb2493f82000-12-15 15:01:37 +0000401 if urltype.lower() != 'https':
Fred Drake567ca8e2000-08-21 21:42:42 +0000402 realhost = None
403 else:
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000404 realhost, rest = splithost(rest)
Fred Drake567ca8e2000-08-21 21:42:42 +0000405 if realhost:
406 user_passwd, realhost = splituser(realhost)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000407 if user_passwd:
408 selector = "%s://%s%s" % (urltype, realhost, rest)
Andrew M. Kuchling7ad47922000-06-10 01:41:48 +0000409 #print "proxy via https:", host, selector
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000410 if not host: raise IOError, ('https error', 'no host given')
Martin v. Löwis3e865952006-01-24 15:51:21 +0000411 if proxy_passwd:
412 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000413 proxy_auth = base64.b64encode(proxy_passwd).strip()
Martin v. Löwis3e865952006-01-24 15:51:21 +0000414 else:
415 proxy_auth = None
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000416 if user_passwd:
417 import base64
Andrew M. Kuchling872dba42006-10-27 17:11:23 +0000418 auth = base64.b64encode(user_passwd).strip()
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000419 else:
420 auth = None
421 h = httplib.HTTPS(host, 0,
422 key_file=self.key_file,
423 cert_file=self.cert_file)
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000424 if data is not None:
425 h.putrequest('POST', selector)
Georg Brandl0619a322006-07-26 07:40:17 +0000426 h.putheader('Content-Type',
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000427 'application/x-www-form-urlencoded')
Georg Brandl0619a322006-07-26 07:40:17 +0000428 h.putheader('Content-Length', '%d' % len(data))
Andrew M. Kuchling141e9892000-04-23 02:53:11 +0000429 else:
430 h.putrequest('GET', selector)
Andrew M. Kuchling52278572006-12-19 15:11:41 +0000431 if proxy_auth: h.putheader('Proxy-Authorization', 'Basic %s' % proxy_auth)
432 if auth: h.putheader('Authorization', 'Basic %s' % auth)
Fred Drake567ca8e2000-08-21 21:42:42 +0000433 if realhost: h.putheader('Host', realhost)
Guido van Rossum68468eb2003-02-27 20:14:51 +0000434 for args in self.addheaders: h.putheader(*args)
Kristján Valur Jónsson84040db2009-01-09 20:27:16 +0000435 h.endheaders(data)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000436 errcode, errmsg, headers = h.getreply()
Neal Norwitzce55e212007-03-20 08:14:57 +0000437 fp = h.getfile()
Georg Brandlf66b6032007-03-14 08:27:52 +0000438 if errcode == -1:
Neal Norwitzce55e212007-03-20 08:14:57 +0000439 if fp: fp.close()
Georg Brandlf66b6032007-03-14 08:27:52 +0000440 # something went wrong with the HTTP status line
441 raise IOError, ('http protocol error', 0,
442 'got a bad status line', None)
Georg Brandl9b915672007-09-24 18:08:24 +0000443 # According to RFC 2616, "2xx" code indicates that the client's
444 # request was successfully received, understood, and accepted.
Kurt B. Kaiser0f7c25d2008-01-02 04:11:28 +0000445 if (200 <= errcode < 300):
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000446 return addinfourl(fp, headers, "https:" + url, errcode)
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000447 else:
Fred Drake567ca8e2000-08-21 21:42:42 +0000448 if data is None:
449 return self.http_error(url, fp, errcode, errmsg, headers)
450 else:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000451 return self.http_error(url, fp, errcode, errmsg, headers,
452 data)
Fred Drake567ca8e2000-08-21 21:42:42 +0000453
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000454 def open_file(self, url):
Neal Norwitzc5d0dbd2006-04-09 04:00:49 +0000455 """Use local file or FTP depending on form of URL."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000456 if not isinstance(url, str):
457 raise IOError, ('file error', 'proxy support for file protocol currently not implemented')
Jack Jansen4ef11032002-09-12 20:14:04 +0000458 if url[:2] == '//' and url[2:3] != '/' and url[2:12].lower() != 'localhost/':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000459 return self.open_ftp(url)
460 else:
461 return self.open_local_file(url)
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000462
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000463 def open_local_file(self, url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000464 """Use local file."""
Georg Brandl5a096e12007-01-22 19:40:21 +0000465 import mimetypes, mimetools, email.utils
Raymond Hettingera6172712004-12-31 19:15:26 +0000466 try:
467 from cStringIO import StringIO
468 except ImportError:
469 from StringIO import StringIO
Guido van Rossumf0713d32001-08-09 17:43:35 +0000470 host, file = splithost(url)
471 localname = url2pathname(file)
Guido van Rossuma2da3052002-04-15 00:25:01 +0000472 try:
473 stats = os.stat(localname)
474 except OSError, e:
475 raise IOError(e.errno, e.strerror, e.filename)
Walter Dörwald92b48b72002-03-22 17:30:38 +0000476 size = stats.st_size
Georg Brandl5a096e12007-01-22 19:40:21 +0000477 modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000478 mtype = mimetypes.guess_type(url)[0]
Raymond Hettingera6172712004-12-31 19:15:26 +0000479 headers = mimetools.Message(StringIO(
Guido van Rossumf0713d32001-08-09 17:43:35 +0000480 'Content-Type: %s\nContent-Length: %d\nLast-modified: %s\n' %
481 (mtype or 'text/plain', size, modified)))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000482 if not host:
Guido van Rossum336a2011999-06-24 15:27:36 +0000483 urlfile = file
484 if file[:1] == '/':
485 urlfile = 'file://' + file
Guido van Rossumf0713d32001-08-09 17:43:35 +0000486 return addinfourl(open(localname, 'rb'),
Guido van Rossum336a2011999-06-24 15:27:36 +0000487 headers, urlfile)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000488 host, port = splitport(host)
489 if not port \
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000490 and socket.gethostbyname(host) in (localhost(), thishost()):
Guido van Rossum336a2011999-06-24 15:27:36 +0000491 urlfile = file
492 if file[:1] == '/':
493 urlfile = 'file://' + file
Guido van Rossumf0713d32001-08-09 17:43:35 +0000494 return addinfourl(open(localname, 'rb'),
Guido van Rossum336a2011999-06-24 15:27:36 +0000495 headers, urlfile)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000496 raise IOError, ('local file error', 'not on local host')
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000497
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000498 def open_ftp(self, url):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000499 """Use FTP protocol."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000500 if not isinstance(url, str):
501 raise IOError, ('ftp error', 'proxy support for ftp protocol currently not implemented')
Raymond Hettingera6172712004-12-31 19:15:26 +0000502 import mimetypes, mimetools
503 try:
504 from cStringIO import StringIO
505 except ImportError:
506 from StringIO import StringIO
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000507 host, path = splithost(url)
508 if not host: raise IOError, ('ftp error', 'no host given')
509 host, port = splitport(host)
510 user, host = splituser(host)
511 if user: user, passwd = splitpasswd(user)
512 else: passwd = None
513 host = unquote(host)
514 user = unquote(user or '')
515 passwd = unquote(passwd or '')
516 host = socket.gethostbyname(host)
517 if not port:
518 import ftplib
519 port = ftplib.FTP_PORT
520 else:
521 port = int(port)
522 path, attrs = splitattr(path)
523 path = unquote(path)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000524 dirs = path.split('/')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000525 dirs, file = dirs[:-1], dirs[-1]
526 if dirs and not dirs[0]: dirs = dirs[1:]
Guido van Rossum5e006a31999-08-18 17:40:33 +0000527 if dirs and not dirs[0]: dirs[0] = '/'
Guido van Rossumb2493f82000-12-15 15:01:37 +0000528 key = user, host, port, '/'.join(dirs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000529 # XXX thread unsafe!
530 if len(self.ftpcache) > MAXFTPCACHE:
531 # Prune the cache, rather arbitrarily
532 for k in self.ftpcache.keys():
533 if k != key:
534 v = self.ftpcache[k]
535 del self.ftpcache[k]
536 v.close()
537 try:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000538 if not key in self.ftpcache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000539 self.ftpcache[key] = \
540 ftpwrapper(user, passwd, host, port, dirs)
541 if not file: type = 'D'
542 else: type = 'I'
543 for attr in attrs:
544 attr, value = splitvalue(attr)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000545 if attr.lower() == 'type' and \
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000546 value in ('a', 'A', 'i', 'I', 'd', 'D'):
Guido van Rossumb2493f82000-12-15 15:01:37 +0000547 type = value.upper()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000548 (fp, retrlen) = self.ftpcache[key].retrfile(file, type)
Guido van Rossum88e0b5b2001-08-23 13:38:15 +0000549 mtype = mimetypes.guess_type("ftp:" + url)[0]
550 headers = ""
551 if mtype:
552 headers += "Content-Type: %s\n" % mtype
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000553 if retrlen is not None and retrlen >= 0:
Guido van Rossum88e0b5b2001-08-23 13:38:15 +0000554 headers += "Content-Length: %d\n" % retrlen
Raymond Hettingera6172712004-12-31 19:15:26 +0000555 headers = mimetools.Message(StringIO(headers))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000556 return addinfourl(fp, headers, "ftp:" + url)
557 except ftperrors(), msg:
558 raise IOError, ('ftp error', msg), sys.exc_info()[2]
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000559
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000560 def open_data(self, url, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000561 """Use "data" URL."""
Martin v. Löwis3e865952006-01-24 15:51:21 +0000562 if not isinstance(url, str):
563 raise IOError, ('data error', 'proxy support for data protocol currently not implemented')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000564 # ignore POSTed data
565 #
566 # syntax of data URLs:
567 # dataurl := "data:" [ mediatype ] [ ";base64" ] "," data
568 # mediatype := [ type "/" subtype ] *( ";" parameter )
569 # data := *urlchar
570 # parameter := attribute "=" value
Raymond Hettingera6172712004-12-31 19:15:26 +0000571 import mimetools
572 try:
573 from cStringIO import StringIO
574 except ImportError:
575 from StringIO import StringIO
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000576 try:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000577 [type, data] = url.split(',', 1)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000578 except ValueError:
579 raise IOError, ('data error', 'bad data URL')
580 if not type:
581 type = 'text/plain;charset=US-ASCII'
Guido van Rossumb2493f82000-12-15 15:01:37 +0000582 semi = type.rfind(';')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000583 if semi >= 0 and '=' not in type[semi:]:
584 encoding = type[semi+1:]
585 type = type[:semi]
586 else:
587 encoding = ''
588 msg = []
Senthil Kumaran1b7f9e52010-05-01 08:01:56 +0000589 msg.append('Date: %s'%time.strftime('%a, %d %b %Y %H:%M:%S GMT',
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000590 time.gmtime(time.time())))
591 msg.append('Content-type: %s' % type)
592 if encoding == 'base64':
593 import base64
594 data = base64.decodestring(data)
595 else:
596 data = unquote(data)
Georg Brandl0619a322006-07-26 07:40:17 +0000597 msg.append('Content-Length: %d' % len(data))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000598 msg.append('')
599 msg.append(data)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000600 msg = '\n'.join(msg)
Raymond Hettingera6172712004-12-31 19:15:26 +0000601 f = StringIO(msg)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000602 headers = mimetools.Message(f, 0)
Georg Brandl1f663572005-11-26 16:50:44 +0000603 #f.fileno = None # needed for addinfourl
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000604 return addinfourl(f, headers, url)
Guido van Rossum6d4d1c21998-03-12 14:32:55 +0000605
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000606
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000607class FancyURLopener(URLopener):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000608 """Derived class with handlers for errors we can handle (perhaps)."""
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000609
Neal Norwitz60e04cd2002-06-11 13:38:51 +0000610 def __init__(self, *args, **kwargs):
Guido van Rossum68468eb2003-02-27 20:14:51 +0000611 URLopener.__init__(self, *args, **kwargs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000612 self.auth_cache = {}
Skip Montanaroc3e11d62001-02-15 16:56:36 +0000613 self.tries = 0
614 self.maxtries = 10
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000615
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000616 def http_error_default(self, url, fp, errcode, errmsg, headers):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000617 """Default error handling -- don't raise an exception."""
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000618 return addinfourl(fp, headers, "http:" + url, errcode)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000619
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000620 def http_error_302(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000621 """Error 302 -- relocated (temporarily)."""
Skip Montanaroc3e11d62001-02-15 16:56:36 +0000622 self.tries += 1
623 if self.maxtries and self.tries >= self.maxtries:
624 if hasattr(self, "http_error_500"):
625 meth = self.http_error_500
626 else:
627 meth = self.http_error_default
628 self.tries = 0
629 return meth(url, fp, 500,
630 "Internal Server Error: Redirect Recursion", headers)
631 result = self.redirect_internal(url, fp, errcode, errmsg, headers,
632 data)
633 self.tries = 0
634 return result
635
636 def redirect_internal(self, url, fp, errcode, errmsg, headers, data):
Raymond Hettinger54f02222002-06-01 14:18:47 +0000637 if 'location' in headers:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000638 newurl = headers['location']
Raymond Hettinger54f02222002-06-01 14:18:47 +0000639 elif 'uri' in headers:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000640 newurl = headers['uri']
641 else:
642 return
643 void = fp.read()
644 fp.close()
Guido van Rossum3527f591999-03-29 20:23:41 +0000645 # In case the server sent a relative URL, join with original:
Moshe Zadka5d87d472001-04-09 14:54:21 +0000646 newurl = basejoin(self.type + ":" + url, newurl)
Guido van Rossumfa19f7c2003-05-16 01:46:51 +0000647 return self.open(newurl)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000648
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000649 def http_error_301(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000650 """Error 301 -- also relocated (permanently)."""
651 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
Guido van Rossume6ad8911996-09-10 17:02:56 +0000652
Raymond Hettinger024aaa12003-04-24 15:32:12 +0000653 def http_error_303(self, url, fp, errcode, errmsg, headers, data=None):
654 """Error 303 -- also relocated (essentially identical to 302)."""
655 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
656
Guido van Rossumfa19f7c2003-05-16 01:46:51 +0000657 def http_error_307(self, url, fp, errcode, errmsg, headers, data=None):
658 """Error 307 -- relocated, but turn POST into error."""
659 if data is None:
660 return self.http_error_302(url, fp, errcode, errmsg, headers, data)
661 else:
662 return self.http_error_default(url, fp, errcode, errmsg, headers)
663
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000664 def http_error_401(self, url, fp, errcode, errmsg, headers, data=None):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000665 """Error 401 -- authentication required.
Martin v. Löwis3e865952006-01-24 15:51:21 +0000666 This function supports Basic authentication only."""
Raymond Hettinger54f02222002-06-01 14:18:47 +0000667 if not 'www-authenticate' in headers:
Tim Peters85ba6732001-02-28 08:26:44 +0000668 URLopener.http_error_default(self, url, fp,
Fred Drakec680ae82001-10-13 18:37:07 +0000669 errcode, errmsg, headers)
Moshe Zadkae99bd172001-02-27 06:27:04 +0000670 stuff = headers['www-authenticate']
671 import re
672 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
673 if not match:
Tim Peters85ba6732001-02-28 08:26:44 +0000674 URLopener.http_error_default(self, url, fp,
Moshe Zadkae99bd172001-02-27 06:27:04 +0000675 errcode, errmsg, headers)
676 scheme, realm = match.groups()
677 if scheme.lower() != 'basic':
Tim Peters85ba6732001-02-28 08:26:44 +0000678 URLopener.http_error_default(self, url, fp,
Moshe Zadkae99bd172001-02-27 06:27:04 +0000679 errcode, errmsg, headers)
680 name = 'retry_' + self.type + '_basic_auth'
681 if data is None:
682 return getattr(self,name)(url, realm)
683 else:
684 return getattr(self,name)(url, realm, data)
Tim Peters92037a12006-01-24 22:44:08 +0000685
Martin v. Löwis3e865952006-01-24 15:51:21 +0000686 def http_error_407(self, url, fp, errcode, errmsg, headers, data=None):
687 """Error 407 -- proxy authentication required.
688 This function supports Basic authentication only."""
689 if not 'proxy-authenticate' in headers:
690 URLopener.http_error_default(self, url, fp,
691 errcode, errmsg, headers)
692 stuff = headers['proxy-authenticate']
693 import re
694 match = re.match('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', stuff)
695 if not match:
696 URLopener.http_error_default(self, url, fp,
697 errcode, errmsg, headers)
698 scheme, realm = match.groups()
699 if scheme.lower() != 'basic':
700 URLopener.http_error_default(self, url, fp,
701 errcode, errmsg, headers)
702 name = 'retry_proxy_' + self.type + '_basic_auth'
703 if data is None:
704 return getattr(self,name)(url, realm)
705 else:
706 return getattr(self,name)(url, realm, data)
Tim Peters92037a12006-01-24 22:44:08 +0000707
Martin v. Löwis3e865952006-01-24 15:51:21 +0000708 def retry_proxy_http_basic_auth(self, url, realm, data=None):
709 host, selector = splithost(url)
710 newurl = 'http://' + host + selector
711 proxy = self.proxies['http']
712 urltype, proxyhost = splittype(proxy)
713 proxyhost, proxyselector = splithost(proxyhost)
714 i = proxyhost.find('@') + 1
715 proxyhost = proxyhost[i:]
716 user, passwd = self.get_user_passwd(proxyhost, realm, i)
717 if not (user or passwd): return None
718 proxyhost = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + proxyhost
719 self.proxies['http'] = 'http://' + proxyhost + proxyselector
720 if data is None:
721 return self.open(newurl)
722 else:
723 return self.open(newurl, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000724
Martin v. Löwis3e865952006-01-24 15:51:21 +0000725 def retry_proxy_https_basic_auth(self, url, realm, data=None):
726 host, selector = splithost(url)
727 newurl = 'https://' + host + selector
728 proxy = self.proxies['https']
729 urltype, proxyhost = splittype(proxy)
730 proxyhost, proxyselector = splithost(proxyhost)
731 i = proxyhost.find('@') + 1
732 proxyhost = proxyhost[i:]
733 user, passwd = self.get_user_passwd(proxyhost, realm, i)
734 if not (user or passwd): return None
735 proxyhost = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + proxyhost
736 self.proxies['https'] = 'https://' + proxyhost + proxyselector
737 if data is None:
738 return self.open(newurl)
739 else:
740 return self.open(newurl, data)
Tim Peters92037a12006-01-24 22:44:08 +0000741
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000742 def retry_http_basic_auth(self, url, realm, data=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000743 host, selector = splithost(url)
Guido van Rossumb2493f82000-12-15 15:01:37 +0000744 i = host.find('@') + 1
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000745 host = host[i:]
746 user, passwd = self.get_user_passwd(host, realm, i)
747 if not (user or passwd): return None
Guido van Rossumafc4f042001-01-15 18:31:13 +0000748 host = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + host
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000749 newurl = 'http://' + host + selector
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000750 if data is None:
751 return self.open(newurl)
752 else:
753 return self.open(newurl, data)
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000754
Guido van Rossum3c8baed2000-02-01 23:36:55 +0000755 def retry_https_basic_auth(self, url, realm, data=None):
Tim Peterse1190062001-01-15 03:34:38 +0000756 host, selector = splithost(url)
757 i = host.find('@') + 1
758 host = host[i:]
759 user, passwd = self.get_user_passwd(host, realm, i)
760 if not (user or passwd): return None
Guido van Rossumafc4f042001-01-15 18:31:13 +0000761 host = quote(user, safe='') + ':' + quote(passwd, safe='') + '@' + host
Martin v. Löwis3e865952006-01-24 15:51:21 +0000762 newurl = 'https://' + host + selector
763 if data is None:
764 return self.open(newurl)
765 else:
766 return self.open(newurl, data)
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000767
Florent Xiclunae127e242010-05-17 10:39:07 +0000768 def get_user_passwd(self, host, realm, clear_cache=0):
Guido van Rossumb2493f82000-12-15 15:01:37 +0000769 key = realm + '@' + host.lower()
Raymond Hettinger54f02222002-06-01 14:18:47 +0000770 if key in self.auth_cache:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000771 if clear_cache:
772 del self.auth_cache[key]
773 else:
774 return self.auth_cache[key]
775 user, passwd = self.prompt_user_passwd(host, realm)
776 if user or passwd: self.auth_cache[key] = (user, passwd)
777 return user, passwd
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000778
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000779 def prompt_user_passwd(self, host, realm):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000780 """Override this in a GUI environment!"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000781 import getpass
782 try:
783 user = raw_input("Enter username for %s at %s: " % (realm,
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000784 host))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000785 passwd = getpass.getpass("Enter password for %s in %s at %s: " %
786 (user, realm, host))
787 return user, passwd
788 except KeyboardInterrupt:
789 print
790 return None, None
Guido van Rossumbbb0a051995-08-04 04:29:05 +0000791
792
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000793# Utility functions
794
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000795_localhost = None
796def localhost():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000797 """Return the IP address of the magic hostname 'localhost'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000798 global _localhost
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000799 if _localhost is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000800 _localhost = socket.gethostbyname('localhost')
801 return _localhost
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000802
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000803_thishost = None
804def thishost():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000805 """Return the IP address of the current host."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000806 global _thishost
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000807 if _thishost is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000808 _thishost = socket.gethostbyname(socket.gethostname())
809 return _thishost
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000810
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000811_ftperrors = None
812def ftperrors():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000813 """Return the set of errors raised by the FTP class."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000814 global _ftperrors
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000815 if _ftperrors is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000816 import ftplib
817 _ftperrors = ftplib.all_errors
818 return _ftperrors
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000819
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000820_noheaders = None
821def noheaders():
Guido van Rossume7b146f2000-02-04 15:28:42 +0000822 """Return an empty mimetools.Message object."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000823 global _noheaders
Raymond Hettinger10ff7062002-06-02 03:04:52 +0000824 if _noheaders is None:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000825 import mimetools
Raymond Hettingera6172712004-12-31 19:15:26 +0000826 try:
827 from cStringIO import StringIO
828 except ImportError:
829 from StringIO import StringIO
830 _noheaders = mimetools.Message(StringIO(), 0)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000831 _noheaders.fp.close() # Recycle file descriptor
832 return _noheaders
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000833
834
835# Utility classes
836
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000837class ftpwrapper:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000838 """Class used by open_ftp() for cache of open FTP connections."""
839
Facundo Batista4f1b1ed2008-05-29 16:39:26 +0000840 def __init__(self, user, passwd, host, port, dirs,
841 timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000842 self.user = user
843 self.passwd = passwd
844 self.host = host
845 self.port = port
846 self.dirs = dirs
Facundo Batista711a54e2007-05-24 17:50:54 +0000847 self.timeout = timeout
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000848 self.init()
Guido van Rossume7b146f2000-02-04 15:28:42 +0000849
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000850 def init(self):
851 import ftplib
852 self.busy = 0
853 self.ftp = ftplib.FTP()
Facundo Batista711a54e2007-05-24 17:50:54 +0000854 self.ftp.connect(self.host, self.port, self.timeout)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000855 self.ftp.login(self.user, self.passwd)
856 for dir in self.dirs:
857 self.ftp.cwd(dir)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000858
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000859 def retrfile(self, file, type):
860 import ftplib
861 self.endtransfer()
862 if type in ('d', 'D'): cmd = 'TYPE A'; isdir = 1
863 else: cmd = 'TYPE ' + type; isdir = 0
864 try:
865 self.ftp.voidcmd(cmd)
866 except ftplib.all_errors:
867 self.init()
868 self.ftp.voidcmd(cmd)
869 conn = None
870 if file and not isdir:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000871 # Try to retrieve as a file
872 try:
873 cmd = 'RETR ' + file
874 conn = self.ftp.ntransfercmd(cmd)
875 except ftplib.error_perm, reason:
Guido van Rossumb2493f82000-12-15 15:01:37 +0000876 if str(reason)[:3] != '550':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000877 raise IOError, ('ftp error', reason), sys.exc_info()[2]
878 if not conn:
879 # Set transfer mode to ASCII!
880 self.ftp.voidcmd('TYPE A')
Georg Brandld5e6cf22008-01-20 12:18:17 +0000881 # Try a directory listing. Verify that directory exists.
882 if file:
883 pwd = self.ftp.pwd()
884 try:
885 try:
886 self.ftp.cwd(file)
887 except ftplib.error_perm, reason:
888 raise IOError, ('ftp error', reason), sys.exc_info()[2]
889 finally:
890 self.ftp.cwd(pwd)
891 cmd = 'LIST ' + file
892 else:
893 cmd = 'LIST'
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000894 conn = self.ftp.ntransfercmd(cmd)
895 self.busy = 1
896 # Pass back both a suitably decorated object and a retrieval length
897 return (addclosehook(conn[0].makefile('rb'),
Fredrik Lundhb49f88b2000-09-24 18:51:25 +0000898 self.endtransfer), conn[1])
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000899 def endtransfer(self):
900 if not self.busy:
901 return
902 self.busy = 0
903 try:
904 self.ftp.voidresp()
905 except ftperrors():
906 pass
Guido van Rossume7b146f2000-02-04 15:28:42 +0000907
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000908 def close(self):
909 self.endtransfer()
910 try:
911 self.ftp.close()
912 except ftperrors():
913 pass
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000914
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000915class addbase:
Guido van Rossume7b146f2000-02-04 15:28:42 +0000916 """Base class for addinfo and addclosehook."""
917
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000918 def __init__(self, fp):
919 self.fp = fp
920 self.read = self.fp.read
921 self.readline = self.fp.readline
Guido van Rossum09c8b6c1999-12-07 21:37:17 +0000922 if hasattr(self.fp, "readlines"): self.readlines = self.fp.readlines
Georg Brandl1f663572005-11-26 16:50:44 +0000923 if hasattr(self.fp, "fileno"):
924 self.fileno = self.fp.fileno
925 else:
926 self.fileno = lambda: None
Raymond Hettinger42182eb2003-03-09 05:33:33 +0000927 if hasattr(self.fp, "__iter__"):
928 self.__iter__ = self.fp.__iter__
929 if hasattr(self.fp, "next"):
930 self.next = self.fp.next
Guido van Rossume7b146f2000-02-04 15:28:42 +0000931
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000932 def __repr__(self):
Walter Dörwald70a6b492004-02-12 17:35:32 +0000933 return '<%s at %r whose fp = %r>' % (self.__class__.__name__,
934 id(self), self.fp)
Guido van Rossume7b146f2000-02-04 15:28:42 +0000935
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000936 def close(self):
937 self.read = None
938 self.readline = None
939 self.readlines = None
940 self.fileno = None
941 if self.fp: self.fp.close()
942 self.fp = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000943
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000944class addclosehook(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000945 """Class to add a close hook to an open file."""
946
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000947 def __init__(self, fp, closehook, *hookargs):
948 addbase.__init__(self, fp)
949 self.closehook = closehook
950 self.hookargs = hookargs
Guido van Rossume7b146f2000-02-04 15:28:42 +0000951
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000952 def close(self):
Guido van Rossumc580dae2000-05-24 13:21:46 +0000953 addbase.close(self)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000954 if self.closehook:
Guido van Rossum68468eb2003-02-27 20:14:51 +0000955 self.closehook(*self.hookargs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000956 self.closehook = None
957 self.hookargs = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000958
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000959class addinfo(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000960 """class to add an info() method to an open file."""
961
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000962 def __init__(self, fp, headers):
963 addbase.__init__(self, fp)
964 self.headers = headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000965
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000966 def info(self):
967 return self.headers
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000968
Guido van Rossume6ad8911996-09-10 17:02:56 +0000969class addinfourl(addbase):
Guido van Rossume7b146f2000-02-04 15:28:42 +0000970 """class to add info() and geturl() methods to an open file."""
971
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000972 def __init__(self, fp, headers, url, code=None):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000973 addbase.__init__(self, fp)
974 self.headers = headers
975 self.url = url
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000976 self.code = code
Guido van Rossume7b146f2000-02-04 15:28:42 +0000977
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000978 def info(self):
979 return self.headers
Guido van Rossume7b146f2000-02-04 15:28:42 +0000980
Georg Brandl9b0d46d2008-01-20 11:43:03 +0000981 def getcode(self):
982 return self.code
983
Jeremy Hyltonf90b0021999-02-25 16:12:12 +0000984 def geturl(self):
985 return self.url
Guido van Rossume6ad8911996-09-10 17:02:56 +0000986
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000987
Guido van Rossum7c395db1994-07-04 22:14:49 +0000988# Utilities to parse URLs (most of these return None for missing parts):
Sjoerd Mullendere0371b81995-11-10 10:36:07 +0000989# unwrap('<URL:type://host/path>') --> 'type://host/path'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000990# splittype('type:opaquestring') --> 'type', 'opaquestring'
991# splithost('//host[:port]/path') --> 'host[:port]', '/path'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000992# splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'
993# splitpasswd('user:passwd') -> 'user', 'passwd'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +0000994# splitport('host:port') --> 'host', 'port'
995# splitquery('/path?query') --> '/path', 'query'
996# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum7c395db1994-07-04 22:14:49 +0000997# splitattr('/path;attr1=value1;attr2=value2;...') ->
998# '/path', ['attr1=value1', 'attr2=value2', ...]
999# splitvalue('attr=value') --> 'attr', 'value'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001000# unquote('abc%20def') -> 'abc def'
1001# quote('abc def') -> 'abc%20def')
1002
Walter Dörwald65230a22002-06-03 15:58:32 +00001003try:
1004 unicode
1005except NameError:
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001006 def _is_unicode(x):
1007 return 0
Walter Dörwald65230a22002-06-03 15:58:32 +00001008else:
1009 def _is_unicode(x):
1010 return isinstance(x, unicode)
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001011
Martin v. Löwis1d994332000-12-03 18:30:10 +00001012def toBytes(url):
1013 """toBytes(u"URL") --> 'URL'."""
1014 # Most URL schemes require ASCII. If that changes, the conversion
1015 # can be relaxed
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001016 if _is_unicode(url):
Martin v. Löwis1d994332000-12-03 18:30:10 +00001017 try:
1018 url = url.encode("ASCII")
1019 except UnicodeError:
Guido van Rossumb2493f82000-12-15 15:01:37 +00001020 raise UnicodeError("URL " + repr(url) +
1021 " contains non-ASCII characters")
Martin v. Löwis1d994332000-12-03 18:30:10 +00001022 return url
1023
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001024def unwrap(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001025 """unwrap('<URL:type://host/path>') --> 'type://host/path'."""
Guido van Rossumb2493f82000-12-15 15:01:37 +00001026 url = url.strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001027 if url[:1] == '<' and url[-1:] == '>':
Guido van Rossumb2493f82000-12-15 15:01:37 +00001028 url = url[1:-1].strip()
1029 if url[:4] == 'URL:': url = url[4:].strip()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001030 return url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001031
Guido van Rossum332e1441997-09-29 23:23:46 +00001032_typeprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001033def splittype(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001034 """splittype('type:opaquestring') --> 'type', 'opaquestring'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001035 global _typeprog
1036 if _typeprog is None:
1037 import re
1038 _typeprog = re.compile('^([^/:]+):')
Guido van Rossum332e1441997-09-29 23:23:46 +00001039
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001040 match = _typeprog.match(url)
1041 if match:
1042 scheme = match.group(1)
Fred Drake9e94afd2000-07-01 07:03:30 +00001043 return scheme.lower(), url[len(scheme) + 1:]
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001044 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001045
Guido van Rossum332e1441997-09-29 23:23:46 +00001046_hostprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001047def splithost(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001048 """splithost('//host[:port]/path') --> 'host[:port]', '/path'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001049 global _hostprog
1050 if _hostprog is None:
1051 import re
Georg Brandl1c168d82006-03-26 20:59:38 +00001052 _hostprog = re.compile('^//([^/?]*)(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001053
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001054 match = _hostprog.match(url)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001055 if match: return match.group(1, 2)
1056 return None, url
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001057
Guido van Rossum332e1441997-09-29 23:23:46 +00001058_userprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001059def splituser(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001060 """splituser('user[:passwd]@host[:port]') --> 'user[:passwd]', 'host[:port]'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001061 global _userprog
1062 if _userprog is None:
1063 import re
Raymond Hettingerf2e45dd2002-08-18 20:08:56 +00001064 _userprog = re.compile('^(.*)@(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001065
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001066 match = _userprog.match(host)
Fred Drake567ca8e2000-08-21 21:42:42 +00001067 if match: return map(unquote, match.group(1, 2))
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001068 return None, host
Guido van Rossum7c395db1994-07-04 22:14:49 +00001069
Guido van Rossum332e1441997-09-29 23:23:46 +00001070_passwdprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001071def splitpasswd(user):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001072 """splitpasswd('user:passwd') -> 'user', 'passwd'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001073 global _passwdprog
1074 if _passwdprog is None:
1075 import re
Senthil Kumaran5e95e762009-03-30 21:51:50 +00001076 _passwdprog = re.compile('^([^:]*):(.*)$',re.S)
Guido van Rossum332e1441997-09-29 23:23:46 +00001077
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001078 match = _passwdprog.match(user)
1079 if match: return match.group(1, 2)
1080 return user, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001081
Guido van Rossume7b146f2000-02-04 15:28:42 +00001082# splittag('/path#tag') --> '/path', 'tag'
Guido van Rossum332e1441997-09-29 23:23:46 +00001083_portprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001084def splitport(host):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001085 """splitport('host:port') --> 'host', 'port'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001086 global _portprog
1087 if _portprog is None:
1088 import re
1089 _portprog = re.compile('^(.*):([0-9]+)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001090
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001091 match = _portprog.match(host)
1092 if match: return match.group(1, 2)
1093 return host, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001094
Guido van Rossum332e1441997-09-29 23:23:46 +00001095_nportprog = None
Guido van Rossum53725a21996-06-13 19:12:35 +00001096def splitnport(host, defport=-1):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001097 """Split host and port, returning numeric port.
1098 Return given default port if no ':' found; defaults to -1.
1099 Return numerical port if a valid number are found after ':'.
1100 Return None if ':' but not a valid number."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001101 global _nportprog
1102 if _nportprog is None:
1103 import re
1104 _nportprog = re.compile('^(.*):(.*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001105
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001106 match = _nportprog.match(host)
1107 if match:
1108 host, port = match.group(1, 2)
1109 try:
Guido van Rossumb2493f82000-12-15 15:01:37 +00001110 if not port: raise ValueError, "no digits"
1111 nport = int(port)
1112 except ValueError:
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001113 nport = None
1114 return host, nport
1115 return host, defport
Guido van Rossum53725a21996-06-13 19:12:35 +00001116
Guido van Rossum332e1441997-09-29 23:23:46 +00001117_queryprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001118def splitquery(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001119 """splitquery('/path?query') --> '/path', 'query'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001120 global _queryprog
1121 if _queryprog is None:
1122 import re
1123 _queryprog = re.compile('^(.*)\?([^?]*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001124
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001125 match = _queryprog.match(url)
1126 if match: return match.group(1, 2)
1127 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001128
Guido van Rossum332e1441997-09-29 23:23:46 +00001129_tagprog = None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001130def splittag(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001131 """splittag('/path#tag') --> '/path', 'tag'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001132 global _tagprog
1133 if _tagprog is None:
1134 import re
1135 _tagprog = re.compile('^(.*)#([^#]*)$')
Guido van Rossum7e7ca0b1998-03-26 21:01:39 +00001136
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001137 match = _tagprog.match(url)
1138 if match: return match.group(1, 2)
1139 return url, None
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001140
Guido van Rossum7c395db1994-07-04 22:14:49 +00001141def splitattr(url):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001142 """splitattr('/path;attr1=value1;attr2=value2;...') ->
1143 '/path', ['attr1=value1', 'attr2=value2', ...]."""
Guido van Rossumb2493f82000-12-15 15:01:37 +00001144 words = url.split(';')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001145 return words[0], words[1:]
Guido van Rossum7c395db1994-07-04 22:14:49 +00001146
Guido van Rossum332e1441997-09-29 23:23:46 +00001147_valueprog = None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001148def splitvalue(attr):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001149 """splitvalue('attr=value') --> 'attr', 'value'."""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001150 global _valueprog
1151 if _valueprog is None:
1152 import re
1153 _valueprog = re.compile('^([^=]*)=(.*)$')
Guido van Rossum332e1441997-09-29 23:23:46 +00001154
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001155 match = _valueprog.match(attr)
1156 if match: return match.group(1, 2)
1157 return attr, None
Guido van Rossum7c395db1994-07-04 22:14:49 +00001158
R. David Murraybfbdefe2010-05-25 15:20:46 +00001159# urlparse contains a duplicate of this method to avoid a circular import. If
1160# you update this method, also update the copy in urlparse. This code
1161# duplication does not exist in Python3.
1162
Senthil Kumaranf3e9b2a2010-03-18 12:14:15 +00001163_hexdig = '0123456789ABCDEFabcdef'
Florent Xiclunae127e242010-05-17 10:39:07 +00001164_hextochr = dict((a + b, chr(int(a + b, 16)))
1165 for a in _hexdig for b in _hexdig)
Raymond Hettinger803ce802005-09-10 06:49:04 +00001166
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001167def unquote(s):
Guido van Rossume7b146f2000-02-04 15:28:42 +00001168 """unquote('abc%20def') -> 'abc def'."""
Raymond Hettinger803ce802005-09-10 06:49:04 +00001169 res = s.split('%')
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001170 # fastpath
1171 if len(res) == 1:
1172 return s
1173 s = res[0]
1174 for item in res[1:]:
Raymond Hettinger803ce802005-09-10 06:49:04 +00001175 try:
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001176 s += _hextochr[item[:2]] + item[2:]
Raymond Hettinger803ce802005-09-10 06:49:04 +00001177 except KeyError:
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001178 s += '%' + item
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +00001179 except UnicodeDecodeError:
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001180 s += unichr(int(item[:2], 16)) + item[2:]
1181 return s
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001182
Guido van Rossum0564e121996-12-13 14:47:36 +00001183def unquote_plus(s):
Skip Montanaro79f1c172000-08-22 03:00:52 +00001184 """unquote('%7e/abc+def') -> '~/abc def'"""
Brett Cannonaaeffaf2004-03-23 23:50:17 +00001185 s = s.replace('+', ' ')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001186 return unquote(s)
Guido van Rossum0564e121996-12-13 14:47:36 +00001187
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001188always_safe = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
Jeremy Hylton6102e292000-08-31 15:48:10 +00001189 'abcdefghijklmnopqrstuvwxyz'
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001190 '0123456789' '_.-')
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001191_safe_map = {}
1192for i, c in zip(xrange(256), str(bytearray(xrange(256)))):
1193 _safe_map[c] = c if (i < 128 and c in always_safe) else '%{:02X}'.format(i)
1194_safe_quoters = {}
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001195
Florent Xiclunae127e242010-05-17 10:39:07 +00001196def quote(s, safe='/'):
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001197 """quote('abc def') -> 'abc%20def'
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001198
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001199 Each part of a URL, e.g. the path info, the query, etc., has a
1200 different set of reserved characters that must be quoted.
1201
1202 RFC 2396 Uniform Resource Identifiers (URI): Generic Syntax lists
1203 the following reserved characters.
1204
1205 reserved = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" |
1206 "$" | ","
1207
1208 Each of these characters is reserved in some component of a URL,
1209 but not necessarily in all of them.
1210
1211 By default, the quote function is intended for quoting the path
1212 section of a URL. Thus, it will not encode '/'. This character
1213 is reserved, but in typical usage the quote function is being
1214 called on a path where the existing slash characters are used as
1215 reserved characters.
1216 """
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001217 # fastpath
1218 if not s:
1219 return s
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001220 cachekey = (safe, always_safe)
1221 try:
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001222 (quoter, safe) = _safe_quoters[cachekey]
Raymond Hettinger199d2f72005-09-09 22:27:13 +00001223 except KeyError:
Florent Xiclunaaf87f9f2010-05-17 13:35:09 +00001224 safe_map = _safe_map.copy()
1225 safe_map.update([(c, c) for c in safe])
1226 quoter = safe_map.__getitem__
1227 safe = always_safe + safe
1228 _safe_quoters[cachekey] = (quoter, safe)
1229 if not s.rstrip(safe):
1230 return s
1231 return ''.join(map(quoter, s))
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001232
Florent Xiclunae127e242010-05-17 10:39:07 +00001233def quote_plus(s, safe=''):
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +00001234 """Quote the query fragment of a URL; replacing ' ' with '+'"""
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001235 if ' ' in s:
Raymond Hettingercf6b6322005-09-10 18:17:54 +00001236 s = quote(s, safe + ' ')
1237 return s.replace(' ', '+')
1238 return quote(s, safe)
Guido van Rossum0564e121996-12-13 14:47:36 +00001239
Florent Xiclunae127e242010-05-17 10:39:07 +00001240def urlencode(query, doseq=0):
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001241 """Encode a sequence of two-element tuples or dictionary into a URL query string.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001242
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001243 If any values in the query arg are sequences and doseq is true, each
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001244 sequence element is converted to a separate parameter.
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001245
1246 If the query arg is a sequence of two-element tuples, the order of the
1247 parameters in the output will match the order of parameters in the
1248 input.
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001249 """
Tim Peters658cba62001-02-09 20:06:00 +00001250
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001251 if hasattr(query,"items"):
1252 # mapping objects
1253 query = query.items()
1254 else:
1255 # it's a bother at times that strings and string-like objects are
1256 # sequences...
1257 try:
1258 # non-sequence items should not work with len()
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001259 # non-empty strings will fail this
Walter Dörwald65230a22002-06-03 15:58:32 +00001260 if len(query) and not isinstance(query[0], tuple):
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001261 raise TypeError
1262 # zero-length sequences of all types will get here and succeed,
1263 # but that's a minor nit - since the original implementation
1264 # allowed empty dicts that type of behavior probably should be
1265 # preserved for consistency
1266 except TypeError:
1267 ty,va,tb = sys.exc_info()
1268 raise TypeError, "not a valid non-string sequence or mapping object", tb
1269
Guido van Rossume7b146f2000-02-04 15:28:42 +00001270 l = []
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001271 if not doseq:
1272 # preserve old behavior
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001273 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001274 k = quote_plus(str(k))
1275 v = quote_plus(str(v))
1276 l.append(k + '=' + v)
1277 else:
Skip Montanaro14f1ad42001-01-28 21:11:12 +00001278 for k, v in query:
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001279 k = quote_plus(str(k))
Walter Dörwald65230a22002-06-03 15:58:32 +00001280 if isinstance(v, str):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001281 v = quote_plus(v)
1282 l.append(k + '=' + v)
Guido van Rossum4b46c0a2002-05-24 17:58:05 +00001283 elif _is_unicode(v):
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001284 # is there a reasonable way to convert to ASCII?
1285 # encode generates a string, but "replace" or "ignore"
1286 # lose information and "strict" can raise UnicodeError
1287 v = quote_plus(v.encode("ASCII","replace"))
1288 l.append(k + '=' + v)
1289 else:
1290 try:
1291 # is this a sufficient test for sequence-ness?
Georg Brandl84fedf72010-02-06 22:59:15 +00001292 len(v)
Skip Montanaroa5d23a12001-01-20 15:56:39 +00001293 except TypeError:
1294 # not a sequence
1295 v = quote_plus(str(v))
1296 l.append(k + '=' + v)
1297 else:
1298 # loop over the sequence
1299 for elt in v:
1300 l.append(k + '=' + quote_plus(str(elt)))
Guido van Rossumb2493f82000-12-15 15:01:37 +00001301 return '&'.join(l)
Guido van Rossum810a3391998-07-22 21:33:23 +00001302
Guido van Rossum442e7201996-03-20 15:33:11 +00001303# Proxy handling
Mark Hammond4f570b92000-07-26 07:04:38 +00001304def getproxies_environment():
1305 """Return a dictionary of scheme -> proxy server URL mappings.
1306
1307 Scan the environment for variables named <scheme>_proxy;
1308 this seems to be the standard convention. If you need a
1309 different way, you can pass a proxies dictionary to the
1310 [Fancy]URLopener constructor.
1311
1312 """
1313 proxies = {}
1314 for name, value in os.environ.items():
Guido van Rossumb2493f82000-12-15 15:01:37 +00001315 name = name.lower()
Mark Hammond4f570b92000-07-26 07:04:38 +00001316 if value and name[-6:] == '_proxy':
1317 proxies[name[:-6]] = value
1318 return proxies
1319
Georg Brandl22350112008-01-20 12:05:43 +00001320def proxy_bypass_environment(host):
1321 """Test if proxies should not be used for a particular host.
1322
1323 Checks the environment for a variable named no_proxy, which should
1324 be a list of DNS suffixes separated by commas, or '*' for all hosts.
1325 """
1326 no_proxy = os.environ.get('no_proxy', '') or os.environ.get('NO_PROXY', '')
1327 # '*' is special case for always bypass
1328 if no_proxy == '*':
1329 return 1
1330 # strip port off host
1331 hostonly, port = splitport(host)
1332 # check if the host ends with any of the DNS suffixes
1333 for name in no_proxy.split(','):
1334 if name and (hostonly.endswith(name) or host.endswith(name)):
1335 return 1
1336 # otherwise, don't bypass
1337 return 0
1338
1339
Jack Jansen11d9b062004-07-16 11:45:00 +00001340if sys.platform == 'darwin':
Ronald Oussoren51f06332009-09-20 10:31:22 +00001341 from _scproxy import _get_proxy_settings, _get_proxies
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001342
1343 def proxy_bypass_macosx_sysconf(host):
1344 """
1345 Return True iff this host shouldn't be accessed using a proxy
1346
1347 This function uses the MacOSX framework SystemConfiguration
1348 to fetch the proxy information.
1349 """
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001350 import re
1351 import socket
1352 from fnmatch import fnmatch
1353
Ronald Oussoren31802d02009-10-18 07:07:00 +00001354 hostonly, port = splitport(host)
1355
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001356 def ip2num(ipAddr):
1357 parts = ipAddr.split('.')
1358 parts = map(int, parts)
1359 if len(parts) != 4:
1360 parts = (parts + [0, 0, 0, 0])[:4]
1361 return (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8) | parts[3]
1362
Ronald Oussoren51f06332009-09-20 10:31:22 +00001363 proxy_settings = _get_proxy_settings()
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001364
Ronald Oussoren51f06332009-09-20 10:31:22 +00001365 # Check for simple host names:
1366 if '.' not in host:
1367 if proxy_settings['exclude_simple']:
1368 return True
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001369
Ronald Oussoren31802d02009-10-18 07:07:00 +00001370 hostIP = None
1371
Ronald Oussoren809073b2009-09-20 10:54:07 +00001372 for value in proxy_settings.get('exceptions', ()):
Ronald Oussoren51f06332009-09-20 10:31:22 +00001373 # Items in the list are strings like these: *.local, 169.254/16
Ronald Oussoren51f06332009-09-20 10:31:22 +00001374 if not value: continue
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001375
Ronald Oussoren51f06332009-09-20 10:31:22 +00001376 m = re.match(r"(\d+(?:\.\d+)*)(/\d+)?", value)
1377 if m is not None:
1378 if hostIP is None:
Ronald Oussoren31802d02009-10-18 07:07:00 +00001379 try:
1380 hostIP = socket.gethostbyname(hostonly)
1381 hostIP = ip2num(hostIP)
1382 except socket.error:
1383 continue
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001384
Ronald Oussoren51f06332009-09-20 10:31:22 +00001385 base = ip2num(m.group(1))
Ronald Oussorenb96fbb82010-06-27 13:59:39 +00001386 mask = m.group(2)
1387 if mask is None:
1388 mask = 8 * (m.group(1).count('.') + 1)
1389
1390 else:
1391 mask = int(mask[1:])
1392 mask = 32 - mask
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001393
Ronald Oussoren51f06332009-09-20 10:31:22 +00001394 if (hostIP >> mask) == (base >> mask):
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001395 return True
1396
Ronald Oussoren51f06332009-09-20 10:31:22 +00001397 elif fnmatch(host, value):
1398 return True
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001399
Ronald Oussoren51f06332009-09-20 10:31:22 +00001400 return False
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001401
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001402 def getproxies_macosx_sysconf():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001403 """Return a dictionary of scheme -> proxy server URL mappings.
Guido van Rossum442e7201996-03-20 15:33:11 +00001404
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001405 This function uses the MacOSX framework SystemConfiguration
1406 to fetch the proxy information.
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001407 """
Ronald Oussoren51f06332009-09-20 10:31:22 +00001408 return _get_proxies()
Mark Hammond4f570b92000-07-26 07:04:38 +00001409
Georg Brandl22350112008-01-20 12:05:43 +00001410 def proxy_bypass(host):
1411 if getproxies_environment():
1412 return proxy_bypass_environment(host)
1413 else:
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001414 return proxy_bypass_macosx_sysconf(host)
Tim Peters55c12d42001-08-09 18:04:14 +00001415
Jack Jansen11d9b062004-07-16 11:45:00 +00001416 def getproxies():
Ronald Oussoren9dd6b1d2008-05-12 11:31:05 +00001417 return getproxies_environment() or getproxies_macosx_sysconf()
Tim Peters182b5ac2004-07-18 06:16:08 +00001418
Mark Hammond4f570b92000-07-26 07:04:38 +00001419elif os.name == 'nt':
1420 def getproxies_registry():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001421 """Return a dictionary of scheme -> proxy server URL mappings.
Mark Hammond4f570b92000-07-26 07:04:38 +00001422
1423 Win32 uses the registry to store proxies.
1424
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001425 """
1426 proxies = {}
Mark Hammond4f570b92000-07-26 07:04:38 +00001427 try:
1428 import _winreg
1429 except ImportError:
1430 # Std module, so should be around - but you never know!
1431 return proxies
1432 try:
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001433 internetSettings = _winreg.OpenKey(_winreg.HKEY_CURRENT_USER,
1434 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
Mark Hammond4f570b92000-07-26 07:04:38 +00001435 proxyEnable = _winreg.QueryValueEx(internetSettings,
1436 'ProxyEnable')[0]
1437 if proxyEnable:
1438 # Returned as Unicode but problems if not converted to ASCII
1439 proxyServer = str(_winreg.QueryValueEx(internetSettings,
1440 'ProxyServer')[0])
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001441 if '=' in proxyServer:
1442 # Per-protocol settings
Mark Hammond4f570b92000-07-26 07:04:38 +00001443 for p in proxyServer.split(';'):
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001444 protocol, address = p.split('=', 1)
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001445 # See if address has a type:// prefix
Guido van Rossum64e5aa92002-04-02 14:38:16 +00001446 import re
1447 if not re.match('^([^/:]+)://', address):
Guido van Rossumb955d6c2002-03-31 23:38:48 +00001448 address = '%s://%s' % (protocol, address)
1449 proxies[protocol] = address
Fredrik Lundhb49f88b2000-09-24 18:51:25 +00001450 else:
1451 # Use one setting for all protocols
1452 if proxyServer[:5] == 'http:':
1453 proxies['http'] = proxyServer
1454 else:
1455 proxies['http'] = 'http://%s' % proxyServer
1456 proxies['ftp'] = 'ftp://%s' % proxyServer
Mark Hammond4f570b92000-07-26 07:04:38 +00001457 internetSettings.Close()
1458 except (WindowsError, ValueError, TypeError):
1459 # Either registry key not found etc, or the value in an
1460 # unexpected format.
1461 # proxies already set up to be empty so nothing to do
1462 pass
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001463 return proxies
Guido van Rossum442e7201996-03-20 15:33:11 +00001464
Mark Hammond4f570b92000-07-26 07:04:38 +00001465 def getproxies():
1466 """Return a dictionary of scheme -> proxy server URL mappings.
1467
1468 Returns settings gathered from the environment, if specified,
1469 or the registry.
1470
1471 """
1472 return getproxies_environment() or getproxies_registry()
Tim Peters55c12d42001-08-09 18:04:14 +00001473
Georg Brandl22350112008-01-20 12:05:43 +00001474 def proxy_bypass_registry(host):
Tim Peters55c12d42001-08-09 18:04:14 +00001475 try:
1476 import _winreg
1477 import re
Tim Peters55c12d42001-08-09 18:04:14 +00001478 except ImportError:
1479 # Std modules, so should be around - but you never know!
1480 return 0
1481 try:
1482 internetSettings = _winreg.OpenKey(_winreg.HKEY_CURRENT_USER,
1483 r'Software\Microsoft\Windows\CurrentVersion\Internet Settings')
1484 proxyEnable = _winreg.QueryValueEx(internetSettings,
1485 'ProxyEnable')[0]
1486 proxyOverride = str(_winreg.QueryValueEx(internetSettings,
1487 'ProxyOverride')[0])
1488 # ^^^^ Returned as Unicode but problems if not converted to ASCII
1489 except WindowsError:
1490 return 0
1491 if not proxyEnable or not proxyOverride:
1492 return 0
1493 # try to make a host list from name and IP address.
Georg Brandl1f636702006-02-18 23:10:23 +00001494 rawHost, port = splitport(host)
1495 host = [rawHost]
Tim Peters55c12d42001-08-09 18:04:14 +00001496 try:
Georg Brandl1f636702006-02-18 23:10:23 +00001497 addr = socket.gethostbyname(rawHost)
1498 if addr != rawHost:
Tim Peters55c12d42001-08-09 18:04:14 +00001499 host.append(addr)
1500 except socket.error:
1501 pass
Georg Brandl1f636702006-02-18 23:10:23 +00001502 try:
1503 fqdn = socket.getfqdn(rawHost)
1504 if fqdn != rawHost:
1505 host.append(fqdn)
1506 except socket.error:
1507 pass
Tim Peters55c12d42001-08-09 18:04:14 +00001508 # make a check value list from the registry entry: replace the
1509 # '<local>' string by the localhost entry and the corresponding
1510 # canonical entry.
1511 proxyOverride = proxyOverride.split(';')
Tim Peters55c12d42001-08-09 18:04:14 +00001512 # now check if we match one of the registry values.
1513 for test in proxyOverride:
Senthil Kumaran4af40d22009-05-01 05:59:52 +00001514 if test == '<local>':
1515 if '.' not in rawHost:
1516 return 1
Tim Petersab9ba272001-08-09 21:40:30 +00001517 test = test.replace(".", r"\.") # mask dots
1518 test = test.replace("*", r".*") # change glob sequence
1519 test = test.replace("?", r".") # change glob char
Tim Peters55c12d42001-08-09 18:04:14 +00001520 for val in host:
1521 # print "%s <--> %s" %( test, val )
1522 if re.match(test, val, re.I):
1523 return 1
1524 return 0
1525
Georg Brandl22350112008-01-20 12:05:43 +00001526 def proxy_bypass(host):
1527 """Return a dictionary of scheme -> proxy server URL mappings.
1528
1529 Returns settings gathered from the environment, if specified,
1530 or the registry.
1531
1532 """
1533 if getproxies_environment():
1534 return proxy_bypass_environment(host)
1535 else:
1536 return proxy_bypass_registry(host)
1537
Mark Hammond4f570b92000-07-26 07:04:38 +00001538else:
1539 # By default use environment variables
1540 getproxies = getproxies_environment
Georg Brandl22350112008-01-20 12:05:43 +00001541 proxy_bypass = proxy_bypass_environment
Guido van Rossum442e7201996-03-20 15:33:11 +00001542
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001543# Test and time quote() and unquote()
1544def test1():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001545 s = ''
1546 for i in range(256): s = s + chr(i)
1547 s = s*4
1548 t0 = time.time()
1549 qs = quote(s)
1550 uqs = unquote(qs)
1551 t1 = time.time()
1552 if uqs != s:
1553 print 'Wrong!'
Walter Dörwald70a6b492004-02-12 17:35:32 +00001554 print repr(s)
1555 print repr(qs)
1556 print repr(uqs)
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001557 print round(t1 - t0, 3), 'sec'
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001558
1559
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001560def reporthook(blocknum, blocksize, totalsize):
1561 # Report during remote transfers
Guido van Rossumb2493f82000-12-15 15:01:37 +00001562 print "Block number: %d, Block size: %d, Total size: %d" % (
1563 blocknum, blocksize, totalsize)
Guido van Rossum9ab96d41998-09-28 14:07:00 +00001564
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001565# Test program
Guido van Rossum23490151998-06-25 02:39:00 +00001566def test(args=[]):
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001567 if not args:
1568 args = [
1569 '/etc/passwd',
1570 'file:/etc/passwd',
1571 'file://localhost/etc/passwd',
Collin Winter071d1ae2007-03-12 01:55:54 +00001572 'ftp://ftp.gnu.org/pub/README',
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001573 'http://www.python.org/index.html',
1574 ]
Guido van Rossum09c8b6c1999-12-07 21:37:17 +00001575 if hasattr(URLopener, "open_https"):
1576 args.append('https://synergy.as.cmu.edu/~geek/')
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001577 try:
1578 for url in args:
1579 print '-'*10, url, '-'*10
1580 fn, h = urlretrieve(url, None, reporthook)
Guido van Rossumb2493f82000-12-15 15:01:37 +00001581 print fn
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001582 if h:
1583 print '======'
1584 for k in h.keys(): print k + ':', h[k]
1585 print '======'
Philip Jenvey0299d0d2009-12-03 02:40:13 +00001586 with open(fn, 'rb') as fp:
1587 data = fp.read()
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001588 if '\r' in data:
1589 table = string.maketrans("", "")
Guido van Rossumb2493f82000-12-15 15:01:37 +00001590 data = data.translate(table, "\r")
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001591 print data
1592 fn, h = None, None
1593 print '-'*40
1594 finally:
1595 urlcleanup()
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001596
Guido van Rossum23490151998-06-25 02:39:00 +00001597def main():
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001598 import getopt, sys
1599 try:
1600 opts, args = getopt.getopt(sys.argv[1:], "th")
1601 except getopt.error, msg:
1602 print msg
1603 print "Use -h for help"
1604 return
1605 t = 0
1606 for o, a in opts:
1607 if o == '-t':
1608 t = t + 1
1609 if o == '-h':
1610 print "Usage: python urllib.py [-t] [url ...]"
1611 print "-t runs self-test;",
1612 print "otherwise, contents of urls are printed"
1613 return
1614 if t:
1615 if t > 1:
1616 test1()
1617 test(args)
1618 else:
1619 if not args:
1620 print "Use -h for help"
1621 for url in args:
1622 print urlopen(url).read(),
Guido van Rossum23490151998-06-25 02:39:00 +00001623
Guido van Rossum7c6ebb51994-03-22 12:05:32 +00001624# Run test program when run as a script
1625if __name__ == '__main__':
Jeremy Hyltonf90b0021999-02-25 16:12:12 +00001626 main()