from django.contrib.syndication.views import Feed as RSSFeed from django.contrib.sites.shortcuts import get_current_site from django.urls import reverse from django.utils.feedgenerator import Rss201rev2Feed from tp.settings import IMAGES_URL, MP3_URL from .models import Feed from datetime import datetime from .rss_utils import get_rss_item_desc, get_feed_image_desc class AudioRssFeedGenerator(Rss201rev2Feed): content_type = 'application/xml; charset=utf-8' def add_root_elements(self, handler): super().add_root_elements(handler) handler.startElement("image", {}) handler.addQuickElement("url", self.feed['image_url']) handler.addQuickElement("title", self.feed['image_title']) handler.addQuickElement("link", self.feed['image_link']) handler.addQuickElement("description", self.feed['image_desc']) handler.endElement("image") def add_item_elements(self, handler, item): super().add_item_elements(handler, item) handler.startElement("image", {}) handler.addQuickElement("url", item['image_url']) handler.addQuickElement("title", item['image_title']) handler.addQuickElement("link", item['image_link']) handler.addQuickElement("description", item['image_desc']) handler.endElement("image") class AudioRssFeed(RSSFeed): feed_type = AudioRssFeedGenerator def get_object(self, request, slug): obj = Feed.objects.get(slug=slug) obj.request = request return obj def items(self, obj): xr = [x for x in obj.episode_set.order_by('pub_date')] for x in xr: x.request = obj.request return xr def item_enclosure_url(self, item): return f'{MP3_URL}{item.mp3}' def item_enclosure_length(self, item): return(item.mp3.size) def item_enclosure_mime_type(self, item): return "audio/mpeg" def item_pubdate(self, item): ''' Need to return datetime.datetime object, but item.pub_date is an datetime.date object ''' return datetime.fromisoformat(item.pub_date.isoformat()) def link(self, obj): return reverse('audio:feed', kwargs={'pk': obj.pk, 'slug': obj.slug}) def title(self, obj): return obj.title def description(self, obj): return obj.description def item_link(self, item): return reverse('audio:episode', kwargs={'pk': item.pk, 'slug': item.slug}) def item_title(self, item): return f'{item.episode_number}: {item.title}' def item_description(self, item): return get_rss_item_desc(item) def item_extra_kwargs(self, item): x = {} x['image_url'] = f'{IMAGES_URL}{item.image.name}' x['image_title'] = item.title x['image_link'] = f'{get_current_site(item.request)}{self.item_link(item)}' x['image_desc'] = f'Image for: {item.title}' return x def feed_extra_kwargs(self, obj): x = {} x['image_url'] = f'{IMAGES_URL}{obj.image.name}' x['image_title'] = obj.title x['image_link'] = f'{get_current_site(obj.request)}{self.link(obj)}' x['image_desc'] = get_feed_image_desc(obj) return x