From 60f106bc2aaa6c1ee7b53574556f0935fc4d4f18 Mon Sep 17 00:00:00 2001
From: DYefremov <dmitriy.v.yefremov@gmail.com>
Date: Fri, 21 Jun 2019 14:54:09 +0300
Subject: [PATCH] added simple parser to handle yt playlists

---
 app/tools/yt.py | 62 ++++++++++++++++++++++++++++++++++++++++++++++---
 app/ui/iptv.py  |  2 +-
 2 files changed, 60 insertions(+), 4 deletions(-)

diff --git a/app/tools/yt.py b/app/tools/yt.py
index 814835ef..0c5895c0 100644
--- a/app/tools/yt.py
+++ b/app/tools/yt.py
@@ -1,10 +1,12 @@
 """ Module for working with YouTube service """
 import re
 import urllib
+from html.parser import HTMLParser
 from urllib.request import Request
 
 _YT_PATTERN = re.compile(r"https://www.youtube.com/.+(?:v=|\/)([\w-]{11})&?(list=)?([\w-]{34})?.*")
 _YT_VIDEO_PATTERN = re.compile(r"https://r\d+---sn-[\w]{10}-[\w]{3,5}.googlevideo.com/videoplayback?.*")
+_HEADERS = {"User-Agent": "Mozilla/5.0"}
 
 
 class YouTube:
@@ -27,10 +29,9 @@ class YouTube:
 
             returns tuple from the video link and title
          """
-        headers = {"User-Agent": "Mozilla/5.0"}
-        req = Request("https://youtube.com/get_video_info?video_id={}".format(video_id), headers=headers)
+        req = Request("https://youtube.com/get_video_info?video_id={}".format(video_id), headers=_HEADERS)
         with urllib.request.urlopen(req, timeout=2) as resp:
-            data = resp.read().decode('utf-8').split("&")
+            data = resp.read().decode("utf-8").split("&")
             out = {k: v for k, sep, v in (str(d).partition("=") for d in map(urllib.request.unquote, data))}
             stream_map = out.get("url_encoded_fmt_stream_map", None)
             if stream_map:
@@ -38,3 +39,58 @@ class YouTube:
                 url, title = s_map.get("url", None), out.get("title", None)
                 return urllib.request.unquote(url) if url else "", title.replace("+", " ") if title else ""
             return "", ""
+
+
+class PlayListParser(HTMLParser):
+    """ Very simple parser to handle YouTube playlist pages. """
+
+    def __init__(self):
+        super().__init__()
+        self._is_header = False
+        self._header = ""
+        self._playlist = []
+
+    def handle_starttag(self, tag, attrs):
+        if tag == "h1" and ("class", "pl-header-title") in attrs:
+            self._is_header = True
+
+        elif tag == "tr" and ("class", "pl-video yt-uix-tile ") in attrs:
+            p_data = {k: v for k, v in attrs}
+            self._playlist.append((p_data.get("data-title", None), p_data.get("data-video-id", None)))
+
+    def handle_data(self, data):
+        if self._is_header:
+            self._header = data.strip()
+
+    def handle_endtag(self, tag):
+        if self._is_header:
+            self._is_header = False
+
+    def error(self, message):
+        pass
+
+    @property
+    def header(self):
+        return self._header
+
+    @property
+    def playlist(self):
+        return self._playlist
+
+    @staticmethod
+    def get_yt_playlist(play_list_id):
+        """ Getting YouTube playlist by id.
+
+           returns tuple from the playlist header and list of tuples (title, video id)
+        """
+        request = Request("https://www.youtube.com/playlist?list={}".format(play_list_id), headers=_HEADERS)
+
+        with urllib.request.urlopen(request, timeout=2) as resp:
+            data = resp.read().decode("utf-8")
+            parser = PlayListParser()
+            parser.feed(data)
+            return parser.header, parser.playlist
+
+
+if __name__ == "__main__":
+    pass
diff --git a/app/ui/iptv.py b/app/ui/iptv.py
index f7b02d8a..7c3130e1 100644
--- a/app/ui/iptv.py
+++ b/app/ui/iptv.py
@@ -170,7 +170,7 @@ class IptvDialog:
             elif stream_type is StreamType.NONE_REC_2:
                 self._stream_type_combobox.set_active(3)
         except ValueError:
-            show_dialog(DialogType.ERROR, "Unknown stream type {}".format(s_type))
+            self.show_info_message("Unknown stream type {}".format(s_type), Gtk.MessageType.ERROR)
 
         self._srv_type_entry.set_text(data[2])
         self._sid_entry.set_text(str(int(data[3], 16)))