added simple parser to handle yt playlists

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

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)))