1 files changed, 106 insertions, 0 deletions

diff --git a/youtube_podcaster/podcastupdater.py b/youtube_podcaster/podcastupdater.py
new file mode 100644
index 0000000..4a0a017
--- /dev/null
+++ b/youtube_podcaster/podcastupdater.py
@@ -0,0 +1,106 @@
+import pickle
+import os
+import time
+import hashlib
+
+from feedgen.feed import FeedGenerator
+
+from . import (
+    youtube,
+)
+
+
+class PodcastUpdater:
+    def __init__(self, youtube_config, podcast_config):
+        self.podcasts = podcast_config
+        self.youtube = youtube.Youtube(youtube_config["api-key"])
+
+        self.feeds_file = "feeds.save"
+        if os.path.isfile(self.feeds_file):
+            with open(self.feeds_file, "rb") as feeds:
+                self.feeds = pickle.load(feeds)
+        else:
+            self.feeds = {}
+
+    def get_xml(self, channel, playlist):
+        channel = channel.replace('_', ' ')
+        playlist = playlist.replace('_', ' ')
+
+        for podcast in self.podcasts:
+            if podcast["username"] == channel:
+                break
+        else:
+            return None
+
+        if playlist not in podcast["playlists"]:
+            return None
+
+        xml = self.update_podcast(channel, playlist)
+
+        if xml:
+            return open(xml).read()
+
+    def update_podcast(self, channel, playlist):
+        feed_id = hashlib.sha1(bytes("%s %s" % (channel, playlist), "UTF-8")).hexdigest()
+        yt_channel = self.youtube.get_channel(channel)[0]
+        yt_playlists = self.youtube.get_playlists(yt_channel, 50)
+
+        for yt_playlist in yt_playlists:
+            if yt_playlist["snippet"]["title"] == playlist:
+                break
+        else:
+            return None
+
+        if feed_id in self.feeds:
+            feed = self.feeds[feed_id]
+        else:
+            feed = self.add_feed(feed_id, yt_playlist)
+
+        if feed.last_updated < time.time() - 600:
+            self.populate_feed(feed, feed_id, yt_playlist)
+
+            feed_file = "%s.xml" % (feed_id)
+            self.feeds[feed_id].rss_file(feed_file)
+
+            with open(self.feeds_file, "wb") as feed:
+                pickle.dump(self.feeds, feed)
+
+        return "%s.xml" % (feed_id)
+
+    def add_feed(self, feed_id, yt_playlist):
+        feed = FeedGenerator()
+        feed.load_extension("podcast")
+        feed.id(feed_id)
+        feed.title(yt_playlist["snippet"]["title"])
+        feed.author({"name": yt_playlist["snippet"]["channelTitle"]})
+        feed.description(yt_playlist["snippet"]["description"])
+        feed.logo(yt_playlist["snippet"]["thumbnails"]["standard"]["url"])
+        feed.link(href="https://www.youtube.com/playlist?list=%s" % (yt_playlist["id"]))
+        feed.rss_str(pretty=True)
+        feed.last_updated = 0
+        self.feeds[feed_id] = feed
+        return feed
+
+    def populate_feed(self, feed, feed_id, yt_playlist, max_results=5):
+        videos = self.youtube.get_playlist_items(yt_playlist, max_results)
+        downloader = youtube.Downloader.get_instance("vorbis", "downloads", "192.168.178.100")
+
+        entries = feed.entry()
+        for video in videos:
+            video_id = hashlib.sha1(bytes(video["id"], "UTF-8")).hexdigest()
+            for entry in entries:
+                if entry.id() == video_id:
+                    break
+            else:
+                url, size, mime = downloader.download(video, video_id, feed_id)
+                feed_entry = feed.add_entry()
+                feed_entry.id(video_id)
+                feed_entry.guid(video_id)
+                feed_entry.title(video["snippet"]["title"])
+                feed_entry.description(video["snippet"]["description"])
+                feed_entry.published(video["snippet"]["publishedAt"])
+                feed_entry.enclosure(url, size, mime)
+
+        feed.last_updated = time.time()
+
+#  vim: set ts=8 sw=4 tw=0 et :