From c6c7165a97fef4b2c9212640851ef227740b76b7 Mon Sep 17 00:00:00 2001
From: Lars Kiesow <lkiesow@uos.de>
Date: Sun, 12 Aug 2018 21:07:47 +0200
Subject: [PATCH] Add Source Element to Feed Entries

This patch implements the source element for feed entries/items. Note
that only the set of RSS elements (URL and title) are implemented right
now. ATOM supports additional elements which cannot be set at the
moment.

This fixes #79
---
 feedgen/entry.py    | 29 +++++++++++++++++++++++++++++
 tests/test_entry.py |  3 +++
 2 files changed, 32 insertions(+)

diff --git a/feedgen/entry.py b/feedgen/entry.py
index d7827c8..92fb5f8 100644
--- a/feedgen/entry.py
+++ b/feedgen/entry.py
@@ -173,6 +173,15 @@ class FeedEntry(object):
             rights = etree.SubElement(entry, 'rights')
             rights.text = self.__atom_rights
 
+        if self.__atom_source:
+            source = etree.SubElement(entry, 'source')
+            if self.__atom_source.get('title'):
+                source_title = etree.SubElement(source, 'title')
+                source_title.text = self.__atom_source['title']
+            if self.__atom_source.get('link'):
+                etree.SubElement(source, 'link',
+                                 href=self.__atom_source['link'])
+
         if extensions:
             for ext in self.__extensions.values() or []:
                 if ext.get('atom'):
@@ -233,6 +242,10 @@ class FeedEntry(object):
         if self.__rss_pubDate:
             pubDate = etree.SubElement(entry, 'pubDate')
             pubDate.text = formatRFC2822(self.__rss_pubDate)
+        if self.__rss_source:
+            source = etree.SubElement(entry, 'source',
+                                      url=self.__rss_source['url'])
+            source.text = self.__rss_source['title']
 
         if extensions:
             for ext in self.__extensions.values() or []:
@@ -614,6 +627,22 @@ class FeedEntry(object):
             self.__rss_comments = comments
         return self.__rss_comments
 
+    def source(self, url=None, title=None):
+        '''Get or set the source for the current feed entry.
+
+        Note that ATOM feeds support a lot more sub elements than title and URL
+        (which is what RSS supports) but these are currently not supported.
+        Patches are welcome.
+
+        :param url: Link to the source.
+        :param title: Title of the linked resource
+        :returns: Source element as dictionaries.
+        '''
+        if url is not None and title is not None:
+            self.__rss_source = {'url': url, 'title': title}
+            self.__atom_source = {'link': url, 'title': title}
+        return self.__rss_source
+
     def enclosure(self, url=None, length=None, type=None):
         '''Get or set the value of enclosure which describes a media object
         that is attached to the item. This is a RSS only value which is
diff --git a/tests/test_entry.py b/tests/test_entry.py
index 1df3da5..6c9835a 100644
--- a/tests/test_entry.py
+++ b/tests/test_entry.py
@@ -91,6 +91,9 @@ class TestSequenceFunctions(unittest.TestCase):
         assert fe.pubDate().year == 2017
         fe.rights('asdfx')
         assert fe.rights() == 'asdfx'
+        source = fe.source(url='https://example.com', title='Test')
+        assert source.get('title') == 'Test'
+        assert source.get('url') == 'https://example.com'
         fe.comments('asdfx')
         assert fe.comments() == 'asdfx'
         fe.enclosure(url='http://lkiesow.de', type='text/plain', length='1')