Ruud Burger
8 years ago
committed by
GitHub
GitHub
17 changed files with 390 additions and 24 deletions
@ -0,0 +1,136 @@ |
|||
from datetime import datetime |
|||
from couchpotato.core.helpers.variable import tryInt |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.helpers.variable import getTitle |
|||
from couchpotato.core.media._base.providers.torrent.base import TorrentMagnetProvider |
|||
import random |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
|
|||
class Base(TorrentMagnetProvider): |
|||
# Only qualities allowed: 720p/1080p/3D - the rest will fail. |
|||
# All YTS.ag torrents are verified |
|||
urls = { |
|||
'detail': 'https://yts.ag/api#list_movies', |
|||
'search': 'https://yts.ag/api/v2/list_movies.json?query_term=%s&limit=%s&page=%s' |
|||
} |
|||
|
|||
def _search(self, movie, quality, results): |
|||
limit = 10 |
|||
page = 1 |
|||
data = self.getJsonData(self.urls['search'] % (getTitle(movie), limit, page)) |
|||
|
|||
if data: |
|||
movie_count = tryInt(data['data']['movie_count']) |
|||
|
|||
if movie_count == 0: |
|||
log.debug('%s - found no results', (self.getName())) |
|||
else: |
|||
|
|||
movie_results = data['data']['movies'] |
|||
for i in range(0,len(movie_results)): |
|||
result = data['data']['movies'][i] |
|||
name = result['title'] |
|||
|
|||
t = movie['info']['original_title'].split(' ') |
|||
|
|||
if all(word in name for word in t) and movie['info']['year'] == result['year']: |
|||
|
|||
year = result['year'] |
|||
detail_url = result['url'] |
|||
|
|||
for torrent in result['torrents']: |
|||
t_quality = torrent['quality'] |
|||
|
|||
if t_quality in quality['label']: |
|||
hash = torrent['hash'] |
|||
size = tryInt(torrent['size_bytes'] / 1048576) |
|||
seeders = tryInt(torrent['seeds']) |
|||
leechers = tryInt(torrent['peers']) |
|||
pubdate = torrent['date_uploaded'] # format: 2017-02-17 18:40:03 |
|||
pubdate = datetime.strptime(pubdate, '%Y-%m-%d %H:%M:%S') |
|||
age = (datetime.now() - pubdate).days |
|||
|
|||
results.append({ |
|||
'id': random.randint(100, 9999), |
|||
'name': '%s (%s) %s %s %s' % (name, year, 'YTS', t_quality, 'BR-Rip'), |
|||
'url': self.make_magnet(hash, name), |
|||
'size': size, |
|||
'seeders': seeders, |
|||
'leechers': leechers, |
|||
'age': age, |
|||
'detail_url': detail_url, |
|||
'score': 1 |
|||
}) |
|||
|
|||
return |
|||
|
|||
def make_magnet(self, hash, name): |
|||
url_encoded_trackers = 'udp%3A%2F%2Fopen.demonii.com%3A1337%2Fannounce&tr=%0Audp%3A%2F%2Ftracker.openbittorr' \ |
|||
'ent.com%3A80&tr=%0Audp%3A%2F%2Ftracker.coppersurfer.tk%3A6969&tr=%0Audp%3A%2F%2Fglot' \ |
|||
'orrents.pw%3A6969%2Fannounce&tr=%0Audp%3A%2F%2Ftracker.opentrackr.org%3A1337%2Fannou' \ |
|||
'nce&tr=%0Audp%3A%2F%2Ftorrent.gresille.org%3A80%2Fannounce&tr=%0Audp%3A%2F%2Fp4p.are' \ |
|||
'nabg.com%3A1337&tr=%0Audp%3A%2F%2Ftracker.leechers-paradise.org%3A6969]' |
|||
|
|||
return 'magnet:?xt=urn:btih:%s&dn=%s&tr=%s' % (hash, name.replace(' ', '+'), url_encoded_trackers) |
|||
|
|||
|
|||
config = [{ |
|||
'name': 'yts', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'searcher', |
|||
'list': 'torrent_providers', |
|||
'name': 'YTS', |
|||
'description': '<a href="https://yts.ag/" target="_blank">YTS</a>', |
|||
'wizard': True, |
|||
'icon': 'iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAIAAACQkWg2AAACL0lEQVR4AS1SPW/UQBAd23fxne/Ld2dvzvHuzPocEBAKokCBqG' |
|||
'iQ6IgACYmvUKRBFEQgKKGg4BAlUoggggYUEQpSHOI7CIEoQs/fYcbLaU/efTvvvZlnA1qydoxU5kcxX0CkgmQZtPy0hCUjvK+W' |
|||
'gEByOZ5dns1O5bzna8fRVkgsxH8B0YouIvBhdD5T11NiVOoKrsttyUcpRW0InUrFnwe9HzuP2uaQZYhF2LQ76TTXw2RVMTK8mY' |
|||
'Ybjfh+zNquMVCrqn93aArLSixPxnafdGDLaz1tjY5rmNa8z5BczEQOxQfCl1GyoqoWxYRN1bkh7ELw3q/vhP6HIL4TG9Kumpjg' |
|||
'vwuyM7OsjSj98E/vszMfZ7xvPtMaWxGO5crwIumKCR5HxDtJ0AWKGG204RfUd/3smJYqwem/Q7BTS1ZGfM4LNpVwuKAz6cMeRO' |
|||
'st0S2EwNE7GjTehO2H3dxqIpdkydat15G3F8SXBi4GlpBNlSz012L/k2+W0CLLk/jbcf13rf41yJeMQ8QWUZiHCfCA9ad+81nE' |
|||
'KPtoS9mJOf9v0NmMJHgUT6xayheK9EIK7JJeU/AF4scDF7Y5SPlJrRcxJ+um4ibNEdObxLiIwJim+eT2AL5D9CIcnZ5zvSJi9e' |
|||
'IlNHVVtZ831dk5svPgvjPWTq+ktWkd/kD0qtm71x+sDQe3kt6DXnM7Ct+GajmTxKlkAokWljyAKSm5oWa2w+BH4P2UuVub7eTy' |
|||
'iGOQYapY/wEztHduSDYz5gAAAABJRU5ErkJggg==', |
|||
|
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'type': 'enabler', |
|||
'default': False |
|||
}, |
|||
{ |
|||
'name': 'seed_ratio', |
|||
'label': 'Seed ratio', |
|||
'type': 'float', |
|||
'default': 1, |
|||
'description': 'Will not be (re)moved until this seed ratio is met.', |
|||
}, |
|||
{ |
|||
'name': 'seed_time', |
|||
'label': 'Seed time', |
|||
'type': 'int', |
|||
'default': 40, |
|||
'description': 'Will not be (re)moved until this seed time (in hours) is met.', |
|||
}, |
|||
{ |
|||
'name': 'info', |
|||
'label': 'Info', |
|||
'type':'bool', |
|||
'default':'False', |
|||
'description': 'YTS will only work if you set the minimum size for 720p to 500 and 1080p to 800', |
|||
}, |
|||
{ |
|||
'name': 'extra_score', |
|||
'advanced': True, |
|||
'label': 'Extra Score', |
|||
'type': 'int', |
|||
'default': 0, |
|||
'description': 'Starting score for each release found via this provider.', |
|||
} |
|||
], |
|||
} |
|||
] |
|||
}] |
|||
@ -0,0 +1,10 @@ |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.media._base.providers.torrent.yts import Base |
|||
from couchpotato.core.media.movie.providers.base import MovieProvider |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
autoload = 'Yts' |
|||
|
|||
class Yts(MovieProvider, Base): |
|||
pass |
|||
@ -0,0 +1,93 @@ |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.notifications.base import Notification |
|||
import json |
|||
import requests |
|||
|
|||
log = CPLog(__name__) |
|||
autoload = 'Discord' |
|||
|
|||
|
|||
class Discord(Notification): |
|||
required_confs = ('webhook_url',) |
|||
|
|||
def notify(self, message='', data=None, listener=None): |
|||
for key in self.required_confs: |
|||
if not self.conf(key): |
|||
log.warning('Discord notifications are enabled, but ' |
|||
'"{0}" is not specified.'.format(key)) |
|||
return False |
|||
|
|||
data = data or {} |
|||
message = message.strip() |
|||
|
|||
if self.conf('include_imdb') and 'identifier' in data: |
|||
template = ' http://www.imdb.com/title/{0[identifier]}/' |
|||
message += template.format(data) |
|||
|
|||
headers = {b"Content-Type": b"application/json"} |
|||
try: |
|||
r = requests.post(self.conf('webhook_url'), data=json.dumps(dict(content=message, username=self.conf('bot_name'), avatar_url=self.conf('avatar_url'), tts=self.conf('discord_tts'))), headers=headers) |
|||
r.status_code |
|||
except Exception as e: |
|||
log.warning('Error Sending Discord response error code: {0}'.format(r.status_code)) |
|||
return False |
|||
return True |
|||
|
|||
|
|||
config = [{ |
|||
'name': 'discord', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'notifications', |
|||
'list': 'notification_providers', |
|||
'name': 'discord', |
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'default': 0, |
|||
'type': 'enabler', |
|||
}, |
|||
{ |
|||
'name': 'webhook_url', |
|||
'description': ( |
|||
'Your Discord authentication webhook URL.', |
|||
'Created under channel settings.' |
|||
) |
|||
}, |
|||
{ |
|||
'name': 'include_imdb', |
|||
'default': True, |
|||
'type': 'bool', |
|||
'descrpition': 'Include a link to the movie page on IMDB.' |
|||
}, |
|||
{ |
|||
'name': 'bot_name', |
|||
'description': 'Name of bot.', |
|||
'default': 'CouchPotato', |
|||
'advanced': True, |
|||
}, |
|||
{ |
|||
'name': 'avatar_url', |
|||
'description': 'URL to an image to use as the avatar for ' |
|||
'notifications.', |
|||
'default': 'https://couchpota.to/media/images/couch.png', |
|||
'advanced': True, |
|||
}, |
|||
{ |
|||
'name': 'discord_tts', |
|||
'default': 0, |
|||
'type': 'bool', |
|||
'advanced': True, |
|||
'description': 'Send notification using text-to-speech.', |
|||
}, |
|||
{ |
|||
'name': 'on_snatch', |
|||
'default': 0, |
|||
'type': 'bool', |
|||
'advanced': True, |
|||
'description': 'Also send message when movie is snatched.', |
|||
}, |
|||
], |
|||
} |
|||
], |
|||
}] |
|||
@ -0,0 +1,58 @@ |
|||
import traceback |
|||
import subprocess |
|||
|
|||
from couchpotato.core.helpers.encoding import toUnicode |
|||
from couchpotato.core.helpers.variable import getIdentifier |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.notifications.base import Notification |
|||
|
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
autoload = 'Script' |
|||
|
|||
class Script(Notification): |
|||
|
|||
def notify(self, message = '', data = None, listener = None): |
|||
if not data: data = {} |
|||
|
|||
script_data = { |
|||
'message': toUnicode(message) |
|||
} |
|||
|
|||
if getIdentifier(data): |
|||
script_data.update({ |
|||
'imdb_id': getIdentifier(data) |
|||
}) |
|||
|
|||
try: |
|||
subprocess.call([self.conf('path'), message]) |
|||
return True |
|||
except: |
|||
log.error('Script notification failed: %s', traceback.format_exc()) |
|||
|
|||
return False |
|||
|
|||
|
|||
config = [{ |
|||
'name': 'script', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'notifications', |
|||
'list': 'notification_providers', |
|||
'name': 'script', |
|||
'label': 'Script', |
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'default': 0, |
|||
'type': 'enabler', |
|||
}, |
|||
{ |
|||
'name': 'path', |
|||
'description': 'The path to the script to execute.' |
|||
} |
|||
] |
|||
} |
|||
] |
|||
}] |
|||
Loading…
Reference in new issue