convert: add bookmark support to the hg sink
authorEdouard Gomez <ed.gomez@free.fr>
Wed, 07 Jul 2010 00:06:59 +0200
changeset 13746 d80b768545cb
parent 13745 9ff22f600c6c
child 13747 cede00420e1e
convert: add bookmark support to the hg sink
hgext/convert/hg.py
--- a/hgext/convert/hg.py	Wed Jul 07 00:06:59 2010 +0200
+++ b/hgext/convert/hg.py	Wed Jul 07 00:06:59 2010 +0200
@@ -21,7 +21,7 @@
 import os, time, cStringIO
 from mercurial.i18n import _
 from mercurial.node import bin, hex, nullid
-from mercurial import hg, util, context, error
+from mercurial import hg, util, context, bookmarks, error
 
 from common import NoRepo, commit, converter_source, converter_sink
 
@@ -214,6 +214,16 @@
     def setfilemapmode(self, active):
         self.filemapmode = active
 
+    def putbookmarks(self, updatedbookmark):
+        if not len(updatedbookmark):
+            return
+
+        self.ui.status(_("updating bookmarks\n"))
+        for bookmark in updatedbookmark:
+            self.repo._bookmarks[bookmark] = bin(updatedbookmark[bookmark])
+            bookmarks.write(self.repo)
+
+
 class mercurial_source(converter_source):
     def __init__(self, ui, path, rev=None):
         converter_source.__init__(self, ui, path, rev)