from mercurial.graphmod import revisions as graph_rev, colored, CHANGESET from mercurial.node import short from pylons import request, response, session, tmpl_context as c, url, config, \ app_globals as g from pylons.controllers.util import abort, redirect from pylons_app.lib.auth import LoginRequired from pylons_app.lib.base import BaseController, render from pylons_app.lib.filters import age as _age, person from pylons_app.lib.utils import get_repo_slug from pylons_app.model.hg_model import HgModel from simplejson import dumps from webhelpers.paginate import Page import logging log = logging.getLogger(__name__) class GraphController(BaseController): @LoginRequired() def __before__(self): super(GraphController, self).__before__() def index(self): # Return a rendered template hg_model = HgModel() if request.POST.get('size'): c.size = int(request.params.get('size', 20)) else: c.size = int(request.params.get('size', 20)) c.jsdata, c.canvasheight = self.graph(hg_model.get_repo(c.repo_name), c.size) return render('/graph.html') def graph(self, repo, size): revcount = size p = int(request.params.get('page', 1)) c.pagination = Page(repo.revisions, page=p, item_count=len(repo.revisions), items_per_page=revcount) if not repo.revisions:return dumps([]), 0 max_rev = repo.revisions[-1] offset = 1 if p == 1 else ((p - 1) * revcount) rev_start = repo.revisions[(-1 * offset)] bg_height = 39 revcount = min(max_rev, revcount) rev_end = max(0, rev_start - revcount) dag = graph_rev(repo.repo, rev_start, rev_end) tree = list(colored(dag)) canvasheight = (len(tree) + 1) * bg_height - 27 data = [] for (id, type, ctx, vtx, edges) in tree: if type != CHANGESET: continue node = short(ctx.node()) age = _age(ctx.date()) desc = ctx.description() user = person(ctx.user()) branch = ctx.branch() branch = branch, repo.repo.branchtags().get(branch) == ctx.node() data.append((node, vtx, edges, desc, user, age, branch, ctx.tags())) return dumps(data), canvasheight