[subtitle] add language in search

modules/attilasub/backend.py

@@ -34,6 +34,7 @@ class AttilasubBackend(BaseBackend, ICapSubtitle):
     VERSION = '0.f'
     DESCRIPTION = '"Attila'' s Website 2.0" french subtitles'
     LICENSE = 'AGPLv3+'
+    LANGUAGE_LIST = ['fr']
     BROWSER = AttilasubBrowser
 
     def create_default_browser(self):
@@ -49,5 +50,7 @@ class AttilasubBackend(BaseBackend, ICapSubtitle):
 
         return self.browser.openurl(subtitle.url.encode('utf-8')).read()
 
-    def iter_subtitles(self, pattern):
-        return self.browser.iter_subtitles(quote_plus(pattern.encode('utf-8')))
+    def iter_subtitles(self, language, pattern):
+        if language not in self.LANGUAGE_LIST:
+            return []
+        return self.browser.iter_subtitles(language,quote_plus(pattern.encode('utf-8')))

modules/attilasub/browser.py

@@ -36,10 +36,10 @@ class AttilasubBrowser(BaseBrowser):
         'http://davidbillemont3.free.fr/.*.htm': SubtitlesPage,
         }
 
-    def iter_subtitles(self, pattern):
+    def iter_subtitles(self, language, pattern):
         self.location('http://search.freefind.com/find.html?id=81131980&_charset_=&bcd=%%F7&scs=1&pageid=r&query=%s&mode=Find%%20pages%%20matching%%20ALL%%20words' % pattern.encode('utf-8'))
         assert self.is_on_page(SearchPage)
-        return self.page.iter_subtitles(pattern)
+        return self.page.iter_subtitles(language,pattern)
 
     def get_subtitle(self, id):
         url_end = id.split('|')[0]

modules/attilasub/pages.py

@@ -35,7 +35,7 @@ __all__ = ['SubtitlesPage','SearchPage']
 
 
 class SearchPage(BasePage):
-    def iter_subtitles(self,pattern):
+    def iter_subtitles(self, language, pattern):
         fontresult = self.parser.select(self.document.getroot(),'div.search-results font.search-results')
         # for each result in freefind, explore the subtitle list page to iter subtitles
         for res in fontresult:
@@ -44,7 +44,7 @@ class SearchPage(BasePage):
             self.browser.location(url)
             assert self.browser.is_on_page(SubtitlesPage)
             # subtitles page does the job
-            for subtitle in self.browser.page.iter_subtitles(pattern):
+            for subtitle in self.browser.page.iter_subtitles(language, pattern):
                 yield subtitle
 
 
@@ -72,11 +72,12 @@ class SubtitlesPage(BasePage):
         subtitle = Subtitle(id,name)
         subtitle.url = url
         subtitle.fps = 0
+        subtitle.language = "fre"
         subtitle.nb_cd = nb_cd
         subtitle.description = "no desc"
         return subtitle
 
-    def iter_subtitles(self,pattern):
+    def iter_subtitles(self,language, pattern):
         pattern = pattern.strip().replace('+',' ')
         pattern_words = pattern.split()
         tab = self.parser.select(self.document.getroot(),'table[bordercolor="#B8C0B2"]')
@@ -84,6 +85,9 @@ class SubtitlesPage(BasePage):
             tab = self.parser.select(self.document.getroot(),'table[bordercolordark="#B8C0B2"]')
             if len(tab) == 0:
                 return
+        # some results of freefind point on useless pages
+        if tab[0].attrib.get('width','') != '100%':
+            return
         for line in tab[0].getiterator('tr'):
             cols = self.parser.select(line,'td')
             traduced_title = self.parser.select(cols[0],'font',1).text.lower()
@@ -112,6 +116,7 @@ class SubtitlesPage(BasePage):
                 subtitle = Subtitle(id,name)
                 subtitle.url = url
                 subtitle.fps = 0
+                subtitle.language = "fre"
                 subtitle.nb_cd = nb_cd
                 subtitle.description = "no desc"
                 yield subtitle

modules/attilasub/test.py

@@ -27,7 +27,7 @@ class AttilasubTest(BackendTest):
     BACKEND = 'attilasub'
 
     def test_subtitle(self):
-        subtitles = list(self.backend.iter_subtitles('spiderman'))
+        subtitles = list(self.backend.iter_subtitles('fr','spiderman'))
         assert (len(subtitles) > 0)
         for subtitle in subtitles:
             path, qs = urllib.splitquery(subtitle.url)