|
''' Mercurial revision DAG visualization library
Implements a gtk.TreeModel which visualizes a Mercurial repository
revision history.
Portions of this code stolen mercilessly from bzr-gtk visualization
dialog. Other portions stolen from graphlog extension.
'''
import gtk
import gobject
import re
from time import (strftime, gmtime)
from mercurial import util
from mercurial.node import short
from mercurial.hgweb import webutil
# FIXME: dirty hack to import toutf() from hggtk.hglib.
#
# Python 2.5's relative imports doesn't seem to work either,
# when running history.py directly.
#
# Besides, we want to be compatible older Python versions.
try:
# when history.py is invoked directly
from hglib import toutf
except ImportError:
# when history.py is imported and called from hgproc.py
from hggtk.hglib import toutf
# treemodel row enumerated attributes
LINES = 0
NODE = 1
REVID = 2
LAST_LINES = 3
MESSAGE = 4
COMMITER = 5
TIMESTAMP = 6
REVISION = 7
PARENTS = 8
WCPARENT = 9
HEAD = 10
TAGS = 11
MARKED = 12
FGCOLOR = 13
HEXID = 14
class TreeModel(gtk.GenericTreeModel):
def __init__ (self, repo, graphdata, color_func):
gtk.GenericTreeModel.__init__(self)
self.revisions = {}
self.repo = repo
self.parents = [x.rev() for x in repo.changectx(None).parents()]
self.heads = [repo.changelog.rev(x) for x in repo.heads()]
self.line_graph_data = graphdata
self.author_re = re.compile('<.*@.*>', 0)
self.color_func = color_func
def on_get_flags(self):
return gtk.TREE_MODEL_LIST_ONLY
def on_get_n_columns(self):
return 14
def on_get_column_type(self, index):
if index == NODE: return gobject.TYPE_PYOBJECT
if index == LINES: return gobject.TYPE_PYOBJECT
if index == REVID: return gobject.TYPE_STRING
if index == LAST_LINES: return gobject.TYPE_PYOBJECT
if index == MESSAGE: return gobject.TYPE_STRING
if index == COMMITER: return gobject.TYPE_STRING
if index == TIMESTAMP: return gobject.TYPE_STRING
if index == REVISION: return gobject.TYPE_PYOBJECT
if index == PARENTS: return gobject.TYPE_PYOBJECT
if index == WCPARENT: return gobject.TYPE_BOOLEAN
if index == HEAD: return gobject.TYPE_BOOLEAN
if index == TAGS: return gobject.TYPE_STRING
if index == MARKED: return gobject.TYPE_BOOLEAN
if index == FGCOLOR: return gobject.TYPE_STRING
if index == HEXID: return gobject.TYPE_STRING
def on_get_iter(self, path):
return path[0]
def on_get_path(self, rowref):
return rowref
def on_get_value(self, rowref, column):
(revid, node, lines, parents) = self.line_graph_data[rowref]
if column == REVID: return revid
if column == NODE: return node
if column == LINES: return lines
if column == PARENTS: return parents
if column == LAST_LINES:
if rowref>0:
return self.line_graph_data[rowref-1][2]
return []
if revid not in self.revisions:
ctx = self.repo.changectx(revid)
summary = ctx.description().replace('\0', '')
summary = summary.split('\n')[0]
summary = gobject.markup_escape_text(toutf(summary))
node = self.repo.lookup(revid)
tags = self.repo.nodetags(node)
taglist = toutf(', '.join(tags))
for tag in tags:
summary = '<span background="#ffffaa"> %s </span> %s' % (
tag, summary)
# show branch names per hgweb's logic
branches = webutil.nodebranchdict(self.repo, ctx)
inbranches = webutil.nodeinbranch(self.repo, ctx)
for branch in branches:
summary = '<span background="#aaffaa"> %s </span> %s' % (
branch['name'], summary)
for branch in inbranches:
summary = '<span background="#d5dde6"> %s </span> %s' % (
branch['name'], summary)
if '<' in ctx.user():
author = toutf(self.author_re.sub('', ctx.user()).strip(' '))
else:
author = toutf(util.shortuser(ctx.user()))
date = strftime("%Y-%m-%d %H:%M:%S", gmtime(ctx.date()[0]))
wc_parent = revid in self.parents
head = revid in self.heads
color = self.color_func(parents, revid, author)
revision = (None, node, revid, None, summary,
author, date, None, parents, wc_parent, head, taglist,
None, color, short(node))
self.revisions[revid] = revision
else:
revision = self.revisions[revid]
if column == REVISION:
return revision
if column == MARKED:
return revid == self.marked_rev
return revision[column]
def on_iter_next(self, rowref):
if rowref < len(self.line_graph_data) - 1:
return rowref+1
return None
def on_iter_children(self, parent):
if parent is None: return 0
return None
def on_iter_has_child(self, rowref):
return False
def on_iter_n_children(self, rowref):
if rowref is None: return len(self.line_graph_data)
return 0
def on_iter_nth_child(self, parent, n):
if parent is None: return n
return None
def on_iter_parent(self, child):
return None
|
Loading...