hgext/interhg.py
changeset 7216 292fb2ad2846
parent 6962 2af657eafeba
child 8225 46293a0c7e9f
--- a/hgext/interhg.py	Wed Oct 22 17:34:50 2008 -0500
+++ b/hgext/interhg.py	Wed Oct 22 17:34:52 2008 -0500
@@ -27,7 +27,7 @@
 
 import re
 from mercurial.hgweb import hgweb_mod
-from mercurial import templatefilters
+from mercurial import templatefilters, extensions
 from mercurial.i18n import _
 
 orig_escape = templatefilters.filters["escape"]
@@ -42,9 +42,7 @@
 
 templatefilters.filters["escape"] = interhg_escape
 
-orig_refresh = hgweb_mod.hgweb.refresh
-
-def interhg_refresh(self):
+def interhg_refresh(orig, self):
     interhg_table[:] = []
     for key, pattern in self.repo.ui.configitems('interhg'):
         # grab the delimiter from the character after the "s"
@@ -79,6 +77,6 @@
         except re.error:
             self.repo.ui.warn(_("interhg: invalid regexp for %s: %s\n")
                               % (key, regexp))
-    return orig_refresh(self)
+    return orig(self)
 
-hgweb_mod.hgweb.refresh = interhg_refresh
+extensions.wrapfunction(hgweb_mod.hgweb, 'refresh', interhg_refresh)