From 45d1dce037fa5336384335e4a28451bce7cf4384 Mon Sep 17 00:00:00 2001 From: JackDandy Date: Mon, 22 Jun 2015 01:00:38 +0100 Subject: [PATCH] Add TorrentShack torrent provider. --- CHANGES.md | 1 + gui/slick/images/providers/torrentshack.png | Bin 0 -> 861 bytes sickbeard/__init__.py | 2 +- sickbeard/providers/__init__.py | 3 +- sickbeard/providers/torrentshack.py | 151 ++++++++++++++++++++ 5 files changed, 155 insertions(+), 2 deletions(-) create mode 100644 gui/slick/images/providers/torrentshack.png create mode 100644 sickbeard/providers/torrentshack.py diff --git a/CHANGES.md b/CHANGES.md index ee670737..b2ff124b 100644 --- a/CHANGES.md +++ b/CHANGES.md @@ -55,6 +55,7 @@ * Add AlphaRatio torrent provider * Add PiSexy torrent provider * Add Strike torrent provider +* Add TorrentShack torrent provider * Fix getManualSearchStatus: object has no attribute 'segment' * Change handling of general HTTP error response codes to prevent issues * Add handling for CloudFlare custom HTTP response codes diff --git a/gui/slick/images/providers/torrentshack.png b/gui/slick/images/providers/torrentshack.png new file mode 100644 index 0000000000000000000000000000000000000000..f9ae440dec05904ba26208bbcf223459e7793735 GIT binary patch literal 861 zcmeAS@N?(olHy`uVBq!ia0vp^0wB!63?wyl`GbKJOS+@4BLl<6e(pbstUx|vage(c z!@6@aFM%9|WRD45bDP46hOx7_4S6Fo+k-*%fF5)W;X# z6XL3A>a1htq+{Z!YwDzJ>ZonvtYhq|ZS0_9Vy|uDs$t@wW9+PD?4)VxsB2=UY2v17 z>Z)a8r(^1(W#+Pe+n)8C_pIBxTieJ z)_v=??OeZU_n9^(9U})l6BjLGms2ea+QxR9Htjjr%5<)kW%Kp}TXr1WuzmlgZTr`3 z-g~BvLD$ql%g9y_=p#cHO=H{t|NrN<_~n3tG|AiDC34oLFA_iwXMsm#F#`jCA_z0C zuwR}D6#U}p;usC({aNkZ4*NoNd*80;q!$~>yqEK|+%#8OFQaBg7 zom*EaY-QGNe1B4GYOl!Be02#uz5wO*8@e_>@s+xG{lV-#g`8(recvg5CP~dn+`;K; z#=2?FZo%8n?7r5>SKDPV-{)j+ynBURCR0M$y8qg*fzo%6{r%9nmordq>gf$AT}{US2M_Z5gM# z%3hQy=H$^Nwzok4tCqM%l%ynPkeY6loS&0l zR9ujl475e3G;}^tQUNYmoS0jXlbM&Uo0gbdl3&DNSYfCG#V@D+DFB(|>FVdQ&MBb@ E0CLDg`~Uy| literal 0 HcmV?d00001 diff --git a/sickbeard/__init__.py b/sickbeard/__init__.py index 2524267b..720edaa8 100755 --- a/sickbeard/__init__.py +++ b/sickbeard/__init__.py @@ -37,7 +37,7 @@ from sickbeard import providers, metadata, config, webserveInit from sickbeard.providers.generic import GenericProvider from providers import btn, newznab, womble, thepiratebay, torrentleech, kat, iptorrents, \ omgwtfnzbs, scc, torrentday, hdbits, speedcd, nyaatorrents, torrentbytes, \ - freshontv, bitsoup, tokyotoshokan, animenzb, totv, rarbg, morethan, alpharatio, pisexy, strike + freshontv, bitsoup, tokyotoshokan, animenzb, totv, rarbg, morethan, alpharatio, pisexy, strike, torrentshack from sickbeard.config import CheckSection, check_setting_int, check_setting_str, check_setting_float, ConfigMigrator, \ naming_ep_type, minimax from sickbeard import searchBacklog, showUpdater, versionChecker, autoPostProcesser, \ diff --git a/sickbeard/providers/__init__.py b/sickbeard/providers/__init__.py index ade7930d..400799d0 100755 --- a/sickbeard/providers/__init__.py +++ b/sickbeard/providers/__init__.py @@ -38,7 +38,8 @@ __all__ = ['womble', 'morethan', 'alpharatio', 'pisexy', - 'strike' + 'strike', + 'torrentshack', ] from os import sys diff --git a/sickbeard/providers/torrentshack.py b/sickbeard/providers/torrentshack.py new file mode 100644 index 00000000..6ba2bc59 --- /dev/null +++ b/sickbeard/providers/torrentshack.py @@ -0,0 +1,151 @@ +# coding=utf-8 +# +# Author: SickGear +# +# 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 . + +import re +import datetime +import traceback + +from . import generic +from sickbeard import logger, tvcache, helpers +from sickbeard.bs4_parser import BS4Parser +from lib.unidecode import unidecode + + +class TorrentShackProvider(generic.TorrentProvider): + + def __init__(self): + generic.TorrentProvider.__init__(self, 'TorrentShack') + + self.url_base = 'https://torrentshack.me/' + self.urls = {'config_provider_home_uri': self.url_base, + 'login': self.url_base + 'login.php?lang=', + 'search': self.url_base + 'torrents.php?searchstr=%s' + + '&release_type=both&searchtags=&tags_type=0&order_by=s3&order_way=desc&torrent_preset=all' + + '&filter_cat[600]=1&filter_cat[620]=1&filter_cat[700]=1' + + '&filter_cat[850]=1&filter_cat[980]=1&filter_cat[981]=1', + 'get': self.url_base + '%s'} + + self.url = self.urls['config_provider_home_uri'] + + self.username, self.password, self.minseed, self.minleech = 4 * [None] + self.cache = TorrentShackCache(self) + + def _doLogin(self): + + logged_in = lambda: 'session' in self.session.cookies + if logged_in(): + return True + + if self._checkAuth(): + login_params = {'username': self.username, 'password': self.password, 'keeplogged': '1', 'login': 'Login'} + response = helpers.getURL(self.urls['login'], post_data=login_params, session=self.session) + if response and logged_in(): + return True + + msg = u'Failed to authenticate with %s, abort provider' + if response and 'username or password was incorrect' in response: + msg = u'Invalid username or password for %s. Check settings' + logger.log(msg % self.name, logger.ERROR) + + return False + + def _doSearch(self, search_params, search_mode='eponly', epcount=0, age=0): + + results = [] + if not self._doLogin(): + return results + + items = {'Season': [], 'Episode': [], 'Cache': []} + + rc = dict((k, re.compile('(?i)' + v)) + for (k, v) in {'info': 'view', 'get': 'download', 'title': 'view\s+torrent\s+'}.items()) + for mode in search_params.keys(): + for search_string in search_params[mode]: + + if isinstance(search_string, unicode): + search_string = unidecode(search_string) + + # fetch 15 results by default, and up to 100 if allowed in user profile + search_url = self.urls['search'] % search_string + html = self.getURL(search_url) + + cnt = len(items[mode]) + try: + if not html or self._has_no_results(html): + raise generic.HaltParseException + + with BS4Parser(html, features=['html5lib', 'permissive']) as soup: + torrent_table = soup.find('table', attrs={'class': 'torrent_table'}) + torrent_rows = [] if not torrent_table else torrent_table.find_all('tr') + + if 2 > len(torrent_rows): + raise generic.HaltParseException + + for tr in torrent_rows[1:]: + try: + seeders, leechers = [int(tr.find_all('td')[x].get_text().strip()) for x in (-2, -1)] + if 'Cache' != mode and (seeders < self.minseed or leechers < self.minleech): + continue + + info = tr.find('a', title=rc['info']) + title = 'title' in info.attrs and rc['title'].sub('', info.attrs['title']) \ + or info.get_text().strip() + + link = str(tr.find('a', title=rc['get'])['href']).replace('&', '&').lstrip('/') + download_url = self.urls['get'] % link + except (AttributeError, TypeError): + continue + + if title and download_url: + items[mode].append((title, download_url, seeders)) + + except generic.HaltParseException: + pass + except Exception: + logger.log(u'Failed to parse. Traceback: %s' % traceback.format_exc(), logger.ERROR) + self._log_result(mode, len(items[mode]) - cnt, search_url) + + # for each search mode sort all the items by seeders + items[mode].sort(key=lambda tup: tup[2], reverse=True) + + results += items[mode] + + return results + + def findPropers(self, search_date=datetime.datetime.today()): + + return self._find_propers(search_date) + + def _get_episode_search_strings(self, ep_obj, add_string='', **kwargs): + + return generic.TorrentProvider._get_episode_search_strings(self, ep_obj, add_string, sep_date='.', use_or=False) + + +class TorrentShackCache(tvcache.TVCache): + + def __init__(self, this_provider): + tvcache.TVCache.__init__(self, this_provider) + + self.minTime = 20 # cache update frequency + + def _getRSSData(self): + + return self.provider.get_cache_data() + +provider = TorrentShackProvider()