iwla/plugins/display/top_hits.py

import time

from iwla import IWLA
from iplugin import IPlugin
from display import *

class IWLADisplayTopHits(IPlugin):
    def __init__(self, iwla):
        super(IWLADisplayTopHits, self).__init__(iwla)
        self.API_VERSION = 1
        self.requires = ['IWLAPostAnalysisTopHits']

    def hook(self):
        top_hits = self.iwla.getMonthStats()['top_hits']
        top_hits = sorted(top_hits.items(), key=lambda t: t[1], reverse=True)

        # All in a file
        title = time.strftime('All Hits - %B %Y', self.iwla.getCurTime())
        filename = 'top_hits.html'
        path = self.iwla.getCurDisplayPath(filename)

        page = DisplayHTMLPage(title, path, self.iwla.getConfValue('css_path', []))
        table = DisplayHTMLBlockTable('All Hits', ['URI', 'Entrance'])        
        table.setColsCSSClass(['', 'iwla_hit'])
        for (uri, entrance) in top_hits:
            table.appendRow([uri, entrance])
        page.appendBlock(table)
                        
        self.iwla.getDisplay().addPage(page)

        link = '<a href=\'%s\'>All hits</a>' % (filename)
        title = '%s - %s' % ('Top Hits', link)

        # Top in index
        index = self.iwla.getDisplayIndex()
        
        table = DisplayHTMLBlockTable(title, ['URI', 'Entrance'])        
        table.setColsCSSClass(['', 'iwla_hit'])
        for (uri, entrance) in top_hits[:10]:
            table.appendRow([uri, entrance])
        index.appendBlock(table)