168 lines
4.4 KiB
Python
168 lines
4.4 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.contrib.syndication import views
|
|
from django.utils import feedgenerator
|
|
from django.utils.timezone import get_fixed_timezone
|
|
|
|
from .models import Article, Entry
|
|
|
|
|
|
class TestRss2Feed(views.Feed):
|
|
title = 'My blog'
|
|
description = 'A more thorough description of my blog.'
|
|
link = '/blog/'
|
|
feed_guid = '/foo/bar/1234'
|
|
author_name = 'Sally Smith'
|
|
author_email = 'test@example.com'
|
|
author_link = 'http://www.example.com/'
|
|
categories = ('python', 'django')
|
|
feed_copyright = 'Copyright (c) 2007, Sally Smith'
|
|
ttl = 600
|
|
|
|
def items(self):
|
|
return Entry.objects.all()
|
|
|
|
def item_description(self, item):
|
|
return "Overridden description: %s" % item
|
|
|
|
def item_pubdate(self, item):
|
|
return item.published
|
|
|
|
def item_updateddate(self, item):
|
|
return item.updated
|
|
|
|
item_author_name = 'Sally Smith'
|
|
item_author_email = 'test@example.com'
|
|
item_author_link = 'http://www.example.com/'
|
|
item_categories = ('python', 'testing')
|
|
item_copyright = 'Copyright (c) 2007, Sally Smith'
|
|
|
|
|
|
class TestRss2FeedWithGuidIsPermaLinkTrue(TestRss2Feed):
|
|
def item_guid_is_permalink(self, item):
|
|
return True
|
|
|
|
|
|
class TestRss2FeedWithGuidIsPermaLinkFalse(TestRss2Feed):
|
|
def item_guid(self, item):
|
|
return str(item.pk)
|
|
|
|
def item_guid_is_permalink(self, item):
|
|
return False
|
|
|
|
|
|
class TestRss091Feed(TestRss2Feed):
|
|
feed_type = feedgenerator.RssUserland091Feed
|
|
|
|
|
|
class TestNoPubdateFeed(views.Feed):
|
|
title = 'Test feed'
|
|
link = '/feed/'
|
|
|
|
def items(self):
|
|
return Entry.objects.all()
|
|
|
|
|
|
class TestAtomFeed(TestRss2Feed):
|
|
feed_type = feedgenerator.Atom1Feed
|
|
subtitle = TestRss2Feed.description
|
|
|
|
|
|
class TestLatestFeed(TestRss2Feed):
|
|
"""
|
|
A feed where the latest entry date is an `updated` element.
|
|
"""
|
|
feed_type = feedgenerator.Atom1Feed
|
|
subtitle = TestRss2Feed.description
|
|
|
|
def items(self):
|
|
return Entry.objects.exclude(pk=5)
|
|
|
|
|
|
class ArticlesFeed(TestRss2Feed):
|
|
"""
|
|
A feed to test no link being defined. Articles have no get_absolute_url()
|
|
method, and item_link() is not defined.
|
|
"""
|
|
def items(self):
|
|
return Article.objects.all()
|
|
|
|
|
|
class TestEnclosureFeed(TestRss2Feed):
|
|
pass
|
|
|
|
|
|
class TemplateFeed(TestRss2Feed):
|
|
"""
|
|
A feed to test defining item titles and descriptions with templates.
|
|
"""
|
|
title_template = 'syndication/title.html'
|
|
description_template = 'syndication/description.html'
|
|
|
|
# Defining a template overrides any item_title definition
|
|
def item_title(self):
|
|
return "Not in a template"
|
|
|
|
|
|
class TemplateContextFeed(TestRss2Feed):
|
|
"""
|
|
A feed to test custom context data in templates for title or description.
|
|
"""
|
|
title_template = 'syndication/title_context.html'
|
|
description_template = 'syndication/description_context.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(TemplateContextFeed, self).get_context_data(**kwargs)
|
|
context['foo'] = 'bar'
|
|
return context
|
|
|
|
|
|
class NaiveDatesFeed(TestAtomFeed):
|
|
"""
|
|
A feed with naive (non-timezone-aware) dates.
|
|
"""
|
|
def item_pubdate(self, item):
|
|
return item.published
|
|
|
|
|
|
class TZAwareDatesFeed(TestAtomFeed):
|
|
"""
|
|
A feed with timezone-aware dates.
|
|
"""
|
|
def item_pubdate(self, item):
|
|
# Provide a weird offset so that the test can know it's getting this
|
|
# specific offset and not accidentally getting on from
|
|
# settings.TIME_ZONE.
|
|
return item.published.replace(tzinfo=get_fixed_timezone(42))
|
|
|
|
|
|
class TestFeedUrlFeed(TestAtomFeed):
|
|
feed_url = 'http://example.com/customfeedurl/'
|
|
|
|
|
|
class MyCustomAtom1Feed(feedgenerator.Atom1Feed):
|
|
"""
|
|
Test of a custom feed generator class.
|
|
"""
|
|
def root_attributes(self):
|
|
attrs = super(MyCustomAtom1Feed, self).root_attributes()
|
|
attrs['django'] = 'rocks'
|
|
return attrs
|
|
|
|
def add_root_elements(self, handler):
|
|
super(MyCustomAtom1Feed, self).add_root_elements(handler)
|
|
handler.addQuickElement('spam', 'eggs')
|
|
|
|
def item_attributes(self, item):
|
|
attrs = super(MyCustomAtom1Feed, self).item_attributes(item)
|
|
attrs['bacon'] = 'yum'
|
|
return attrs
|
|
|
|
def add_item_elements(self, handler, item):
|
|
super(MyCustomAtom1Feed, self).add_item_elements(handler, item)
|
|
handler.addQuickElement('ministry', 'silly walks')
|
|
|
|
|
|
class TestCustomFeed(TestAtomFeed):
|
|
feed_type = MyCustomAtom1Feed
|