mirror
/
bazarr
mirror of https://github.com/morpheus65535/bazarr
1
0
Fork 0
Code Issues Releases Wiki Activity

stop sorting query results by language in zimuku

This commit is contained in:
gyh1621 2020-03-25 07:29:45 +00:00
parent b12cb42146
commit 6478635163
1 changed files with 9 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
libs/subliminal_patch/providers/zimuku.py
View File

@ -96,7 +96,7 @@ class ZimukuProvider(Provider):
r.content.decode("utf-8", "ignore"), ["html.parser"] r.content.decode("utf-8", "ignore"), ["html.parser"]
) )
subs_body = bs_obj.find("div", class_="subs box clearfix").find("tbody") subs_body = bs_obj.find("div", class_="subs box clearfix").find("tbody")
subs, lan_scores = [], {} subs = []
for sub in subs_body.find_all("tr"): for sub in subs_body.find_all("tr"):
a = sub.find("a") a = sub.find("a")
name = _extract_name(a.text) name = _extract_name(a.text)
@ -104,20 +104,15 @@ class ZimukuProvider(Provider):
0 0
] # remove ext because it can be an archive type ] # remove ext because it can be an archive type
lan_score = 0 language = Language("eng")
for img in sub.find("td", class_="tac lang").find_all("img"): for img in sub.find("td", class_="tac lang").find_all("img"):
if "uk" in img.attrs["src"]: if (
lan_score += 1 "hongkong" in img.attrs["src"]
elif "hongkong" in img.attrs["src"]: or "china" in img.attrs["src"]
lan_score += 2 or "jollyroger" in img.attrs["src"]
elif "china" in img.attrs["src"]: ):
lan_score += 4 language = Language("zho")
elif "jollyroger" in img.attrs["src"]: break
lan_score += 8
if lan_score == 1:
language = Language("eng")
else:
language = Language("zho")
sub_page_link = urljoin(self.server_url, a.attrs["href"]) sub_page_link = urljoin(self.server_url, a.attrs["href"])
backup_session = copy.deepcopy(self.session) backup_session = copy.deepcopy(self.session)
backup_session.headers["Referer"] = link backup_session.headers["Referer"] = link
@ -125,10 +120,7 @@ class ZimukuProvider(Provider):
subs.append( subs.append(
self.subtitle_class(language, sub_page_link, name, backup_session) self.subtitle_class(language, sub_page_link, name, backup_session)
) )
lan_scores[name] = lan_score
# prefer double languages
subs.sort(key=lambda s: lan_scores[s.version], reverse=True)
return subs return subs
def query(self, keyword, season=None, episode=None, year=None): def query(self, keyword, season=None, episode=None, year=None):