Mercurial > hg > hg-git

changeset 813:2e77e332f3b3

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
git_handler.update_references: switch to using get_exportable
author Siddharth Agarwal <sid0@fb.com>
date Wed, 29 Oct 2014 10:35:36 -0700
parents 256a232e689d
children 345919960ea4
files hggit/git_handler.py
diffstat 1 files changed, 6 insertions(+), 5 deletions(-) [+]
line wrap: on
line diff
--- a/hggit/git_handler.py
+++ b/hggit/git_handler.py
@@ -1084,14 +1084,15 @@
         return filteredrefs
 
     def update_references(self):
-        heads = self.local_heads()
+        exportable = self.get_exportable()
 
         # Create a local Git branch name for each
         # Mercurial bookmark.
-        for key, (git_ref, hg_sha) in heads.iteritems():
-            git_sha = self.map_git_get(hg_sha)
-            if git_sha:
-                self.git.refs[git_ref] = git_sha
+        for hg_sha, refs in exportable.iteritems():
+            for git_ref in refs.heads:
+                git_sha = self.map_git_get(hg_sha)
+                if git_sha:
+                    self.git.refs[git_ref] = git_sha
 
     def export_hg_tags(self):
         for tag, sha in self.repo.tags().iteritems():