Thelie/python-feedgen
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Enable Multiple media:content/media:group Elements

Browse source 
One item can have multiple media:content elements which may be located
in multiple media:group element to indicate that content is the same but
for the format.

This patch adds the ability to add multiple content elements and define
the group to which they go belong.

If no group is specified, all elements are located in a `default` group.

If the group is set to None, the content element is directly attached to
the item element.

Part of #58
This commit is contained in:
Lars Kiesow 2017-05-21 22:06:50 +02:00
parent a7ae36cb4f
commit 4970dab6d7
No known key found for this signature in database
GPG key ID: 5DAFE8D9C823CE73
3 changed files with 84 additions and 90 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

154
feedgen/ext/media.py
View file

@ -1,7 +1,7 @@
# -*- coding: utf-8 -*-
'''
feedgen.ext.media
~~~~~~~~~~~~~~~~~~~~~~~~~
~~~~~~~~~~~~~~~~~
Extends the feedgen to produce media tags.
@ -11,6 +11,7 @@
'''
from lxml import etree
from feedgen.util import ensure_format
from feedgen.ext.base import BaseExtension, BaseEntryExtension
MEDIA_NS = 'http://search.yahoo.com/mrss/'
@ -38,39 +39,21 @@ class MediaEntryExtension(BaseEntryExtension):
:param feed: The RSS item XML element to use.
'''
groups = {None: entry}
for media_content in self.__media_content:
# Define current media:group
group = groups.get(media_content.get('group'))
if group is None:
group = etree.SubElement(entry, '{%s}group' % MEDIA_NS)
if self.__media_content:
groups[media_content.get('group')] = group
# Add content
content = etree.SubElement(group, '{%s}content' % MEDIA_NS)
if self.__media_content.get('url'):
content.set('url', self.__media_content.get('url'))
if self.__media_content.get('fileSize'):
content.set('fileSize', self.__media_content.get('fileSize'))
if self.__media_content.get('type'):
content.set('type', self.__media_content.get('type'))
if self.__media_content.get('medium'):
content.set('medium', self.__media_content.get('medium'))
if self.__media_content.get('isDefault'):
content.set('isDefault', self.__media_content.get('isDefault'))
if self.__media_content.get('expression'):
content.set(
'expression', self.__media_content.get('expression'))
if self.__media_content.get('bitrate'):
content.set('bitrate', self.__media_content.get('bitrate'))
if self.__media_content.get('framerate'):
content.set('framerate', self.__media_content.get('framerate'))
if self.__media_content.get('samplingrate'):
content.set('samplingrate',
self.__media_content.get('samplingrate'))
if self.__media_content.get('channels'):
content.set('channels', self.__media_content.get('channels'))
if self.__media_content.get('duration'):
content.set('duration', self.__media_content.get('duration'))
if self.__media_content.get('height'):
content.set('height', self.__media_content.get('height'))
if self.__media_content.get('width'):
content.set('width', self.__media_content.get('width'))
if self.__media_content.get('lang'):
content.set('lang', self.__media_content.get('lang'))
for attr in ('url', 'fileSize', 'type', 'medium', 'isDefault',
'expression', 'bitrate', 'framerate', 'samplingrate',
'channels', 'duration', 'height', 'width', 'lang'):
if media_content.get(attr):
content.set(attr, media_content[attr])
if self.__media_thumbnail:
thumbnail = etree.SubElement(group, '{%s}thumbnail' % MEDIA_NS)
if self.__media_thumbnail.get('url'):
@ -87,69 +70,68 @@ class MediaEntryExtension(BaseEntryExtension):
def extend_rss(self, item):
return self.extend_atom(item)
def content(self, url=None, fileSize=None, type=None, medium=None,
isDefault=None, expression=None, bitrate=None, framerate=None,
samplingrate=None, channels=None, duration=None, height=None,
width=None, lang=None):
'''<media:content> is a sub-element of either <item> or <media:group>.
def content(self, content=None, replace=False, group='default', **kwargs):
'''Get or set media:content data.
This method can be called with:
- the fields of a media:content as keyword arguments
- the fields of a media:content as a dictionary
- a list of dictionaries containing the media:content fields
<media:content> is a sub-element of either <item> or <media:group>.
Media objects that are not the same content should not be included in
the same <media:group> element. The sequence of these items implies
the order of presentation. While many of the attributes appear to be
audio/video specific, this element can be used to publish any type
of media. It contains 14 attributes, most of which are optional.
:param url: should specify the direct URL to the media object.
:param fileSize: number of bytes of the media object.
:param type: standard MIME type of the object.
:param medium: type of object
(image | audio | video | document | executable).
:param isDefault: determines if this is the default object.
:param expression: determines if the object is a sample or the full
version of the object, or even if it is a
continuous stream (sample | full | nonstop).
:param bitrate: kilobits per second rate of media.
:param framerate: number of frames per second for the media object.
:param samplingrate: number of samples per second taken to create the
media object. It is expressed in thousands of
samples per second (kHz).
:param channels: number of audio channels in the media object.
:param duration: number of seconds the media object plays.
:param height: height of the media object.
:param width: width of the media object.
:param lang: is the primary language encapsulated in the media object.
media:content has the following fields:
- *url* should specify the direct URL to the media object.
- *fileSize* number of bytes of the media object.
- *type* standard MIME type of the object.
- *medium* type of object (image | audio | video | document |
executable).
- *isDefault* determines if this is the default object.
- *expression* determines if the object is a sample or the full version
of the object, or even if it is a continuous stream (sample | full |
nonstop).
- *bitrate* kilobits per second rate of media.
- *framerate* number of frames per second for the media object.
- *samplingrate* number of samples per second taken to create the media
object. It is expressed in thousands of samples per second (kHz).
- *channels* number of audio channels in the media object.
- *duration* number of seconds the media object plays.
- *height* height of the media object.
- *width* width of the media object.
- *lang* is the primary language encapsulated in the media object.
:param content: Dictionary or list of dictionaries with content data.
:param replace: Add or replace old data.
:param group: Media group to put this content in.
:returns: The media content tag.
'''
if url is not None:
self.__media_content = {'url': url}
if fileSize is not None:
self.__media_content['fileSize'] = fileSize
if type is not None:
self.__media_content['type'] = type
if medium is not None:
self.__media_content['medium'] = medium
if isDefault is not None:
self.__media_content['isDefault'] = isDefault
if expression is not None:
self.__media_content['expression'] = expression
if bitrate is not None:
self.__media_content['bitrate'] = bitrate
if framerate is not None:
self.__media_content['framerate'] = framerate
if samplingrate is not None:
self.__media_content['samplingrate'] = samplingrate
if channels is not None:
self.__media_content['channels'] = channels
if duration is not None:
self.__media_content['duration'] = duration
if height is not None:
self.__media_content['height'] = height
if width is not None:
self.__media_content['width'] = width
if lang is not None:
self.__media_content['lang'] = lang
# Handle kwargs
if content is None and kwargs:
content = kwargs
# Handle new data
if content is not None:
# Reset data if we want to replace them
if replace or self.__media_content is None:
self.__media_content = []
# Ensure list
if not isinstance(content, list):
content = [content]
# define media group
for c in content:
c['group'] = c.get('group', group)
self.__media_content += ensure_format(
content,
set(['url', 'fileSize', 'type', 'medium', 'isDefault',
'expression', 'bitrate', 'framerate', 'samplingrate',
'channels', 'duration', 'height', 'width', 'lang',
'group']),
set(['url', 'group']))
return self.__media_content
def thumbnail(self, url=None, height=None, width=None, time=None):

