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