Add filters to list - like Seasons and Availability

Author: alladdin

addon.xml

@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
-<addon id="plugin.video.primaplay" name="Prima Play" version="0.1.1" provider-name="Ladislav Dokulil">
+<addon id="plugin.video.primaplay" name="Prima Play" version="0.1.2" provider-name="Ladislav Dokulil">
   <requires>
     <import addon="xbmc.python" version="2.1.0"/>
   </requires>

changelog.txt

@@ -1,3 +1,6 @@
+2016-06-01 [0.1.2]
+ * Add filter lists for Seasons and others
+
 2016-06-01 [0.1.1]
  * Don't use BeautifulSoap - replaced by regexps
 

default.py

@@ -47,6 +47,9 @@ def _toString(text):
     def main_menu(pageurl):
         page = _play_parser.get_page(pageurl)
         if page.player: add_player(page.player)
+        for filter_list in page.filter_lists:
+            add_title(filter_list)
+            add_item_list(filter_list.item_list)
         for video_list in page.video_lists:
             if video_list.title: add_title(video_list)
             add_item_list(video_list.item_list)

libPrimaPlay/PrimaPlay.py

@@ -71,7 +71,8 @@ def get_page(self, link):
         content = self.ua.get(link)
         return Page(
             self.get_page_player(content),
-            self.get_video_lists(content, link)
+            self.get_video_lists(content, link),
+            self.get_filter_lists(content, link)
         )
 
     def get_page_player(self, content):
@@ -144,6 +145,34 @@ def get_items_from_wrapper(self, content, src_link):
             list.append(Item(title.decode('utf-8'), link, image_url))
         return list
 
+    def get_filter_lists(self, content, src_link):
+        list = []
+        before_wrapper_re = re.compile('(.*)<div class="loading-wrapper">', re.S)
+        before_content = before_wrapper_re.search(content).group(1)
+
+        filter_wrappers = re.split('<li class="hamburger-parent[^"]*">', before_content)
+
+        title_re = re.compile('<span data-jnp="[^"]+" class="hamburger-toggler">([^<]+)</span>')
+        for filter_wrapper in filter_wrappers:
+            title_result = title_re.search(filter_wrapper)
+            if title_result is None: continue
+            title = title_result.group(1)
+            items = self.get_filter_items(filter_wrapper, src_link)
+            if (len(items) <= 0): continue 
+            list.append(PageVideoList(title.decode('utf-8'), None, None, items))
+        return list
+
+    def get_filter_items(self, content, src_link):
+        list = []
+
+        filter_item_re = re.compile('<li>[^<]*<a class="tdi" href="([^"]+)"[^>]*>([^<]+)</a>[^<]*</li>', re.S)
+        for raw_link, raw_title in filter_item_re.findall(content):
+            link = self.make_full_link(raw_link, src_link)
+            title = self.strip_tags(raw_title)
+            list.append(Item(title.decode('utf-8'), link))
+
+        return list
+
     def make_full_link(self, target_link, src_link):
         if target_link is None:
             return None
@@ -169,8 +198,9 @@ def strip_tags(self, string):
         return result.strip()
 
 class Page:
-    def __init__(self, player = None, video_lists = []):
+    def __init__(self, player = None, video_lists = [], filter_lists = []):
         self.video_lists = video_lists
+        self.filter_lists = filter_lists
         self.player = player
 
 class PageVideoList:

libPrimaPlay/PrimaPlay_unittest.py

@@ -101,5 +101,13 @@ def test_get_page__episodes(self):
         self.assertEqual(page.video_lists[0].item_list[0].link,
             'http://play.iprima.cz/prostreno-ix-10')
 
+        self.assertEqual(len(page.filter_lists), 3)
+        self.assertEqual(page.filter_lists[0].title, u'Řada')
+
+        self.assertEqual(len(page.filter_lists[0].item_list), 11)
+        self.assertEqual(page.filter_lists[0].item_list[0].title, u'Sezóna 1')
+        self.assertEqual(page.filter_lists[0].item_list[0].link,
+            'http://play.iprima.cz/prostreno?season=p14883&sort[]=Rord&sort[]=latest')
+
 if __name__ == '__main__':
     unittest.main()