comparison 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
comparison
equal deleted inserted replaced
105:3ac4350b7848 106:a86c8de926b4
1 from pylons import request, response, session, tmpl_context as c, url, config, \
2 app_globals as g
3 from pylons.controllers.util import abort, redirect
4 from pylons_app.lib.base import BaseController, render
5 from pylons_app.lib.utils import get_repo_slug
6 from pylons_app.model.hg_model import HgModel
1 import logging 7 import logging
2 8
3 from pylons import request, response, session, tmpl_context as c, url
4 from pylons.controllers.util import abort, redirect
5
6 from pylons_app.lib.base import BaseController, render
7 9
8 log = logging.getLogger(__name__) 10 log = logging.getLogger(__name__)
9 11
10 class GraphController(BaseController): 12 class GraphController(BaseController):
11 13 def __before__(self):
14 c.repos_prefix = config['repos_name']
15 c.repo_name = get_repo_slug(request)
16
12 def index(self): 17 def index(self):
13 # Return a rendered template 18 # Return a rendered template
14 #return render('/graph.mako') 19 hg_model = HgModel()
15 # or, return a string 20 if request.POST.get('size'):
16 return 'Hello World' 21 c.size = int(request.params.get('size', 20))
22 else:
23 c.size = int(request.params.get('size', 20))
24 c.jsdata, c.canvasheight = self.graph(hg_model.get_repo(c.repo_name), c.size)
25
26 return render('/graph.html')
27
28
29 def graph(self, repo, size):
30 from mercurial.graphmod import revisions as graph_rev, colored, CHANGESET
31 from pylons_app.lib.filters import age as _age, person
32 from simplejson import dumps
33 from mercurial.node import short
34 from webhelpers.paginate import Page
35 revcount = size
36 p = int(request.params.get('page', 1))
37 c.pagination = Page(repo.revisions, page=p, item_count=len(repo.revisions), items_per_page=revcount)
38 max_rev = repo.revisions[-1]
39 offset = 1 if p == 1 else ((p - 1) * revcount)
40 rev_start = repo.revisions[(-1 * offset)]
41 bg_height = 39
42
43 revcount = min(max_rev, revcount)
44 rev_end = max(0, rev_start - revcount)
45 print rev_start, rev_end
46 print 'x' * 100
47 dag = graph_rev(repo.repo, rev_start, rev_end)
48 tree = list(colored(dag))
49 canvasheight = (len(tree) + 1) * bg_height - 27
50 data = []
51 for (id, type, ctx, vtx, edges) in tree:
52 if type != CHANGESET:
53 continue
54 node = short(ctx.node())
55 age = _age(ctx.date())
56 desc = ctx.description()
57 user = person(ctx.user())
58 branch = ctx.branch()
59 branch = branch, repo.repo.branchtags().get(branch) == ctx.node()
60 data.append((node, vtx, edges, desc, user, age, branch, ctx.tags()))
61
62 return dumps(data), canvasheight