Added timeout options for Python 2.6 and above

This commit is contained in:
2011-06-15 09:41:27 +01:00
parent ac69b7450a
commit 2e17210ca3

View File

@@ -1,3 +1,4 @@
import sys
import urllib, urllib2
from hashlib import sha1
from datetime import datetime, timedelta
@@ -65,7 +66,10 @@ class CachedDocumentManager(models.Manager):
req = urllib2.Request(url)
req.add_header('CCP-Contact', 'matalok@pleaseignore.com')
try:
if sys.version_info < (2, 6):
conn = urllib2.urlopen(req)
else:
conn = urllib2.urlopen(req, timeout=5)
except urllib2.HTTPError, e:
print "HTTP Error Code: %s" % e.code
raise DocumentRetrievalError(e.code)