iwla/plugins/display/top_downloads.py

import time

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

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

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

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

        page = display.createPage(title, path, self.iwla.getConfValue('css_path', []))
        table = display.createBlock(DisplayHTMLBlockTable, 'All Downloads', ['URI', 'Hit'])
        table.setColsCSSClass(['', 'iwla_hit'])
        
        total_entrance = [0]*2
        for (uri, entrance) in top_downloads:
            table.appendRow([generateHTMLLink(uri), entrance])
            total_entrance[1] += entrance
        page.appendBlock(table)
                        
        display.addPage(page)

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

        # Top in index
        index = self.iwla.getDisplayIndex()
        
        table = display.createBlock(DisplayHTMLBlockTable, title, ['URI', 'Hits'])
        table.setColsCSSClass(['', 'iwla_hit'])
        for (uri, entrance) in top_downloads[:10]:
            table.appendRow([generateHTMLLink(uri), entrance])
            total_entrance[1] -= entrance
        if total_entrance[1]:
            total_entrance[0] = 'Others'
            table.appendRow(total_entrance)
            table.setCellCSSClass(table.getNbRows()-1, 0, 'iwla_others')
        index.appendBlock(table)