From 6097457aca5544287eacd985550cca38d9c90cdc Mon Sep 17 00:00:00 2001
From: adaur <adaur@users.noreply.github.com>
Date: Sun, 24 Sep 2017 23:40:58 -0400
Subject: [PATCH] Add provider YggTorrent (#3963)
* Add provider YggTorrent
* Fix connection method + leechers / seeders values (#4034)
---
gui/slick/images/providers/yggtorrent.png | Bin 0 -> 2150 bytes
sickbeard/providers/__init__.py | 4 +-
sickbeard/providers/yggtorrent.py | 151 ++++++++++++++++++++++
3 files changed, 153 insertions(+), 2 deletions(-)
create mode 100644 gui/slick/images/providers/yggtorrent.png
create mode 100644 sickbeard/providers/yggtorrent.py
diff --git a/gui/slick/images/providers/yggtorrent.png b/gui/slick/images/providers/yggtorrent.png
new file mode 100644
index 0000000000000000000000000000000000000000..de04ec0b111250b9610fd5fb2d26c8435c487057
GIT binary patch
literal 2150
zcmeAS@N?(olHy`uVBq!ia0y~yU{C;I4mJh`hT^KKFANMEx}GkMArY-xr)3vNUll!G
zzdLH}lBqi0o1TUz-MrD4`{+ug)kfhHvzdau4BvjxxZEf<gU2WRQr!!lzfSGjG>uzl
z`n1iQc72I%NswHo^Fmb{pWy3^LTX3M6hq3Mgy)=|IDgZU*pu3C^Xx4WGhQ8;a^P>t
zd)w-B|F?giSN*QlH7YkZ_u{*|yT8XuOrJK*>ifI9xA!I;?K*X|#@=68{CC3ZYis>E
z14<KZ%*>{}eEXJnecaw(6W7Zu*t_@c_BG3%J==c!*0-9ne?LAx)?XL3^HbaM`E^wn
z<##AGY}&L*>-450GiT1+tT)|z<@)vUd>w6xFDff5UG=7)u3XA+>1of+oUmY#*0ZzC
z?FBzPHD4Bda+a@jbll1nE7qJ(+q}#7`q%X$2?+@bmoDG7zIpX#W%lP?rWv!2UHNcA
zch$6IJ9hk-|NGn9)3KTjZ;OsGv-8d9T69Bo_KwAgKZ`<4TW9$R&z_+m)Okm~`{=Lr
zA_*5&tg61ecq|=d*5`hHuJXJ~E7q)1yZ`#HZ+Ut7?WtXdm7*CEB;5Sw&;1yAGSX{r
zp4%_w8BG%ZjgHuhH{8ve&Qvyg&(bUXmtQ_|u=35)d;ND4gV<dr&q+&O_{_<uX<K7=
z{K*rMO?Th@-o&6cyXEN7qpG*J=ih&7{Mg!>{a44?zQp2Pg-?HGXT5$c&EnYM=i}qT
z__~{kW4h566Z89<*X?-GnUa5FvSv|HQUcTZ!*0pG9sa+P?kuUWkk(O&4*GV@o1K5#
zvdfuoyBT7NxSRB@_a0o>aa=X(h_L29v66tN(=AN}jxKv$k#oVq);He$`1SHPH$L9@
zue?g5>(Zr5N(WUZ1q6zu>Mg%-ls#ibTkBbo6?#()bsA4haoc&hleJ`->6^024;LO}
zfBpJ(y*OXHe~irPb?esc+_meM)agwno|~rbjy?A^``X(0TlenWW9rf82nY-eoZs7f
zN|5DE6R)y*Q0N1db*UduC>6K0X3gMvA$jVqo!jzdJ6A1yc(K5u=5DKm-Td>{FI_5%
z_Fee(vfBBTrzXCXx%_RGkBX+vuMda$?dP3L+2Xyf;cS}rmxSaZKlOQM1E)QA5n#Eq
z$s=H*#Ecov8+3A-8hp0|J}9_w|Nj2sx3{+5ww3D-3Y6Nh`_%NEn&PjE%Gz}`iYD;o
zWM#j$_;}z!g8P%rQQS-s(^OBI&FrysPZjJrw@lLahGqd*$gejlo2?#Jv=uFHc`@zd
z!z2qc&&yYrzSo$k_~fOW(Y&iO6*CWWy9Lziu1SBoeZKt<nS{7CTNN5Kx*o^vZ|8N?
zm(1=7;kqY#X-C~m>2?)?FAjRy>6sf^VrQFu=-wWmdNyeCjer@S+I*FjFTZrZSO0Tw
z^XAQ)yTi_~G6t=@QW6?#=@<NbZT)eMjtMiG?Q2cEgJ(`YGN<+O%HqeLQbl&$JtN8I
z*PGGc;3TIR6#8}Xd1b!o7Ht<ZOriopoYc}6q(6WDNk&ZD%a`k~R$ckDi7f|DUX;jh
ztNv(xCq_f_@ZrNnk&!nS255wA4S3UiyJ_l%ee2fk+`8@;`zGG1$IVxCP6mmbxMZJt
z=^p!DVWz0C1^HQRD?^%=eM+*E=&XHZZ)Z2}`i&bmvLdfZ8yg!>p3XdhfyGfk_q37(
zkJ!5WtX+!_JiZb7W{dFba}TB!Z~4mQvw}%n+|N@aGfHU5g|&bDADlaPF2LFO@W%j+
zlK&Hx-T5{-Fx7E=SeqZSeR+{*!S4AVO&>SQd(FIQ)A#N9jML97%v2_tihoH}n|pA%
z#*1Tz!VkW@ynL_S{P#z#-|yL+TToDNeq-|Sf2onz>RE-~?S8*+Z{L{@vYijtO%kzN
z7qxZrQqilI+v9&{J6xOeBsR4BY}(;X&s<X?)#Umgm+aEe(NVCJTYToyOkO=jx%ua<
z_uoH%<;#~nhbs5qx)mkjYFWK$^XDZKjeRzBxF}Vtc*{?#`*PP<wf+9t<lDb=mQ~JZ
zlQMi@ntijRbV}+%VVkeNwATH*w|U3jpL3+;7JYC2zH`}EGrNlCx>~PZX=<*2a`$?C
zvTHH-&qk+PeESaQUccL5eg5=BXZ_7jpG8-G{_ekV)7NhS3BLImuAkL(PCwaL{r-o`
zo0sAI-o;GPt9;m$)xNTtMZF8k%zv-8B*NtW(ocU*%YXX(`@P!AgX`sXOh5F>Ew1X5
z=d}Di^F9_A^1Z$9U;EkbSn_6>Tl?ql`g(iD9FI@0@A7}@5C7LQZELgoi6a$F>h<3Q
z79V~rQ}=9pXWLWW*+=*7+xOiqIrZ7Dpw7p!YH2dlez;^Ynf2^&IH4ajZ@ICViHleB
z#eK#qrp+gBsA%$T;j#Ga?YY#^_x$&(TN-9HPQ4xVaeBz4OACsgZe1bS!RO+!goo|i
zoXW5Kzvt(Q9zS;M(cRiUJ!3xgDxMjlJGMnV`gYYwg=L%0<R9mxBu_Uz`6(TLx?$dI
z`D{rJ)4ZjR^V<W?fA9bISARujz*U}Ic@si(pYf<?rQM77ZB9IJ;>3w992d;zaQeGB
zdPh$XFu0VU#@f=-u#$6z)_kUj=Zu@$JhXUk$$Tl9f8;6u!qf%JU#qNhV9k=U;<I)C
ze_Vdjl?RWLWiwt@HiSs^d`@>{`Ecr6gSM{0#7#5$4)%xdtXORAoh4Lv<HeHf3ob<(
zOH4Lc-}zF1$My1dd8MLN-D{qvY1GX6b0MlI$<En!x#o&2x9xXoKC9QKoVMMsGPTX)
zdG%*)1&$ZXxZ5Yib13B8*%5ea@9W>PJCC#&nPg9RF5olYS)g5L+RNgGaIqtsJx@<b
z5${^hQJThc+2Dg%XveEFmcbvz7A>EA;?y7K$m>EUy23?Tek#qrYN~RxRjmKv8SVvs
puD+@PY0q}2zMXmBZsS|~$3K<_`hR=Y#K6G7;OXk;vd$@?2>`UTMC<?n
literal 0
HcmV?d00001
diff --git a/sickbeard/providers/__init__.py b/sickbeard/providers/__init__.py
index cdd98b9ee..1f3906080 100644
--- a/sickbeard/providers/__init__.py
+++ b/sickbeard/providers/__init__.py
@@ -28,7 +28,7 @@ from sickbeard.providers import (abnormal, alpharatio, archetorrent, binsearch,
hd4free, hdbits, hdspace, hdtorrents, hdtorrents_it, horriblesubs, hounddawgs, ilcorsaronero, immortalseed, iptorrents,
limetorrents, morethantv, ncore, nebulance, newpct, norbits, nyaa, omgwtfnzbs, pretome, rarbg, scc, scenetime, shazbat,
skytorrents, speedcd, thepiratebay, tntvillage, tokyotoshokan, torrent9, torrentbytes, torrentday, torrentleech,
- torrentproject, torrentz, tvchaosuk, xthor)
+ torrentproject, torrentz, tvchaosuk, xthor, yggtorrent)
__all__ = [
'abnormal', 'alpharatio', 'archetorrent', 'binsearch', 'bitcannon', 'btn', 'cpasbien', 'danishbits',
@@ -36,7 +36,7 @@ __all__ = [
'horriblesubs', 'hounddawgs', 'ilcorsaronero', 'immortalseed', 'iptorrents', 'limetorrents', 'morethantv',
'ncore', 'nebulance', 'newpct', 'norbits', 'nyaa', 'omgwtfnzbs', 'pretome', 'rarbg', 'scc', 'scenetime',
'shazbat', 'skytorrents', 'speedcd', 'thepiratebay', 'tntvillage', 'tokyotoshokan', 'torrent9',
- 'torrentbytes', 'torrentday', 'torrentleech', 'torrentproject', 'torrentz', 'tvchaosuk', 'xthor'
+ 'torrentbytes', 'torrentday', 'torrentleech', 'torrentproject', 'torrentz', 'tvchaosuk', 'xthor', 'yggtorrent'
]
diff --git a/sickbeard/providers/yggtorrent.py b/sickbeard/providers/yggtorrent.py
new file mode 100644
index 000000000..c76533bdc
--- /dev/null
+++ b/sickbeard/providers/yggtorrent.py
@@ -0,0 +1,151 @@
+# coding=utf-8
+# Author: adaur <adaur.underground@gmail.com>
+#
+# URL: https://sickrage.github.io
+#
+# This file is part of SickRage.
+#
+# SickRage 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.
+#
+# SickRage 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 SickRage. If not, see <http://www.gnu.org/licenses/>.
+
+from __future__ import unicode_literals
+
+import re
+
+from requests.compat import urljoin
+from requests.utils import dict_from_cookiejar
+
+from sickbeard import logger, tvcache
+from sickbeard.bs4_parser import BS4Parser
+from sickrage.helper.common import convert_size, try_int
+from sickrage.providers.torrent.TorrentProvider import TorrentProvider
+
+
+class YggTorrentProvider(TorrentProvider): # pylint: disable=too-many-instance-attributes
+
+ def __init__(self):
+
+ # Provider Init
+ TorrentProvider.__init__(self, 'YggTorrent')
+
+ # Credentials
+ self.username = None
+ self.password = None
+
+ # Torrent Stats
+ self.minseed = None
+ self.minleech = None
+
+ # URLs
+ self.url = 'https://yggtorrent.com/'
+ self.urls = {
+ 'login': urljoin(self.url, 'user/login'),
+ 'search': urljoin(self.url, 'engine/search'),
+ }
+
+ # Proper Strings
+ self.proper_strings = ['PROPER']
+
+ # Cache
+ self.cache = tvcache.TVCache(self, min_time=30)
+
+ def login(self):
+ if any(dict_from_cookiejar(self.session.cookies).values()):
+ return True
+
+ login_params = {
+ 'id': self.username,
+ 'pass': self.password,
+ }
+
+ response = self.get_url(self.urls['login'], post_data=login_params, returns='text')
+ if response: # Yggtorrent return empty response if user is logged, so ...
+ logger.log('Invalid username or password. Check your settings', logger.WARNING)
+ return False
+
+ return True
+
+ def search(self, search_strings, age=0, ep_obj=None): # pylint: disable=too-many-locals, too-many-branches
+ results = []
+ if not self.login():
+ return results
+
+ for mode in search_strings:
+ items = []
+ logger.log('Search Mode: {0}'.format(mode), logger.DEBUG)
+
+ for search_string in search_strings[mode]:
+
+ if mode != 'RSS':
+ logger.log('Search string: {0}'.format
+ (search_string.decode('utf-8')), logger.DEBUG)
+
+ try:
+ search_params = {
+ 'q': re.sub(r'[()]', '', search_string)
+ }
+ data = self.get_url(self.urls['search'], params=search_params, returns='text')
+ if not data:
+ continue
+
+ with BS4Parser(data, 'html5lib') as html:
+ torrent_table = html.find(class_='table table-striped')
+ torrent_rows = torrent_table('tr') if torrent_table else []
+
+ # Continue only if at least one Release is found
+ if len(torrent_rows) < 2:
+ logger.log('Data returned from provider does not contain any torrents', logger.DEBUG)
+ continue
+
+ # Skip column headers
+ for result in torrent_rows[1:]:
+ cells = result('td')
+ if len(cells) < 5:
+ continue
+
+ title = cells[0].find('a', class_='torrent-name').get_text(strip=True)
+ download_url = urljoin(self.url, cells[0].find('a', target='_blank')['href'])
+ if not (title and download_url):
+ continue
+
+ seeders = try_int(cells[4].get_text(strip=True))
+ leechers = try_int(cells[5].get_text(strip=True))
+
+ torrent_size = cells[2].get_text()
+ size = convert_size(torrent_size) or -1
+
+ # Filter unseeded torrent
+ if seeders < self.minseed or leechers < self.minleech:
+ if mode != 'RSS':
+ logger.log('Discarding torrent because it doesn\'t meet the minimum seeders or leechers: {0} (S:{1} L:{2})'.format
+ (title, seeders, leechers), logger.DEBUG)
+ continue
+
+ item = {'title': title, 'link': download_url, 'size': size, 'seeders': seeders, 'leechers': leechers, 'hash': ''}
+ if mode != 'RSS':
+ logger.log('Found result: {0} with {1} seeders and {2} leechers'.format
+ (title, seeders, leechers), logger.DEBUG)
+
+ items.append(item)
+
+ except (AttributeError, TypeError, KeyError, ValueError):
+ logger.log('Failed parsing provider {}.'.format(self.name), logger.ERROR)
+
+ # For each search mode sort all the items by seeders if available
+ items.sort(key=lambda d: try_int(d.get('seeders', 0)), reverse=True)
+ results += items
+
+ return results
+
+
+provider = YggTorrentProvider()
--
GitLab