root/trunk/plugins/Atom.py

import cherrypy as cpy
import time
from cgi import escape
from urllib import basejoin as urljoin

from FileCabinet import get_most_recent, get_entries_by_meta
from utils import config, run_callback

class EntryStruct(object):
    def __init__(self):
        self.desc = ''
        self.link = ''
        self.relpath = ''
        self.text = ''
        self.time = ''
        self.title = ''

class Atom(object):
    _cp_config = {"tools.etags.on": True,
        "tools.etags.autotags": True,
        "tools.response_headers.on": True,
        "tools.response_headers.headers": [('Content-Type', 'application/xml')]}

    def __init__(self, parent):
        self.parent = parent

    @cpy.expose
    def index(self):
        datadir = config('datadir')
        num_entries = config('num_entries', 10)
        entries = get_most_recent(datadir, num_entries)
        return self.prepare_atom_template(entries)

    def prepare_atom_template(self, entries):
        ns = cpy.config.get('/').copy()
        entry_structs = []
        last_updated = ''
        for e in entries:
            es = EntryStruct()       
            es.title = e.title

            #this callback gives any interested plugins the chance to change
            #the text of a story, as presented in a feed. It gives an Entry
            #object, and ignores any return value
            run_callback(self.parent.plugins, "cb_feed_story", e)
            fulltext = escape(e.text)

            #If you only want short descriptions:
            #es.desc = escape(e.text[:255])
            #for full text descriptions:
            es.desc = fulltext
            es.text = fulltext
            es.time = time.strftime('%Y-%m-%dT%H:%M:%SZ', e.time_tuple)
            if not last_updated: last_updated = es.time
            es.link = urljoin(config('base_url'), e.relpath + '.html')
            entry_structs.append(es)
        ns['last_updated'] = last_updated
        ns['entries'] = entry_structs
        return ('atom', ns)

    @cpy.expose
    def default(self, *args, **kwargs):
        if args[0].lower().startswith('keyword') and len(args) > 1:
            return self.keyword_atom(args[1])
        else: return self.index()

    def keyword_atom(self, kw):
        num_entries = config('num_entries', 10)
        entries = get_entries_by_meta('keywords')
        entries = [e for e in entries if kw in e.metadata['keywords']]
        return self.prepare_atom_template(entries[:num_entries])