Changeset 193

Timestamp:

08/18/08 21:16:35 (11 months ago)

Author:

rgrp

Message:

[shakespeare/stats][m]: add charting to support to text stats page using pygooglechart.

Location:

trunk

Files:

• setup.py (modified) (1 diff)
• shakespeare/controllers/stats.py (modified) (2 diffs)
• shakespeare/templates/layout.html (modified) (1 diff)
• shakespeare/templates/stats/text.html (modified) (2 diffs)

trunk/setup.py

@@ -21 +21 @@
         'SQLObject>=0.6',
         'Genshi>=0.3',
-        'annotater>=0.1',
+        'pygooglechart>=0.2,<0.3',
+        # 'annotater>=0.1',
         ],
     test_suite='nose.collector',

trunk/shakespeare/controllers/stats.py

@@ -1 +1 @@
 import logging
 
+import pygooglechart
+
 from shakespeare.lib.base import *
-
 log = logging.getLogger(__name__)
-
 import shakespeare.stats
 
@@ -18 +18 @@
         c.text = text
         c.stats = stats.text_stats(text)
+        # 40 seems limit for google
+        data = [ (s.word, s.freq) for s in c.stats[:40] ]
+        c.img_url = self.vertical_bar_chart(data)
         return render('stats/text')
 
+    # TODO: factor this out to its module (?)
+    def vertical_bar_chart(self, data, width=500):
+        # tranpose
+        tdata = zip(*data)
+        labels = list(tdata[0])
+        values = tdata[1]
+        bar_width = 10
+        # add 5 for space between bars
+        height = (bar_width + 5) * len(values)
+        # was setting x_range but automatic behaviour seems better
+        # x_range = (min(values), max(values))
+        chart = pygooglechart.StackedHorizontalBarChart(width, height)
+        chart.set_bar_width(bar_width)
+        chart.set_colours(['cc0033'])
+        chart.add_data(values)
+        # have to reverse the labels for vertical
+        labels.reverse()
+        chart.set_axis_labels(pygooglechart.Axis.LEFT, labels)
+        chart.set_axis_range(pygooglechart.Axis.BOTTOM, 0, max(values))
+        chart.set_axis_range(pygooglechart.Axis.TOP, 0, max(values))
+        url = chart.get_url()
+        return url
+

trunk/shakespeare/templates/layout.html

@@ -59 +59 @@
       <li><a href="${h.url_for(controller='text', action='index')}">Texts</a></li>
       <li><a href="${h.url_for(controller='search', action='index')}">Search</a></li>
+      <li><a href="${h.url_for(controller='stats', action='index')}">Stats</a></li>
       <li><a href="${h.url_for(controller='site', action='guide')}">Guide</a></li>
     </ul>

trunk/shakespeare/templates/stats/text.html

@@ -2 +2 @@
   xmlns:xi="http://www.w3.org/2001/XInclude">
   
-  <py:def function="page_title">Stats for ${c.text.name}</py:def>
+  <py:def function="page_title">Stats for ${c.text.title}</py:def>
 
   <div py:match="content">
-    <table border="1">
+    <img style="float: left;"
+      src="${c.img_url}"
+      alt="Word Statistics Bar Chart" />
+
+    <table border="1" style="margin-left: 550px;">
     <thead>
       <tr>
+        <th>Index</th>
         <th>
           Word  
@@ -17 +22 @@
     </thead>
     <tbody>
-      <tr py:for="stat in c.stats">
+      <tr py:for="index, stat in enumerate(c.stats)">
+        <td>
+          ${index + 1}
+        </td>
         <td>
           ${stat.word}