mercurial/httprepo.py
changeset 2015 1a09814a5b1f
parent 1870 8a8ab47cccde
child 2281 7761597b5da3
--- a/mercurial/httprepo.py	Tue Mar 28 09:26:38 2006 -0800
+++ b/mercurial/httprepo.py	Wed Mar 29 12:45:33 2006 +0200
@@ -9,7 +9,7 @@
 from remoterepo import *
 from i18n import gettext as _
 from demandload import *
-demandload(globals(), "hg os urllib urllib2 urlparse zlib util")
+demandload(globals(), "hg os urllib urllib2 urlparse zlib util httplib")
 
 class httprepository(remoterepository):
     def __init__(self, ui, path):
@@ -129,8 +129,11 @@
 
         def zgenerator(f):
             zd = zlib.decompressobj()
-            for chnk in f:
-                yield zd.decompress(chnk)
+            try:
+                for chnk in f:
+                    yield zd.decompress(chnk)
+            except httplib.HTTPException, inst:
+                raise IOError(None, _('connection ended unexpectedly'))
             yield zd.flush()
 
         return util.chunkbuffer(zgenerator(util.filechunkiter(f)))