mercurial-scm/hg: mercurial/cmdutil.py comparison

comparison mercurial/cmdutil.py @ 3645:b984dcb1df71

Refactor log ui buffering and patch display

author	Matt Mackall <mpm@selenic.com>
date	Mon, 13 Nov 2006 13:26:57 -0600
parents	b4ad640a3bcf
children	2801a3efc7c3

comparison

equal deleted inserted replaced

-:b7547efe78fb
+:b984dcb1df71
 from demandload import demandload
 from node import *
 from i18n import gettext as _
 demandload(globals(), 'os sys')
-demandload(globals(), 'mdiff util templater cStringIO')
+demandload(globals(), 'mdiff util templater cStringIO patch')
 revrangesep = ':'
 def revpair(ui, repo, revs):
 '''return pair of nodes, given list of revisions. second item can
 '(%d%% similar)\n') %
 (oldrel, newrel, score * 100))
 if not dry_run:
 repo.copy(old, new, wlock=wlock)
+class uibuffer(object):
+# Implement and delegate some ui protocol.  Save hunks of
+# output for later display in the desired order.
+def __init__(self, ui):
+self.ui = ui
+self.hunk = {}
+self.header = {}
+self.quiet = ui.quiet
+self.verbose = ui.verbose
+self.debugflag = ui.debugflag
+self.lastheader = None
+def note(self, *args):
+if self.verbose:
+self.write(*args)
+def status(self, *args):
+if not self.quiet:
+self.write(*args)
+def debug(self, *args):
+if self.debugflag:
+self.write(*args)
+def write(self, *args):
+self.hunk.setdefault(self.rev, []).extend(args)
+def write_header(self, *args):
+self.header.setdefault(self.rev, []).extend(args)
+def mark(self, rev):
+self.rev = rev
+def flush(self, rev):
+if rev in self.header:
+h = "".join(self.header[rev])
+if h != self.lastheader:
+self.lastheader = h
+self.ui.write(h)
+del self.header[rev]
+if rev in self.hunk:
+self.ui.write("".join(self.hunk[rev]))
+del self.hunk[rev]
+return 1
+return 0
 class changeset_printer(object):
 '''show changeset information when templating not requested.'''
-def __init__(self, ui, repo):
+def __init__(self, ui, repo, patch, buffered):
 self.ui = ui
 self.repo = repo
+self.buffered = buffered
+self.patch = patch
+if buffered:
+self.ui = uibuffer(ui)
+def flush(self, rev):
+return self.ui.flush(rev)
 def show(self, rev=0, changenode=None, brinfo=None, copies=None):
 '''show a single changeset or file revision'''
+if self.buffered:
+self.ui.mark(rev)
 log = self.repo.changelog
 if changenode is None:
 changenode = log.node(rev)
 elif not rev:
 rev = log.rev(changenode)
 else:
 self.ui.write(_("summary:     %s\n") %
 description.splitlines()[0])
 self.ui.write("\n")
-class changeset_templater(object):
+self.showpatch(changenode)
+def showpatch(self, node):
+if self.patch:
+prev = self.repo.changelog.parents(node)[0]
+patch.diff(self.repo, prev, node, fp=self.ui)
+self.ui.write("\n")
+class changeset_templater(changeset_printer):
 '''format changeset information.'''
-def __init__(self, ui, repo, mapfile, dest=None):
+def __init__(self, ui, repo, patch, mapfile, buffered):
+changeset_printer.__init__(self, ui, repo, patch, buffered)
 self.t = templater.templater(mapfile, templater.common_filters,
 cache={'parent': '{rev}:{node|short} ',
 'manifest': '{rev}:{node|short}',
 'filecopy': '{name} ({source})'})
-self.ui = ui
-self.dest = dest
-self.repo = repo
 def use_template(self, t):
 '''set template string to use'''
 self.t.cache['changeset'] = t
 def show(self, rev=0, changenode=None, brinfo=None, copies=[], **props):
 '''show a single changeset or file revision'''
+if self.buffered:
+self.ui.mark(rev)
 log = self.repo.changelog
 if changenode is None:
 changenode = log.node(rev)
 elif not rev:
 rev = log.rev(changenode)
 }
 props = props.copy()
 props.update(defprops)
 try:
-dest = self.dest or self.ui
 if self.ui.debugflag and 'header_debug' in self.t:
 key = 'header_debug'
 elif self.ui.quiet and 'header_quiet' in self.t:
 key = 'header_quiet'
 elif self.ui.verbose and 'header_verbose' in self.t:
 elif 'header' in self.t:
 key = 'header'
 else:
 key = ''
 if key:
-dest.write_header(templater.stringify(self.t(key, **props)))
+h = templater.stringify(self.t(key, **props))
+if self.buffered:
+self.ui.write_header(h)
+else:
+self.ui.write(h)
 if self.ui.debugflag and 'changeset_debug' in self.t:
 key = 'changeset_debug'
 elif self.ui.quiet and 'changeset_quiet' in self.t:
 key = 'changeset_quiet'
 elif self.ui.verbose and 'changeset_verbose' in self.t:
 key = 'changeset_verbose'
 else:
 key = 'changeset'
-dest.write(templater.stringify(self.t(key, **props)))
+self.ui.write(templater.stringify(self.t(key, **props)))
+self.showpatch(changenode)
 except KeyError, inst:
 raise util.Abort(_("%s: no key named '%s'") % (self.t.mapfile,
 inst.args[0]))
 except SyntaxError, inst:
 raise util.Abort(_('%s: %s') % (self.t.mapfile, inst.args[0]))
 write_header = write
 def __getattr__(self, key):
 return getattr(self.fp, key)
-def show_changeset(ui, repo, opts):
+def show_changeset(ui, repo, opts, buffered=False):
 """show one changeset using template or regular display.
 Display format will be the first non-empty hit of:
 1. option 'template'
 2. option 'style'
 4. [ui] setting 'style'
 If all of these values are either the unset or the empty string,
 regular display via changeset_printer() is done.
 """
 # options
+patch = opts.get('patch')
 tmpl = opts.get('template')
 mapfile = None
 if tmpl:
 tmpl = templater.parsestring(tmpl, quoted=False)
 else:
 if not os.path.split(mapfile)[0]:
 mapname = (templater.templatepath('map-cmdline.' + mapfile)
 or templater.templatepath(mapfile))
 if mapname: mapfile = mapname
 try:
-t = changeset_templater(ui, repo, mapfile)
+t = changeset_templater(ui, repo, patch, mapfile, buffered)
 except SyntaxError, inst:
 raise util.Abort(inst.args[0])
 if tmpl: t.use_template(tmpl)
 return t
-return changeset_printer(ui, repo)
+return changeset_printer(ui, repo, patch, buffered)

Mercurial > public > mercurial-scm > hg

comparison mercurial/cmdutil.py @ 3645:b984dcb1df71