From 969786e299a26983e499460c6b9bc9b0fdbe3f15 Mon Sep 17 00:00:00 2001
From: Micke Nordin <kano@sunet.se>
Date: Sun, 19 Dec 2021 18:08:05 +0100
Subject: [PATCH] Working YouTube

---
 ChannelProvider/YouTube/__init__.py | 96 +++++++++++++++++++++++++++++
 dpkg.lst                            |  1 +
 main.py                             |  4 +-
 3 files changed, 99 insertions(+), 2 deletions(-)

diff --git a/ChannelProvider/YouTube/__init__.py b/ChannelProvider/YouTube/__init__.py
index e69de29..d789768 100644
--- a/ChannelProvider/YouTube/__init__.py
+++ b/ChannelProvider/YouTube/__init__.py
@@ -0,0 +1,96 @@
+import hashlib
+import io
+import os
+import pickle
+import threading
+from typing import Union
+
+import feedparser
+import requests
+import wx
+from ChannelProvider import ChannelProvider
+from Items import Item
+from youtube_dl import YoutubeDL as yt
+
+
+class YouTube(ChannelProvider):
+    m_cache: dict = dict()
+    m_cachefile = '/tmp/yt_cache'
+
+    def __init__(self, channel_id) -> None:
+        super().__init__(
+            channel_id,
+            'https://www.youtube.com/feeds/videos.xml?channel_id={}'.format(
+                channel_id))
+        self.m_items: Union[list[Item], None] = None
+
+        if os.path.exists(self.m_cachefile):
+            with open(self.m_cachefile, 'rb') as cachehandle:
+                self.m_cache = pickle.load(cachehandle)
+        self.m_thr = threading.Thread(target=self.parse_feed,
+                                      args=(),
+                                      kwargs={})
+        self.m_thr.start()
+
+    def wait(self) -> bool:
+        return self.m_thr.is_alive()
+
+    def parse_feed(self) -> None:
+        feed = feedparser.parse(self.get_feed())
+
+        ydl_opts = {
+            'format': 'worst',
+            'container': 'webm_dash',
+        }
+        entries = feed['entries']
+        self.m_items: list[Item] = list()
+
+        for entry in entries:
+            key = hashlib.sha256(entry['link'].encode('utf-8')).hexdigest()
+
+            if key in self.m_cache.keys():
+                thumbnail_link = self.m_cache[key]['thumbnail_link']
+                content = self.m_cache[key]['content']
+                resolved_link = self.m_cache[key]['resolved_link']
+                description = self.m_cache[key]['description']
+                published_parsed = self.m_cache[key]['published_parsed']
+                title = self.m_cache[key]['title']
+            else:
+
+                title = str(entry['title'])
+                thumbnail_link = str(entry['media_thumbnail'][0]['url'])
+                description = str(entry['description'])
+                link = ''
+                with yt(ydl_opts) as ydl:
+                    video = yt(ydl_opts).extract_info(entry['link'],
+                                                      download=False)
+
+                    for form in video['formats']:
+                        if form['height']:
+                            if form['height'] < 480 and form[
+                                    'acodec'] != 'none':
+                                link = form['url']
+
+                resolved_link = link
+
+                published_parsed = entry['published_parsed']
+                res = requests.get(thumbnail_link)
+                content = res.content
+
+            if not resolved_link:
+                continue
+            content_bytes = io.BytesIO(content)
+            self.m_cache[key] = {'thumbnail_link': thumbnail_link}
+            self.m_cache[key]['content'] = content
+            self.m_cache[key]['resolved_link'] = resolved_link
+            self.m_cache[key]['description'] = description
+            self.m_cache[key]['published_parsed'] = published_parsed
+            self.m_cache[key]['title'] = title
+            image = wx.Image(content_bytes, type=wx.BITMAP_TYPE_ANY, index=-1)
+            thumbnail = wx.Bitmap(image)
+            item = Item(description, resolved_link, self.m_provider_name,
+                        published_parsed, thumbnail, title)
+            self.m_items.append(item)
+        # write to cache file
+        with open(self.m_cachefile, 'wb') as cachehandle:
+            pickle.dump(self.m_cache, cachehandle)
diff --git a/dpkg.lst b/dpkg.lst
index a33db2f..cd264d1 100644
--- a/dpkg.lst
+++ b/dpkg.lst
@@ -25,3 +25,4 @@ libgstreamer1.0-0:amd64
 python3-wxgtk-media4.0
 python3-wxgtk4.0
 python3-feedparser
+youtube-dl
diff --git a/main.py b/main.py
index c29ebaa..2b6024e 100644
--- a/main.py
+++ b/main.py
@@ -8,7 +8,7 @@ import wx
 import wx.lib.scrolledpanel as scrolled
 import wx.media
 
-from ChannelProvider import SVT, ChannelProvider
+from ChannelProvider import SVT, ChannelProvider, YouTube
 
 ChannelProvider = NewType('ChannelProvider', ChannelProvider)
 
@@ -34,7 +34,7 @@ class Cast(wx.Frame):
         self.m_control = None
         self.m_panel.SetupScrolling()
         self.m_panel.SetSizer(self.m_sizer)
-        self.m_channels: list[ChannelProvider] = [SVT.SVT()]
+        self.m_channels: list[ChannelProvider] = [YouTube.YouTube('UCu6mSoMNzHQiBIOCkHUa2Aw')]#,SVT.SVT()]
         self.show_list(None)
 
     def get_chromecasts(self):