Fixing the errors trigerred in test_urllib2net. Related to issue8656.

This commit is contained in:
Senthil Kumaran 2010-05-08 05:00:11 +00:00
parent 6057ba1f97
commit 18e4dd74e0
1 changed files with 8 additions and 5 deletions

View File

@ -1276,13 +1276,13 @@ class FileHandler(BaseHandler):
import email.utils
import mimetypes
host = req.get_host()
file = req.get_selector()
localfile = url2pathname(file)
filename = req.get_selector()
localfile = url2pathname(filename)
try:
stats = os.stat(localfile)
size = stats.st_size
modified = email.utils.formatdate(stats.st_mtime, usegmt=True)
mtype = mimetypes.guess_type(file)[0]
mtype = mimetypes.guess_type(filename)[0]
headers = mimetools.Message(StringIO(
'Content-type: %s\nContent-length: %d\nLast-modified: %s\n' %
(mtype or 'text/plain', size, modified)))
@ -1290,8 +1290,11 @@ class FileHandler(BaseHandler):
host, port = splitport(host)
if not host or \
(not port and socket.gethostbyname(host) in self.get_names()):
return addinfourl(open(localfile, 'rb'),
headers, 'file://'+ host + file)
if host:
origurl = 'file://' + host + filename
else:
origurl = 'file://' + filename
return addinfourl(open(localfile, 'rb'), headers, origurl)
except OSError, msg:
# urllib2 users shouldn't expect OSErrors coming from urlopen()
raise URLError(msg)