diff pylons_app/controllers/changelog.py @ 142:f7218849798a

Changeg graph to changelog, and changelog to shortlog
author Marcin Kuzminski <marcin@python-works.com>
date Sun, 09 May 2010 17:22:30 +0200
parents 20dc7a5eb748
children f191f99f59c9
line wrap: on
line diff
--- a/pylons_app/controllers/changelog.py	Sun May 09 16:51:42 2010 +0200
+++ b/pylons_app/controllers/changelog.py	Sun May 09 17:22:30 2010 +0200
@@ -1,12 +1,18 @@
 import logging
-
-from pylons import tmpl_context as c, app_globals as g, session, request, config, url
+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
+
+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
+        
+        
 
 log = logging.getLogger(__name__)
 
@@ -16,12 +22,44 @@
         c.repo_name = get_repo_slug(request)
         
     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('changelog/changelog.html')
+
+
+    def graph(self, repo, size):
+        revcount = size
         p = int(request.params.get('page', 1))
-        repo = hg_model.get_repo(c.repo_name)
-        c.repo_changesets = Page(repo, page=p, items_per_page=20)
-        c.shortlog_data = render('shortlog/shortlog_data.html')
-        if request.params.get('partial'):
-            return c.shortlog_data
-        r = render('shortlog/shortlog.html')
-        return r
+        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
+