dinislage/CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use urlopen for HD Bits requests

Browse Source
This commit is contained in:
Ruud
2014-01-23 22:00:12 +01:00
parent c55404699e
commit e20c776364
1 changed files with 17 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
couchpotato/core/providers/torrent/hdbits/main.py
View File

@@ -4,7 +4,6 @@ from couchpotato.core.providers.torrent.base import TorrentProvider
import re
import json
import requests
import traceback
log = CPLog(__name__)
@@ -13,20 +12,12 @@ log = CPLog(__name__)
class HDBits(TorrentProvider):
urls = {
'test' : 'https://hdbits.org/',
'detail' : 'https://hdbits.org/details.php?id=%s',
'download' : 'https://hdbits.org/download.php?id=%s&passkey=%s',
'test': 'https://hdbits.org/',
'detail': 'https://hdbits.org/details.php?id=%s',
'download': 'https://hdbits.org/download.php?id=%s&passkey=%s',
'api': 'https://hdbits.org/api/torrents'
}
status_codes = {
1: 'Failure (something bad happened)',
2: 'SSL Required - You should be on https://',
3: 'JSON Malformed',
4: 'Auth data missing',
5: 'Auth failed'
}
http_time_between_calls = 1 #seconds
def _post_query(self, **params):
@@ -37,23 +28,24 @@ class HDBits(TorrentProvider):
}
post_data.update(params)
req = requests.post(self.urls['api'],
data=json.dumps(post_data))
try:
result = self.getJsonData(self.urls['api'], data = json.dumps(post_data))
response = json.loads(req.text)
if result:
if result['status'] != 0:
log.error('Error searching hdbits: %s' % result['message'])
else:
return result['data']
except:
pass
if response['status'] != 0:
log.error('Error searching hdbits: %s'
% self.status_codes[response['status']])
else:
return response['data']
return None
def _search(self, movie, quality, results):
match = re.match(r'tt(\d{7})',
movie['library']['identifier'])
match = re.match(r'tt(\d{7})', movie['library']['identifier'])
data = self._post_query(imdb={'id': match.group(1)})
data = self._post_query(imdb = {'id': match.group(1)})
if data:
try:
@@ -61,14 +53,11 @@ class HDBits(TorrentProvider):
results.append({
'id': result['id'],
'name': result['name'],
'url': self.urls['download']
% (result['id'], self.conf('passkey')),
'url': self.urls['download'] % (result['id'], self.conf('passkey')),
'detail_url': self.urls['detail'] % result['id'],
'size': self.parseSize(result['size']),
'seeders': tryInt(result['seeders']),
'leechers': tryInt(result['leechers'])
})
except:
log.error('Failed getting results from %s: %s',
(self.getName(), traceback.format_exc()))
log.error('Failed getting results from %s: %s', (self.getName(), traceback.format_exc()))