Add search feature

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.2" provider-name="Ladislav Dokulil">
+<addon id="plugin.video.primaplay" name="Prima Play" version="0.1.3" provider-name="Ladislav Dokulil">
   <requires>
     <import addon="xbmc.python" version="2.1.0"/>
   </requires>

changelog.txt

@@ -1,3 +1,6 @@
+2016-07-01 [0.1.3]
+ * Add search feature
+
 2016-06-01 [0.1.2]
  * Add filter lists for Seasons and others
 

default.py

@@ -44,6 +44,12 @@ def _toString(text):
     _baseurl_ = sys.argv[0]
     _play_parser = PrimaPlay.Parser()
 
+
+    def first_menu():
+        li = list_item('Hledej')
+        url = get_menu_link( pagesearch = '1' )
+        xbmcplugin.addDirectoryItem(handle=_handle_, url=url, listitem=li, isFolder=True)
+
     def main_menu(pageurl):
         page = _play_parser.get_page(pageurl)
         if page.player: add_player(page.player)
@@ -64,6 +70,18 @@ def next_menu(nexturl):
 
         xbmcplugin.endOfDirectory(_handle_, updateListing=True)
 
+    def search():
+        keyboard = xbmc.Keyboard('','Hledej')
+        keyboard.doModal()
+        if (not keyboard.isConfirmed()):
+            xbmcplugin.endOfDirectory(_handle_)
+            return
+        search_query = keyboard.getText()
+        if len(search_query) <= 1:
+            xbmcplugin.endOfDirectory(_handle_)
+            return
+        main_menu(_play_parser.get_search_url(search_query))
+
     def add_title(video_list):
         li = list_item('[B]'+video_list.title+'[/B]')
         url = '#'
@@ -115,18 +133,25 @@ def assign_params(params):
                 pass
 
 
-    pageurl = "http://play.iprima.cz"
+    pageurl = None
     nexturl = None
+    pagesearch = None
     params = get_params()
     assign_params(params)
     logDbg("PrimaPlay Parameters!!!")
     logDbg("PAGE: "+str(pageurl))
     logDbg("NEXT PAGE: "+str(nexturl))
+    logDbg("PAGE SEARCH: "+str(pagesearch))
     try:
         if nexturl:
             next_menu(nexturl)
-        else:
+        elif pagesearch:
+            search()
+        elif pageurl:
             main_menu(pageurl)
+        else:
+            first_menu()
+            main_menu("http://play.iprima.cz")
     except Exception as ex:
         exc_type, exc_value, exc_traceback = sys.exc_info()
         _exception_log(exc_type, exc_value, exc_traceback)

libPrimaPlay/PrimaPlay.py

@@ -35,6 +35,7 @@ class Parser:
     def __init__(self, ua = UserAgent(), time_obj = time):
         self.ua = ua
         self.player_init_url = 'http://play.iprima.cz/prehravac/init?'
+        self.search_url = 'http://play.iprima.cz/vysledky-hledani-vse?'
         self.time = time_obj
 
     def get_player_init_url(self, productID):
@@ -45,6 +46,11 @@ def get_player_init_url(self, productID):
         })
         #http://play.iprima.cz/prehravac/init?_infuse=1&_ts=1450864235286&productId=p135603
 
+    def get_search_url(self, query):
+        return self.search_url + urllib.urlencode({
+            'query': query
+        })
+
     def get_video_link(self, productID):
         content = self.ua.get(self.get_player_init_url(productID))
         link_re = re.compile("'src'\s*:\s+'(https?://[^']+\\.m3u8)'")
@@ -147,8 +153,10 @@ def get_items_from_wrapper(self, content, src_link):
 
     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)
+        before_wrapper_re = re.compile('^(.*)<div class="loading-wrapper">', re.S)
+        before_wrapper_result = before_wrapper_re.search(content)
+        if before_wrapper_result is None: return list
+        before_content = before_wrapper_result.group(1)
 
         filter_wrappers = re.split('<li class="hamburger-parent[^"]*">', before_content)
 

libPrimaPlay/PrimaPlay_unittest.py

@@ -109,5 +109,20 @@ def test_get_page__episodes(self):
         self.assertEqual(page.filter_lists[0].item_list[0].link,
             'http://play.iprima.cz/prostreno?season=p14883&sort[]=Rord&sort[]=latest')
 
+    def test_get_page__search(self):
+        prima_play = PrimaPlay.Parser(mockUserAgent(['test_search_page.html']), mockTime())
+        page = prima_play.get_page('http://play.iprima.cz')
+
+        self.assertEqual(page.player, None)
+        self.assertEqual(len(page.video_lists), 3)
+        self.assertEqual(page.video_lists[0].title, u'Mezi seriály')
+        self.assertEqual(page.video_lists[0].link,
+            'http://play.iprima.cz/vysledky-hledani?query=prostreno&searchGroup=SERIES')
+        self.assertEqual(len(page.video_lists[0].item_list), 2)
+        self.assertEqual(page.video_lists[0].item_list[0].title,
+            u'VIP PROSTŘENO! 3 Řady , 32 Epizod')
+        self.assertEqual(page.video_lists[0].item_list[0].link,
+            'http://play.iprima.cz/vip-prostreno')
+
 if __name__ == '__main__':
     unittest.main()