mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-05 02:43:37 +00:00
ee458bd211
Show root dirs can not be set from general config menu. Mass editing shows now has the ability to delete root dirs as well as edit them. Daily search no longer is restricted to just 1 week of results for searching from which now allows for replacing lower quality downloads with higher quality ones if available. RSS Cache is updated for each provider on demand now when performing manual, failed, backlog, or daily searches.
220 lines
8 KiB
Python
220 lines
8 KiB
Python
# 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/>.
|
|
|
|
import re
|
|
import time
|
|
import datetime
|
|
import urllib
|
|
import urlparse
|
|
import sys
|
|
import generic
|
|
import sickbeard
|
|
|
|
from lib import requests
|
|
from lib.requests import exceptions
|
|
from sickbeard import classes
|
|
from sickbeard import logger, tvcache, exceptions
|
|
from sickbeard import helpers
|
|
from sickbeard import clients
|
|
from sickbeard.common import cpu_presets
|
|
from sickbeard.exceptions import ex, AuthException
|
|
try:
|
|
import json
|
|
except ImportError:
|
|
from lib import simplejson as json
|
|
|
|
|
|
class HDBitsProvider(generic.TorrentProvider):
|
|
def __init__(self):
|
|
|
|
generic.TorrentProvider.__init__(self, "HDBits")
|
|
|
|
self.supportsBacklog = True
|
|
|
|
self.enabled = False
|
|
self.username = None
|
|
self.passkey = None
|
|
self.ratio = None
|
|
|
|
self.cache = HDBitsCache(self)
|
|
|
|
self.url = 'https://hdbits.org'
|
|
self.search_url = 'https://hdbits.org/api/torrents'
|
|
self.rss_url = 'https://hdbits.org/api/torrents'
|
|
self.download_url = 'https://hdbits.org/download.php?'
|
|
|
|
def isEnabled(self):
|
|
return self.enabled
|
|
|
|
def _checkAuth(self):
|
|
|
|
if not self.username or not self.passkey:
|
|
raise AuthException("Your authentication credentials for " + self.name + " are missing, check your config.")
|
|
|
|
return True
|
|
|
|
def _checkAuthFromData(self, parsedJSON):
|
|
|
|
if 'status' in parsedJSON and 'message' in parsedJSON:
|
|
if parsedJSON.get('status') == 5:
|
|
logger.log(u"Incorrect authentication credentials for " + self.name + " : " + parsedJSON['message'],
|
|
logger.DEBUG)
|
|
raise AuthException(
|
|
"Your authentication credentials for " + self.name + " are incorrect, check your config.")
|
|
|
|
return True
|
|
|
|
def _get_season_search_strings(self, ep_obj):
|
|
season_search_string = [self._make_post_data_JSON(show=ep_obj.show, season=ep_obj)]
|
|
return season_search_string
|
|
|
|
def _get_episode_search_strings(self, ep_obj, add_string=''):
|
|
episode_search_string = [self._make_post_data_JSON(show=ep_obj.show, episode=ep_obj)]
|
|
return episode_search_string
|
|
|
|
def _get_title_and_url(self, item):
|
|
|
|
title = item['name']
|
|
if title:
|
|
title = u'' + title
|
|
title = title.replace(' ', '.')
|
|
|
|
url = self.download_url + urllib.urlencode({'id': item['id'], 'passkey': self.passkey})
|
|
|
|
return (title, url)
|
|
|
|
def _doSearch(self, search_params, search_mode='eponly', epcount=0, age=0):
|
|
results = []
|
|
|
|
self._checkAuth()
|
|
|
|
logger.log(u"Search url: " + self.search_url + " search_params: " + search_params, logger.DEBUG)
|
|
|
|
parsedJSON = self.getURL(self.search_url, post_data=search_params, json=True)
|
|
if not parsedJSON:
|
|
return []
|
|
|
|
if self._checkAuthFromData(parsedJSON):
|
|
if parsedJSON and 'data' in parsedJSON:
|
|
items = parsedJSON['data']
|
|
else:
|
|
logger.log(u"Resulting JSON from " + self.name + " isn't correct, not parsing it", logger.ERROR)
|
|
items = []
|
|
|
|
for item in items:
|
|
results.append(item)
|
|
|
|
return results
|
|
|
|
def findPropers(self, search_date=None):
|
|
results = []
|
|
|
|
search_terms = [' proper ', ' repack ']
|
|
|
|
for term in search_terms:
|
|
for item in self._doSearch(self._make_post_data_JSON(search_term=term)):
|
|
if item['utadded']:
|
|
try:
|
|
result_date = datetime.datetime.fromtimestamp(int(item['utadded']))
|
|
except:
|
|
result_date = None
|
|
|
|
if result_date:
|
|
if not search_date or result_date > search_date:
|
|
title, url = self._get_title_and_url(item)
|
|
results.append(classes.Proper(title, url, result_date, self.show))
|
|
|
|
return results
|
|
|
|
def _make_post_data_JSON(self, show=None, episode=None, season=None, search_term=None):
|
|
|
|
post_data = {
|
|
'username': self.username,
|
|
'passkey': self.passkey,
|
|
'category': [2],
|
|
# TV Category
|
|
}
|
|
|
|
if episode:
|
|
if show.air_by_date:
|
|
post_data['tvdb'] = {
|
|
'id': show.indexerid,
|
|
'episode': str(episode.airdate).replace('-', '|')
|
|
}
|
|
elif show.sports:
|
|
post_data['tvdb'] = {
|
|
'id': show.indexerid,
|
|
'episode': episode.airdate.strftime('%b')
|
|
}
|
|
elif show.anime:
|
|
post_data['tvdb'] = {
|
|
'id': show.indexerid,
|
|
'episode': "%i" % int(episode.scene_absolute_number)
|
|
}
|
|
else:
|
|
post_data['tvdb'] = {
|
|
'id': show.indexerid,
|
|
'season': episode.scene_season,
|
|
'episode': episode.scene_episode
|
|
}
|
|
|
|
if season:
|
|
if show.air_by_date or show.sports:
|
|
post_data['tvdb'] = {
|
|
'id': show.indexerid,
|
|
'season': str(season.airdate)[:7],
|
|
}
|
|
elif show.anime:
|
|
post_data['tvdb'] = {
|
|
'id': show.indexerid,
|
|
'season': "%d" % season.scene_absolute_number,
|
|
}
|
|
else:
|
|
post_data['tvdb'] = {
|
|
'id': show.indexerid,
|
|
'season': season.scene_season,
|
|
}
|
|
|
|
if search_term:
|
|
post_data['search'] = search_term
|
|
|
|
return json.dumps(post_data)
|
|
|
|
def seedRatio(self):
|
|
return self.ratio
|
|
|
|
|
|
class HDBitsCache(tvcache.TVCache):
|
|
def __init__(self, provider):
|
|
|
|
tvcache.TVCache.__init__(self, provider)
|
|
|
|
# only poll HDBits every 15 minutes max
|
|
self.minTime = 15
|
|
|
|
def _getRSSData(self):
|
|
parsedJSON = self.provider.getURL(self.provider.rss_url, post_data=self.provider._make_post_data_JSON(), json=True)
|
|
|
|
if not self.provider._checkAuthFromData(parsedJSON):
|
|
return []
|
|
|
|
if parsedJSON and 'data' in parsedJSON:
|
|
return parsedJSON['data']
|
|
else:
|
|
return []
|
|
|
|
|
|
|
|
provider = HDBitsProvider()
|