root/trunk/plugins/Keywords.py

import re
import cherrypy as cpy
from urllib import quote_plus, unquote_plus
from FileCabinet import get_entries_by_meta
from utils import config

#find meta info in entries
META_RE = re.compile('^<!-- ?([ \w]+): ?([\w ,.:\/-]+)-->')

def keysplit(str):
    try:
        return [s.strip() for s in str.split(',')]
    except:
        return [str]

#TODO: add + for keyword combination support!
class Keywords(object):
    def __init__(self, parent):
        self.keyword = ''
        self.atom_link = ''
        self.parent = parent

    @cpy.expose
    def default(self, *args, **kargs):
        if len(args) > 1:
            return self.parent.error_page("too many arguments to Keywords")
        elif len(args) < 1:
            return self.parent.error_page("Too few arguments to Keywords")

        try:
            offset = int(kargs.get('offset', 0))
        except ValueError:
            offset = 0
        
        self.keyword = unquote_plus(args[0])
        print "Key =  %s" % self.keyword
        
        #remember we're not sure if base_url has a trailing '/' or not...
        if 'Atom' in config("plugins"):
            self.atom_link = config('base_url').rstrip('/') + \
                '/Atom/keywords/' + self.keyword
        
        entries = get_entries_by_meta('keywords')
        entries = [e for e in entries if self.keyword in keysplit(e.metadata['keywords'])]
        entries = entries[offset:offset + config("num_entries")]
        return self.parent.render_page(entries, self.keyword, offset)

    def cb_add_data(self):
        if self.atom_link:
            return {'keywordatom': self.atom_link, 'keyword': self.keyword}

    def cb_story(self, entry):
        """Add a $keywords variable to an entry which is a linkified,
            comma-seperated string"""
        kwstring = ''
        base_url = config('base_url')
        base_url = base_url.rstrip('/')
        #list comp for 2.3 compatibility
        kws = [k.strip() for k in keysplit(entry.metadata.get('keywords', "")) if k != '']
        links = ['<a href=%s/Keywords/%s>%s</a>' % (base_url, quote_plus(kw), kw) 
                    for kw in kws]
        #add a comma seperated list of keywords to the entry
        entry.keywords = ', '.join(links)

    def cb_page_end(self):
        """clean up our data"""
        self.keyword = ''
        self.atom_link = ''