from django.contrib.syndication.views import Feed
from django.utils.feedgenerator import Rss201rev2Feed

from mhackspace.feeds.models import Article


class MediaRssFeed(Rss201rev2Feed):
    """
    Implement thumbnails which adhere to Yahoo Media RSS (mrss) feed.
  
    @see http://djangosnippets.org/snippets/1648/
    """
    def rss_attributes(self):
        attrs = super(MediaRssFeed, self).rss_attributes()
        attrs['xmlns:dc'] = "http://purl.org/dc/elements/1.1/"
        attrs['xmlns:media'] = 'http://search.yahoo.com/mrss/'
        return attrs

    def add_item_elements(self, handler, item):
        super(MediaRssFeed, self).add_item_elements(handler, item)

        thumbnail = {'url': item['thumbnail_url']}

        if 'thumbnail_width' in item:
            thumbnail['width'] = str(item['thumbnail_width'])

        if 'thumbnail_height' in item:
            thumbnail['height'] = str(item['thumbnail_height'])

        handler.addQuickElement(u"media:thumbnail", '', thumbnail)


class LatestEntriesFeed(Feed):
    title = "Maidstone hackspace site news"
    link = "/latest/"
    description = "Latest creations from our users."

    def items(self):
        return Article.objects.all()[:5]

    def item_title(self, item):
        return item.title

    def item_description(self, item):
        return item.description

    def item_link(self, item):
        return item.url