Mercurial > public > src > rhodecode
diff pylons_app/controllers/graph.py @ 106:a86c8de926b4
some fixes in graph tab. Little fixes in files
author | Marcin Kuzminski <marcin@python-works.com> |
---|---|
date | Sun, 25 Apr 2010 23:26:14 +0200 |
parents | aec4c0071cb3 |
children | cf32c4cc26e8 |
line wrap: on
line diff
--- a/pylons_app/controllers/graph.py Sun Apr 25 22:45:54 2010 +0200 +++ b/pylons_app/controllers/graph.py Sun Apr 25 23:26:14 2010 +0200 @@ -1,16 +1,62 @@ +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.base import BaseController, render +from pylons_app.lib.utils import get_repo_slug +from pylons_app.model.hg_model import HgModel import logging -from pylons import request, response, session, tmpl_context as c, url -from pylons.controllers.util import abort, redirect - -from pylons_app.lib.base import BaseController, render log = logging.getLogger(__name__) class GraphController(BaseController): - + def __before__(self): + c.repos_prefix = config['repos_name'] + c.repo_name = get_repo_slug(request) + def index(self): # Return a rendered template - #return render('/graph.mako') - # or, return a string - return 'Hello World' + 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): + from mercurial.graphmod import revisions as graph_rev, colored, CHANGESET + from pylons_app.lib.filters import age as _age, person + from simplejson import dumps + from mercurial.node import short + from webhelpers.paginate import Page + 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) + 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) + print rev_start, rev_end + print 'x' * 100 + 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