7 changed files with 139 additions and 15 deletions
After Width: | Height: | Size: 368 B |
@ -0,0 +1,118 @@ |
|||||
|
# 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 <http://www.gnu.org/licenses/>. |
||||
|
|
||||
|
from . import generic |
||||
|
from ..helpers import try_int |
||||
|
|
||||
|
from six import string_types |
||||
|
from _23 import filter_list, map_list, unidecode |
||||
|
|
||||
|
|
||||
|
class SpeedAppProvider(generic.TorrentProvider): |
||||
|
|
||||
|
def __init__(self): |
||||
|
|
||||
|
generic.TorrentProvider.__init__(self, 'SpeedApp') |
||||
|
|
||||
|
self.url_base = 'https://speedapp.io/' |
||||
|
|
||||
|
self.api = self.url_base + 'api/' |
||||
|
self.urls = dict( |
||||
|
config_provider_home_uri=self.url_base, |
||||
|
login=self.api + 'test', |
||||
|
search=self.api + 'torrent?itemsPerPage=50&search=%s&%s', cats=self.api + 'category', |
||||
|
get=self.api + 'torrent/%s/download' |
||||
|
) |
||||
|
|
||||
|
self.perms_needed = self.perms = ('torrent.read', 'torrent.download', 'snatch.read') |
||||
|
self.api_key, self._authd, self.minseed, self.minleech, self.cats = 5 * [None] |
||||
|
|
||||
|
def _authorised(self, **kwargs): |
||||
|
|
||||
|
return super(SpeedAppProvider, self)._authorised( |
||||
|
logged_in=self.logged_in, parse_json=True, headers=self.auth_header(), |
||||
|
failed_msg=(lambda y=None: u'Invalid token or permissions for %s. Check settings')) |
||||
|
|
||||
|
def logged_in(self, resp=None): |
||||
|
|
||||
|
self._authd = None |
||||
|
self.perms_needed = self.perms |
||||
|
if isinstance(resp, dict) and isinstance(resp.get('scopes'), list): |
||||
|
self._authd = True |
||||
|
self.perms_needed = filter_list(lambda x: True is not x, [p in resp.get('scopes') or p for p in self.perms]) |
||||
|
if not self.perms_needed: |
||||
|
self.categories = None |
||||
|
resp = self.get_url(self.urls['cats'], skip_auth=True, parse_json=True, headers=self.auth_header()) |
||||
|
if isinstance(resp, list): |
||||
|
categories = [category['id'] for category in filter_list( |
||||
|
lambda c: isinstance(c.get('id'), int) and isinstance(c.get('name'), string_types) |
||||
|
and c.get('name').upper() in ('TV PACKS', 'TV HD', 'TV SD'), resp)] |
||||
|
self.categories = {'Cache': categories, 'Episode': categories, 'Season': categories} |
||||
|
|
||||
|
return not any(self.perms_needed) |
||||
|
|
||||
|
def auth_header(self): |
||||
|
return {'X-Client-Id': '328b70a3f869e26de994', 'Authorization': 'Bearer %s' % self.api_key} |
||||
|
|
||||
|
def _search_provider(self, search_params, **kwargs): |
||||
|
|
||||
|
results = [] |
||||
|
if not self._authorised() or not self.categories: |
||||
|
return results |
||||
|
|
||||
|
items = {'Cache': [], 'Season': [], 'Episode': [], 'Propers': []} |
||||
|
|
||||
|
for mode in search_params: |
||||
|
for search_string in search_params[mode]: |
||||
|
search_url = self.urls['search'] % ( |
||||
|
unidecode(search_string), self._categories_string(mode, template='categories[]=%s')) |
||||
|
|
||||
|
data_json = self.get_url(search_url, skip_auth=True, parse_json=True, headers=self.auth_header()) |
||||
|
if self.should_skip(): |
||||
|
return results |
||||
|
|
||||
|
cnt = len(items[mode]) |
||||
|
if isinstance(data_json, list): |
||||
|
for tr in data_json or []: |
||||
|
seeders, leechers, size, title, tid = (try_int(n, n) for n in [ |
||||
|
tr.get(x) for x in ('seeders', 'leechers', 'size', 'name', 'id')]) |
||||
|
if not self._reject_item(seeders, leechers) and title and tid: |
||||
|
items[mode].append((title, self._link(tid), seeders, self._bytesizer(size))) |
||||
|
|
||||
|
self._log_search(mode, len(items[mode]) - cnt, search_url) |
||||
|
|
||||
|
results = self._sort_seeding(mode, results + items[mode]) |
||||
|
|
||||
|
return results |
||||
|
|
||||
|
def ui_string(self, key): |
||||
|
return ('%s_api_key' % self.get_id()) == key and 'API Token' or \ |
||||
|
('%s_api_key_tip' % self.get_id()) == key and \ |
||||
|
((not self._authd and not self._authorised() or self.perms_needed) |
||||
|
and ('create token at <a href="%sprofile/api-tokens">%s site</a><br>' |
||||
|
'with perms %s' % (self.url_base, self.name, map_list( |
||||
|
lambda p: 't.read' in p and 'Read torrents' |
||||
|
or 't.down' in p and 'Download torrents' |
||||
|
or 'ch.read' in p and 'Read snatches', self.perms_needed))) |
||||
|
.replace('[', '').replace(']', '') |
||||
|
or 'token is valid and required permissions are enabled') \ |
||||
|
or '' |
||||
|
|
||||
|
|
||||
|
provider = SpeedAppProvider() |
Loading…
Reference in new issue