[opensubtitles] search is faster, fillobj is handled for info command

This commit is contained in:
Julien Veyssier 2013-03-13 14:37:53 +01:00
commit 2c3eb10ac3
2 changed files with 49 additions and 6 deletions

View file

@ -17,7 +17,7 @@
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
from weboob.capabilities.subtitle import ICapSubtitle,LanguageNotSupported
from weboob.capabilities.subtitle import ICapSubtitle,LanguageNotSupported,Subtitle
from weboob.applications.suboob.suboob import LANGUAGE_CONV
from weboob.tools.backend import BaseBackend
@ -54,3 +54,14 @@ class OpensubtitlesBackend(BaseBackend, ICapSubtitle):
if language not in LANGUAGE_CONV.keys():
raise LanguageNotSupported()
return self.browser.iter_subtitles(language,quote_plus(pattern.encode('utf-8')))
def fill_subtitle(self, subtitle, fields):
if 'description' in fields:
sub = self.get_subtitle(subtitle.id)
subtitle.description = sub.description
return subtitle
OBJECTS = {
Subtitle:fill_subtitle,
}

View file

@ -19,7 +19,7 @@
from weboob.capabilities.subtitle import Subtitle
from weboob.capabilities.base import NotAvailable
from weboob.capabilities.base import NotAvailable, NotLoaded
from weboob.tools.browser import BasePage
from weboob.applications.suboob.suboob import LANGUAGE_CONV
@ -69,10 +69,42 @@ class SubtitlesPage(BasePage):
links = self.parser.select(line,'a')
a = links[0]
urldetail = a.attrib.get('href','')
self.browser.location("http://www.opensubtitles.org%s"%urldetail)
assert self.browser.is_on_page(SubtitlePage)
# subtitle page does the job
return self.browser.page.get_subtitle()
#self.browser.location("http://www.opensubtitles.org%s"%urldetail)
#assert self.browser.is_on_page(SubtitlePage)
## subtitle page does the job
#return self.browser.page.get_subtitle()
name = u" ".join(a.text.strip().split())
first_cell = cells[0]
spanlist = self.parser.select(first_cell,'span')
if len(spanlist) > 0:
long_name = spanlist[0].attrib.get('title','')
else:
texts = first_cell.itertext()
long_name = texts.next()
long_name = texts.next()
if "Download at 25" in long_name:
long_name = "---"
name = "%s (%s)"%(name,long_name)
second_cell = cells[1]
link = self.parser.select(second_cell,'a',1)
lang = link.attrib.get('href','').split('/')[-1].split('-')[-1]
for lshort,llong in LANGUAGE_CONV.items():
if lang == llong:
lang = unicode(lshort)
break
nb_cd = int(cells[2].text.strip().lower().replace('cd',''))
cell_dl = cells[4]
href = self.parser.select(cell_dl,'a',1).attrib.get('href','')
url = "http://www.opensubtitles.org%s"%href
id = href.split('/')[-1]
subtitle = Subtitle(id,name)
subtitle.url = url
subtitle.language = lang
subtitle.nb_cd = nb_cd
subtitle.description = NotLoaded
return subtitle
class SubtitlePage(BasePage):