diff hgext/mq.py @ 10770:fe39f0160c74 stable

localrepo: change _updatebranchcache to use a context generator
author Sune Foldager <cryo@cyanite.org>
date Fri, 26 Mar 2010 17:02:23 +0100
parents 32023a0a389b
children bd36e5c0ccb1 ce2ae8bf3ae3 1ebba857b371
line wrap: on
line diff
--- a/hgext/mq.py	Wed Mar 24 19:44:30 2010 +0100
+++ b/hgext/mq.py	Fri Mar 26 17:02:23 2010 +0100
@@ -2603,7 +2603,8 @@
             start = lrev + 1
             if start < qbase:
                 # update the cache (excluding the patches) and save it
-                self._updatebranchcache(partial, lrev + 1, qbase)
+                ctxgen = (self[r] for r in xrange(lrev + 1, qbase))
+                self._updatebranchcache(partial, ctxgen)
                 self._writebranchcache(partial, cl.node(qbase - 1), qbase - 1)
                 start = qbase
             # if start = qbase, the cache is as updated as it should be.
@@ -2611,7 +2612,8 @@
             # we might as well use it, but we won't save it.
 
             # update the cache up to the tip
-            self._updatebranchcache(partial, start, len(cl))
+            ctxgen = (self[r] for r in xrange(start, len(cl)))
+            self._updatebranchcache(partial, ctxgen)
 
             return partial