changeset 2199:f1986a61ccff

move stringio class from bugzilla extension into templater module.
author Vadim Gelfer <vadim.gelfer@gmail.com>
date Thu, 04 May 2006 12:02:32 -0700
parents 564034552f7f
children 9f43b6e24232
files hgext/bugzilla.py mercurial/templater.py
diffstat 2 files changed, 16 insertions(+), 16 deletions(-) [+]
line wrap: on
line diff
--- a/hgext/bugzilla.py	Thu May 04 12:00:45 2006 -0700
+++ b/hgext/bugzilla.py	Thu May 04 12:02:32 2006 -0700
@@ -43,7 +43,7 @@
 from mercurial.demandload import *
 from mercurial.i18n import gettext as _
 from mercurial.node import *
-demandload(globals(), 'cStringIO mercurial:templater,util os re time')
+demandload(globals(), 'mercurial:templater,util os re time')
 
 try:
     import MySQLdb
@@ -241,23 +241,9 @@
                 count -= 1
             return root
 
-        class stringio(object):
-            '''wrap cStringIO.'''
-            def __init__(self):
-                self.fp = cStringIO.StringIO()
-
-            def write(self, *args):
-                for a in args:
-                    self.fp.write(a)
-
-            write_header = write
-
-            def getvalue(self):
-                return self.fp.getvalue()
-
         mapfile = self.ui.config('bugzilla', 'style')
         tmpl = self.ui.config('bugzilla', 'template')
-        sio = stringio()
+        sio = templater.stringio()
         t = templater.changeset_templater(self.ui, self.repo, mapfile, sio)
         if not mapfile and not tmpl:
             tmpl = _('changeset {node|short} in repo {root} refers '
--- a/mercurial/templater.py	Thu May 04 12:00:45 2006 -0700
+++ b/mercurial/templater.py	Thu May 04 12:02:32 2006 -0700
@@ -499,3 +499,17 @@
                                                            inst.args[0]))
         except SyntaxError, inst:
             raise util.Abort(_('%s: %s') % (self.t.mapfile, inst.args[0]))
+
+class stringio(object):
+    '''wrap cStringIO for use by changeset_templater.'''
+    def __init__(self):
+        self.fp = cStringIO.StringIO()
+
+    def write(self, *args):
+        for a in args:
+            self.fp.write(a)
+
+    write_header = write
+
+    def getvalue(self):
+        return self.fp.getvalue()