2
feedgen/util.py
View file

@ -26,7 +26,7 @@ def ensure_format(val, allowed, required, allowed_values=None, defaults=None):
:returns: List of checked dictionaries.
'''
if not val:
return None
return []
if allowed_values is None:
allowed_values = {}
if defaults is None:

16
tests/test_extension.py
View file

@ -212,6 +212,10 @@ class TestExtensionMedia(unittest.TestCase):
fe.title('title')
fe.content('content')
fe.media.content(url='file1.xy')
fe.media.content(url='file2.xy')
fe.media.content(url='file1.xy', group=2)
fe.media.content(url='file2.xy', group=2)
fe.media.content(url='file.xy', group=None)
ns = {'media': 'http://search.yahoo.com/mrss/',
'a': 'http://www.w3.org/2005/Atom'}
@ -219,10 +223,18 @@ class TestExtensionMedia(unittest.TestCase):
root = etree.fromstring(self.fg.rss_str())
url = root.xpath('/rss/channel/item/media:group/media:content[1]/@url',
namespaces=ns)
assert url == ['file1.xy']
assert url == ['file1.xy', 'file1.xy']
# There is one without a group
url = root.xpath('/rss/channel/item/media:content[1]/@url',
namespaces=ns)
assert url == ['file.xy']
# Check that we have the item in the resulting Atom feed
root = etree.fromstring(self.fg.atom_str())
url = root.xpath('/a:feed/a:entry/media:group/media:content[1]/@url',
namespaces=ns)
assert url == ['file1.xy']
assert url == ['file1.xy', 'file1.xy']
fe.media.content(content=[], replace=True)
assert fe.media.content() == []