[enh] subtitleseeker: better language handling

This commit is contained in:
Adam Tauber 2014-12-24 21:02:26 +01:00
parent b4f39237a3
commit 3b3921fc59
1 changed files with 11 additions and 1 deletions

View File

@ -11,6 +11,7 @@
from cgi import escape from cgi import escape
from urllib import quote_plus from urllib import quote_plus
from lxml import html from lxml import html
from searx.languages import language_codes
# engine dependent config # engine dependent config
categories = ['videos'] categories = ['videos']
@ -38,13 +39,22 @@ def response(resp):
dom = html.fromstring(resp.text) dom = html.fromstring(resp.text)
search_lang = ""
if resp.search_params['language'] != 'all':
search_lang = [lc[1]
for lc in language_codes
if lc[0][:2] == resp.search_params['language']][0]
# parse results # parse results
for result in dom.xpath(results_xpath): for result in dom.xpath(results_xpath):
link = result.xpath(".//a")[0] link = result.xpath(".//a")[0]
href = link.attrib.get('href') href = link.attrib.get('href')
if language is not "": if language is not "":
href = href + language + "/" href = href + language + '/'
elif search_lang:
href = href + search_lang + '/'
title = escape(link.xpath(".//text()")[0]) title = escape(link.xpath(".//text()")[0])