Ruud
10 years ago
8 changed files with 383 additions and 5 deletions
@ -0,0 +1,232 @@ |
|||
import re |
|||
import traceback |
|||
import random |
|||
import time |
|||
from datetime import datetime |
|||
|
|||
from couchpotato.core.helpers.variable import tryInt, getIdentifier |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.media._base.providers.torrent.base import TorrentMagnetProvider |
|||
|
|||
log = CPLog(__name__) |
|||
tokenreceived = 0 |
|||
rarbgtoken = 0 |
|||
|
|||
class Base(TorrentMagnetProvider): |
|||
|
|||
urls = { |
|||
'test': 'https://torrentapi.org/pubapi_v2.php?app_id=couchpotato', |
|||
'token': 'https://torrentapi.org/pubapi_v2.php?get_token=get_token&app_id=couchpotato', |
|||
'search': 'https://torrentapi.org/pubapi_v2.php?token=%s&mode=search&search_imdb=%s&min_seeders=%s&min_leechers' |
|||
'=%s&ranked=%s&category=movies&format=json_extended&app_id=couchpotato', |
|||
} |
|||
|
|||
http_time_between_calls = 2 # Seconds |
|||
user_agent = 'CouchPotato/1.0' |
|||
|
|||
@staticmethod |
|||
def find_info(filename): |
|||
# CODEC # |
|||
codec = 'x264' |
|||
v = re.search("(?i)(x265|h265|h\.265)", filename) |
|||
if v: |
|||
codec = 'x265' |
|||
|
|||
v = re.search("(?i)(xvid)", filename) |
|||
if v: |
|||
codec = 'xvid' |
|||
|
|||
# RESOLUTION # |
|||
resolution = 'SD' |
|||
a = re.search("(?i)(720p)", filename) |
|||
if a: |
|||
resolution = '720p' |
|||
|
|||
a = re.search("(?i)(1080p)", filename) |
|||
if a: |
|||
resolution = '1080p' |
|||
|
|||
a = re.search("(?i)(2160p)", filename) |
|||
if a: |
|||
resolution = '2160p' |
|||
|
|||
# SOURCE # |
|||
source = 'HD-Rip' |
|||
s = re.search("(?i)(WEB-DL|WEB_DL|WEB\.DL)", filename) |
|||
if s: |
|||
source = 'WEB-DL' |
|||
|
|||
s = re.search("(?i)(WEBRIP)", filename) |
|||
if s: |
|||
source = 'WEBRIP' |
|||
|
|||
s = re.search("(?i)(DVDR|DVDRip|DVD-Rip)", filename) |
|||
if s: |
|||
source = 'DVD-R' |
|||
|
|||
s = re.search("(?i)(BRRIP|BDRIP|BluRay)", filename) |
|||
if s: |
|||
source = 'BR-Rip' |
|||
|
|||
s = re.search("(?i)BluRay(.*)REMUX", filename) |
|||
if s: |
|||
source = 'BluRay-Remux' |
|||
|
|||
s = re.search("(?i)BluRay(.*)\.(AVC|VC-1)\.", filename) |
|||
if s: |
|||
source = 'BluRay-Full' |
|||
|
|||
return_info = [codec, resolution, source] |
|||
return return_info |
|||
|
|||
@staticmethod |
|||
def get_token(self): |
|||
global tokenreceived |
|||
global rarbgtoken |
|||
now = int(time.time()) |
|||
|
|||
if (rarbgtoken == 0) or (tokenreceived == 0) or (now > (tokenreceived+(15*60))): |
|||
log.debug("RARBG: Getting Rarbg token") |
|||
tokendata = self.getJsonData(self.urls['token']) |
|||
|
|||
if tokendata: |
|||
try: |
|||
tokenreceived = int(time.time()) |
|||
rarbgtoken = tokendata['token'] |
|||
log.debug("RARBG: GOT TOKEN: %s", rarbgtoken) |
|||
except RuntimeError: |
|||
log.error('RARBG: Failed getting token from Rarbg') |
|||
rarbgtoken = 0 |
|||
|
|||
# return token |
|||
|
|||
def _search(self, movie, quality, results): |
|||
hasresults = 0 |
|||
curryear = datetime.now().year |
|||
self.get_token(self) |
|||
movieid = getIdentifier(movie) |
|||
try: |
|||
movieyear = movie['info']['year'] |
|||
except: |
|||
log.error("RARBG: Couldn't get movie year") |
|||
movieyear = 0 |
|||
|
|||
if (rarbgtoken != 0) and (movieyear == 0 or movieyear <= curryear): |
|||
data = self.getJsonData(self.urls['search'] % (rarbgtoken, movieid, self.conf('min_seeders'), |
|||
self.conf('min_leechers'), self.conf('ranked_only'))) |
|||
|
|||
if data: |
|||
if 'error_code' in data: |
|||
if data['error'] == 'No results found': |
|||
log.debug("RARBG: No results returned from Rarbg") |
|||
else: |
|||
if data['error_code'] == 10: |
|||
log.error(data['error'], movieid) |
|||
else: |
|||
log.error("RARBG: There is an error in the returned JSON: %s", data['error']) |
|||
else: |
|||
hasresults = 1 |
|||
try: |
|||
if hasresults: |
|||
for result in data['torrent_results']: |
|||
name = result['title'] |
|||
titlesplit = re.split("-", name) |
|||
releasegroup = titlesplit[len(titlesplit)-1] |
|||
|
|||
xtrainfo = self.find_info(name) |
|||
encoding = xtrainfo[0] |
|||
resolution = xtrainfo[1] |
|||
# source = xtrainfo[2] |
|||
pubdate = result['pubdate'] # .strip(" +0000") |
|||
try: |
|||
pubdate = datetime.strptime(pubdate, '%Y-%m-%d %H:%M:%S +0000') |
|||
now = datetime.utcnow() |
|||
age = (now - pubdate).days |
|||
except ValueError: |
|||
log.debug("RARBG: Bad pubdate") |
|||
age = 0 |
|||
|
|||
torrentscore = self.conf('extra_score') |
|||
seeders = tryInt(result['seeders']) |
|||
torrent_desc = '/ %s / %s / %s / %s seeders' % (releasegroup, resolution, encoding, seeders) |
|||
|
|||
if seeders == 0: |
|||
torrentscore = 0 |
|||
|
|||
sliceyear = result['pubdate'][0:4] |
|||
year = tryInt(sliceyear) |
|||
|
|||
results.append({ |
|||
'id': random.randint(100, 9999), |
|||
'name': re.sub('[^A-Za-z0-9\-_ \(\).]+', '', '%s (%s) %s' % (name, year, torrent_desc)), |
|||
'url': result['download'], |
|||
'detail_url': result['info_page'], |
|||
'size': tryInt(result['size']/1048576), # rarbg sends in bytes |
|||
'seeders': tryInt(result['seeders']), |
|||
'leechers': tryInt(result['leechers']), |
|||
'age': tryInt(age), |
|||
'score': torrentscore |
|||
}) |
|||
|
|||
except RuntimeError: |
|||
log.error('RARBG: Failed getting results from %s: %s', (self.getName(), traceback.format_exc())) |
|||
config = [{ |
|||
'name': 'rarbg', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'searcher', |
|||
'list': 'torrent_providers', |
|||
'name': 'Rarbg', |
|||
'wizard': True, |
|||
'description': '<a href="https://rarbg.to/torrents.php">RARBG</a>', |
|||
'icon': 'iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAB+UlEQVQ4jYXTP2hcRxDH8c8JJZjbYNy8V7gIr0qhg5AiFnETX' |
|||
'+PmVAtSmKDaUhUiFyGxjXFlp0hhHy5cqFd9lSGcU55cBU6EEMIj5dsmMewSjNGmOJ3852wysMyww37n94OdXimlh49xDR/hxGr' |
|||
'8hZ/xx0qnlHK5lPKk/H/8U0r5oZTyQSmltzzr+AKfT+ed8UFLeHNAH1UVbA2r88NBfQcX8O2yv74sUqKNWT+T01sy2+zpUbS/w' |
|||
'/awvo7H+O0NQEA/LPKlQWXrSgUmR9HxcZQwmbZGw/pc4MsVAIT+IjcNw80aTjaaem1vPCNlGakj1C6uWFiqeDtyTvoyqAKhBn+' |
|||
'+E7CkxC6Zzjop57XpUSenpIuMhpXAc/zyHkAicRSjw6fHZ1ewPdqwszWAB2hXACln8+NWSlld9zX9YN7GhajQXz5+joPXR66de' |
|||
'U1J27Zi7FzaqE0OdmwNGzF2Ymzt3j+E8/gJH64AFlozKS4+Be7tjwyaIKVsOpnavX0II9x8ByDLKco5SwvjL0MI/z64tyOcwsf' |
|||
'jQw8PJvAdvsb6GSBlxI7UyTnD37i7OWhe3NrflvOit3djbDKdwR181SulXMXdrkubbdvKaOpK09S/4jP8iG9m8zmJjCoEg0HzO' |
|||
'77vna7zp7ju1TqfYIyZxT7dwCd4eWr7BR7h2X8S6gShJlbKYQAAAABJRU5ErkJggg==', |
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'type': 'enabler', |
|||
'default': False, |
|||
}, |
|||
{ |
|||
'name': 'ranked_only', |
|||
'advanced': True, |
|||
'label': 'Ranked Only', |
|||
'type': 'int', |
|||
'default': 1, |
|||
'description': 'Only ranked torrents (internal), scene releases, rarbg releases. ' |
|||
'Enter 1 (true) or 0 (false)', |
|||
}, |
|||
{ |
|||
'name': 'min_seeders', |
|||
'advanced': True, |
|||
'label': 'Minimum Seeders', |
|||
'type': 'int', |
|||
'default': 10, |
|||
'description': 'Minium amount of seeders the release must have.', |
|||
}, |
|||
{ |
|||
'name': 'min_leechers', |
|||
'advanced': True, |
|||
'label': 'Minimum leechers', |
|||
'type': 'int', |
|||
'default': 0, |
|||
'description': 'Minium amount of leechers the release must have.', |
|||
}, |
|||
{ |
|||
'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,11 @@ |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.media._base.providers.torrent.rarbg import Base |
|||
from couchpotato.core.media.movie.providers.base import MovieProvider |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
autoload = 'Rarbg' |
|||
|
|||
|
|||
class Rarbg(MovieProvider, Base): |
|||
pass |
|||
@ -0,0 +1,126 @@ |
|||
import json |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.notifications.base import Notification |
|||
|
|||
log = CPLog(__name__) |
|||
autoload = 'Slack' |
|||
|
|||
|
|||
class Slack(Notification): |
|||
url = 'https://slack.com/api/chat.postMessage' |
|||
required_confs = ('token', 'channels',) |
|||
|
|||
def notify(self, message='', data=None, listener=None): |
|||
for key in self.required_confs: |
|||
if not self.conf(key): |
|||
log.warning('Slack 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) |
|||
|
|||
payload = { |
|||
'token': self.conf('token'), |
|||
'text': message, |
|||
'username': self.conf('bot_name'), |
|||
'unfurl_links': self.conf('include_imdb'), |
|||
'as_user': self.conf('as_user'), |
|||
'icon_url': self.conf('icon_url'), |
|||
'icon_emoji': self.conf('icon_emoji') |
|||
} |
|||
|
|||
channels = self.conf('channels').split(',') |
|||
for channel in channels: |
|||
payload['channel'] = channel.strip() |
|||
response = self.urlopen(self.url, data=payload) |
|||
response = json.loads(response) |
|||
if not response['ok']: |
|||
log.warning('Notification sending to Slack has failed. Error ' |
|||
'code: %s.', response['error']) |
|||
return False |
|||
return True |
|||
|
|||
|
|||
config = [{ |
|||
'name': 'slack', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'notifications', |
|||
'list': 'notification_providers', |
|||
'name': 'slack', |
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'default': 0, |
|||
'type': 'enabler', |
|||
}, |
|||
{ |
|||
'name': 'token', |
|||
'description': ( |
|||
'Your Slack authentication token.', |
|||
'Can be created at https://api.slack.com/web' |
|||
) |
|||
}, |
|||
{ |
|||
'name': 'channels', |
|||
'description': ( |
|||
'Channel to send notifications to.', |
|||
'Can be a public channel, private group or IM ' |
|||
'channel. Can be an encoded ID or a name ' |
|||
'(staring with a hashtag, e.g. #general). ' |
|||
'Separate with commas in order to notify multiple ' |
|||
'channels. It is however recommended to send ' |
|||
'notifications to only one channel due to ' |
|||
'the Slack API rate limits.' |
|||
) |
|||
}, |
|||
{ |
|||
'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': 'as_user', |
|||
'description': 'Send message as the authentication token ' |
|||
' user.', |
|||
'default': False, |
|||
'type': 'bool', |
|||
'advanced': True |
|||
}, |
|||
{ |
|||
'name': 'icon_url', |
|||
'description': 'URL to an image to use as the icon for ' |
|||
'notifications.', |
|||
'advanced': True, |
|||
}, |
|||
{ |
|||
'name': 'icon_emoji', |
|||
'description': ( |
|||
'Emoji to use as the icon for notifications.', |
|||
'Overrides icon_url' |
|||
), |
|||
'advanced': True, |
|||
}, |
|||
{ |
|||
'name': 'on_snatch', |
|||
'default': 0, |
|||
'type': 'bool', |
|||
'advanced': True, |
|||
'description': 'Also send message when movie is snatched.', |
|||
}, |
|||
], |
|||
} |
|||
], |
|||
}] |
|||
Loading…
Reference in new issue