changeset 5352:f0931c0240b4

convert: add before/after hooks for converter sources
author Bryan O'Sullivan <bos@serpentine.com>
date Tue, 02 Oct 2007 13:46:17 -0700
parents cc34be74eeec
children c6adf2be6069
files hgext/convert/__init__.py hgext/convert/common.py
diffstat 2 files changed, 11 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/hgext/convert/__init__.py
+++ b/hgext/convert/__init__.py
@@ -235,6 +235,7 @@ class converter(object):
 
     def convert(self):
         try:
+            self.source.before()
             self.dest.before()
             self.source.setrevmap(self.map)
             self.ui.status("scanning source...\n")
@@ -273,7 +274,10 @@ class converter(object):
             self.cleanup()
 
     def cleanup(self):
-        self.dest.after()
+        try:
+            self.dest.after()
+        finally:
+            self.source.after()
         if self.revmapfilefd:
             self.revmapfilefd.close()
 
--- a/hgext/convert/common.py
+++ b/hgext/convert/common.py
@@ -38,6 +38,12 @@ class converter_source(object):
 
         self.encoding = 'utf-8'
 
+    def before(self):
+        pass
+
+    def after(self):
+        pass
+
     def setrevmap(self, revmap):
         """set the map of already-converted revisions"""
         pass