forked from 00dani/lemoncurry
95 lines
2.3 KiB
Python
95 lines
2.3 KiB
Python
from django.conf import settings
|
|
from django.contrib.sites.models import Site
|
|
from django.contrib.syndication.views import Feed
|
|
from django.urls import reverse
|
|
from django.utils.feedgenerator import Atom1Feed
|
|
from lemoncurry.templatetags.markdown import markdown
|
|
from ..kinds import on_home
|
|
from ..models import Entry
|
|
|
|
|
|
class Atom1FeedWithHub(Atom1Feed):
|
|
def add_root_elements(self, handler):
|
|
super().add_root_elements(handler)
|
|
handler.startElement('link', {'rel': 'hub', 'href': settings.PUSH_HUB})
|
|
handler.endElement('link')
|
|
|
|
|
|
class EntriesFeed(Feed):
|
|
item_guid_is_permalink = True
|
|
|
|
def item_link(self, entry):
|
|
return entry.absolute_url
|
|
|
|
def item_title(self, entry):
|
|
return entry.title
|
|
|
|
def item_description(self, entry):
|
|
return markdown(entry.content)
|
|
|
|
def item_author_name(self, entry):
|
|
return entry.author.name
|
|
|
|
def item_author_email(self, entry):
|
|
return entry.author.email
|
|
|
|
def item_author_link(self, entry):
|
|
return entry.author.absolute_url
|
|
|
|
def item_pubdate(self, entry):
|
|
return entry.published
|
|
|
|
def item_updateddate(self, entry):
|
|
return entry.updated
|
|
|
|
def item_categories(self, entry):
|
|
return (cat.name for cat in entry.cats.all())
|
|
|
|
|
|
class RssByKind(EntriesFeed):
|
|
def get_object(self, request, kind):
|
|
return kind
|
|
|
|
def title(self, kind):
|
|
return "{0} ~ {1}".format(
|
|
kind.plural,
|
|
Site.objects.get_current().name,
|
|
)
|
|
|
|
def link(self, kind):
|
|
return kind.index
|
|
|
|
def description(self, kind):
|
|
return "all {0} at {1}".format(
|
|
kind.plural,
|
|
Site.objects.get_current().name,
|
|
)
|
|
|
|
def items(self, kind):
|
|
return Entry.objects.filter(kind=kind.id)
|
|
|
|
|
|
class AtomByKind(RssByKind):
|
|
feed_type = Atom1FeedWithHub
|
|
subtitle = RssByKind.description
|
|
|
|
|
|
class RssHomeEntries(EntriesFeed):
|
|
def title(self):
|
|
return Site.objects.get_current().name
|
|
|
|
def link(self):
|
|
return reverse('home:index')
|
|
|
|
def description(self):
|
|
return "content from {0}".format(
|
|
Site.objects.get_current().name,
|
|
)
|
|
|
|
def items(self):
|
|
return Entry.objects.filter(kind__in=on_home)
|
|
|
|
|
|
class AtomHomeEntries(RssHomeEntries):
|
|
feed_type = Atom1FeedWithHub
|
|
subtitle = RssHomeEntries.description
|