Browse Source
Change improve RSS validation (particularly for anime). Change improve support for legacy magnet encoding.tags/release_0.12.15^2
6 changed files with 30 additions and 119 deletions
Before Width: | Height: | Size: 1.9 KiB |
@ -1,107 +0,0 @@ |
|||||
# |
|
||||
# This file is part of SickGear. |
|
||||
# |
|
||||
# SickGear is free software: you can redistribute it and/or modify |
|
||||
# it under the terms of the GNU General Public License as published by |
|
||||
# the Free Software Foundation, either version 3 of the License, or |
|
||||
# (at your option) any later version. |
|
||||
# |
|
||||
# SickGear is distributed in the hope that it will be useful, |
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
||||
# GNU General Public License for more details. |
|
||||
# |
|
||||
# You should have received a copy of the GNU General Public License |
|
||||
# along with SickGear. If not, see <http://www.gnu.org/licenses/>. |
|
||||
|
|
||||
import re |
|
||||
import urllib |
|
||||
|
|
||||
from . import generic |
|
||||
from sickbeard import logger, show_name_helpers, tvcache |
|
||||
from sickbeard.helpers import tryInt |
|
||||
|
|
||||
|
|
||||
class NyaaProvider(generic.TorrentProvider): |
|
||||
|
|
||||
def __init__(self): |
|
||||
generic.TorrentProvider.__init__(self, 'NyaaTorrents', anime_only=True) |
|
||||
|
|
||||
self.url_base = self.url = 'https://www.nyaa.se/' |
|
||||
|
|
||||
self.minseed, self.minleech = 2 * [None] |
|
||||
|
|
||||
self.cache = NyaaCache(self) |
|
||||
|
|
||||
def _search_provider(self, search_string, search_mode='eponly', **kwargs): |
|
||||
|
|
||||
if self.show and not self.show.is_anime: |
|
||||
return [] |
|
||||
|
|
||||
params = urllib.urlencode({'term': search_string.encode('utf-8'), |
|
||||
'cats': '1_37', # Limit to English-translated Anime (for now) |
|
||||
# 'sort': '2', # Sort Descending By Seeders |
|
||||
}) |
|
||||
|
|
||||
return self.get_data(getrss_func=self.cache.getRSSFeed, |
|
||||
search_url='%s?page=rss&%s' % (self.url, params), |
|
||||
mode=('Episode', 'Season')['sponly' == search_mode]) |
|
||||
|
|
||||
def get_data(self, getrss_func, search_url, mode='cache'): |
|
||||
|
|
||||
data = getrss_func(search_url) |
|
||||
|
|
||||
results = [] |
|
||||
if data and 'entries' in data: |
|
||||
|
|
||||
rc = dict((k, re.compile('(?i)' + v)) for (k, v) in { |
|
||||
'stats': '(\d+)\W+seed[^\d]+(\d+)\W+leech[^\d]+\d+\W+down[^\d]+([\d.,]+\s\w+)'}.iteritems()) |
|
||||
|
|
||||
for cur_item in data.get('entries', []): |
|
||||
try: |
|
||||
seeders, leechers, size = 0, 0, 0 |
|
||||
stats = rc['stats'].findall(cur_item.get('summary_detail', {'value': ''}).get('value', '')) |
|
||||
if len(stats): |
|
||||
seeders, leechers, size = (tryInt(n, n) for n in stats[0]) |
|
||||
if self._peers_fail(mode, seeders, leechers): |
|
||||
continue |
|
||||
title, download_url = self._title_and_url(cur_item) |
|
||||
download_url = self._link(download_url) |
|
||||
except (AttributeError, TypeError, ValueError, IndexError): |
|
||||
continue |
|
||||
|
|
||||
if title and download_url: |
|
||||
results.append((title, download_url, seeders, self._bytesizer(size))) |
|
||||
|
|
||||
self._log_search(mode, len(results), search_url) |
|
||||
|
|
||||
return self._sort_seeding(mode, results) |
|
||||
|
|
||||
def _season_strings(self, ep_obj, **kwargs): |
|
||||
|
|
||||
return show_name_helpers.makeSceneShowSearchStrings(self.show) |
|
||||
|
|
||||
def _episode_strings(self, ep_obj, **kwargs): |
|
||||
|
|
||||
return self._season_strings(ep_obj) |
|
||||
|
|
||||
|
|
||||
class NyaaCache(tvcache.TVCache): |
|
||||
|
|
||||
def __init__(self, this_provider): |
|
||||
tvcache.TVCache.__init__(self, this_provider) |
|
||||
|
|
||||
self.update_freq = 15 |
|
||||
|
|
||||
def _cache_data(self): |
|
||||
|
|
||||
params = urllib.urlencode({'page': 'rss', # Use RSS page |
|
||||
'order': '1', # Sort Descending By Date |
|
||||
'cats': '1_37' # Limit to English-translated Anime (for now) |
|
||||
}) |
|
||||
|
|
||||
return self.provider.get_data(getrss_func=self.getRSSFeed, |
|
||||
search_url='%s?%s' % (self.provider.url, params)) |
|
||||
|
|
||||
|
|
||||
provider = NyaaProvider() |
|
Loading…
Reference in new issue