Ruud
13 years ago
55 changed files with 1084 additions and 549 deletions
@ -0,0 +1,46 @@ |
|||
from .main import NZBVortex |
|||
|
|||
def start(): |
|||
return NZBVortex() |
|||
|
|||
config = [{ |
|||
'name': 'nzbvortex', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'downloaders', |
|||
'name': 'nzbvortex', |
|||
'label': 'NZBVortex', |
|||
'description': 'Use <a href="http://www.nzbvortex.com/landing/" target="_blank">NZBVortex</a> to download NZBs.', |
|||
'wizard': True, |
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'default': 0, |
|||
'type': 'enabler', |
|||
'radio_group': 'nzb', |
|||
}, |
|||
{ |
|||
'name': 'host', |
|||
'default': 'https://localhost:4321', |
|||
}, |
|||
{ |
|||
'name': 'api_key', |
|||
'label': 'Api Key', |
|||
}, |
|||
{ |
|||
'name': 'manual', |
|||
'default': False, |
|||
'type': 'bool', |
|||
'advanced': True, |
|||
'description': 'Disable this downloader for automated searches, but use it when I manually send a release.', |
|||
}, |
|||
{ |
|||
'name': 'delete_failed', |
|||
'default': True, |
|||
'type': 'bool', |
|||
'description': 'Delete a release after the download has failed.', |
|||
}, |
|||
], |
|||
} |
|||
], |
|||
}] |
|||
@ -0,0 +1,176 @@ |
|||
from base64 import b64encode |
|||
from couchpotato.core.downloaders.base import Downloader |
|||
from couchpotato.core.helpers.encoding import tryUrlencode, ss |
|||
from couchpotato.core.helpers.variable import cleanHost |
|||
from couchpotato.core.logger import CPLog |
|||
from urllib2 import URLError |
|||
from uuid import uuid4 |
|||
import hashlib |
|||
import httplib |
|||
import json |
|||
import socket |
|||
import ssl |
|||
import sys |
|||
import traceback |
|||
import urllib2 |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
class NZBVortex(Downloader): |
|||
|
|||
type = ['nzb'] |
|||
api_level = None |
|||
session_id = None |
|||
|
|||
def download(self, data = {}, movie = {}, manual = False, filedata = None): |
|||
|
|||
if self.isDisabled(manual) or not self.isCorrectType(data.get('type')) or not self.getApiLevel(): |
|||
return |
|||
|
|||
# Send the nzb |
|||
try: |
|||
nzb_filename = self.createFileName(data, filedata, movie) |
|||
self.call('nzb/add', params = {'file': (ss(nzb_filename), filedata)}, multipart = True) |
|||
|
|||
return True |
|||
except: |
|||
log.error('Something went wrong sending the NZB file: %s', traceback.format_exc()) |
|||
return False |
|||
|
|||
def getAllDownloadStatus(self): |
|||
|
|||
if self.isDisabled(manual = True): |
|||
return False |
|||
|
|||
raw_statuses = self.call('nzb') |
|||
|
|||
statuses = [] |
|||
for item in raw_statuses.get('nzbs', []): |
|||
|
|||
# Check status |
|||
status = 'busy' |
|||
if item['state'] == 20: |
|||
status = 'completed' |
|||
elif item['state'] in [21, 22, 24]: |
|||
status = 'failed' |
|||
|
|||
statuses.append({ |
|||
'id': item['id'], |
|||
'name': item['uiTitle'], |
|||
'status': status, |
|||
'original_status': item['state'], |
|||
'timeleft':-1, |
|||
}) |
|||
|
|||
return statuses |
|||
|
|||
def removeFailed(self, item): |
|||
|
|||
if not self.conf('delete_failed', default = True): |
|||
return False |
|||
|
|||
log.info('%s failed downloading, deleting...', item['name']) |
|||
|
|||
try: |
|||
self.call('nzb/%s/cancel' % item['id']) |
|||
except: |
|||
log.error('Failed deleting: %s', traceback.format_exc(0)) |
|||
return False |
|||
|
|||
return True |
|||
|
|||
def login(self): |
|||
|
|||
nonce = self.call('auth/nonce', auth = False).get('authNonce') |
|||
cnonce = uuid4().hex |
|||
hashed = b64encode(hashlib.sha256('%s:%s:%s' % (nonce, cnonce, self.conf('api_key'))).digest()) |
|||
|
|||
params = { |
|||
'nonce': nonce, |
|||
'cnonce': cnonce, |
|||
'hash': hashed |
|||
} |
|||
|
|||
login_data = self.call('auth/login', parameters = params, auth = False) |
|||
|
|||
# Save for later |
|||
if login_data.get('loginResult') == 'successful': |
|||
self.session_id = login_data.get('sessionID') |
|||
return True |
|||
|
|||
log.error('Login failed, please check you api-key') |
|||
return False |
|||
|
|||
|
|||
def call(self, call, parameters = {}, repeat = False, auth = True, *args, **kwargs): |
|||
|
|||
# Login first |
|||
if not self.session_id and auth: |
|||
self.login() |
|||
|
|||
# Always add session id to request |
|||
if self.session_id: |
|||
parameters['sessionid'] = self.session_id |
|||
|
|||
params = tryUrlencode(parameters) |
|||
|
|||
url = cleanHost(self.conf('host')) + 'api/' + call |
|||
url_opener = urllib2.build_opener(HTTPSHandler()) |
|||
|
|||
try: |
|||
data = self.urlopen('%s?%s' % (url, params), opener = url_opener, *args, **kwargs) |
|||
|
|||
if data: |
|||
return json.loads(data) |
|||
except URLError, e: |
|||
if hasattr(e, 'code') and e.code == 403: |
|||
# Try login and do again |
|||
if not repeat: |
|||
self.login() |
|||
return self.call(call, parameters = parameters, repeat = True, *args, **kwargs) |
|||
|
|||
log.error('Failed to parsing %s: %s', (self.getName(), traceback.format_exc())) |
|||
except: |
|||
log.error('Failed to parsing %s: %s', (self.getName(), traceback.format_exc())) |
|||
|
|||
return {} |
|||
|
|||
def getApiLevel(self): |
|||
|
|||
if not self.api_level: |
|||
|
|||
url = cleanHost(self.conf('host')) + 'api/app/apilevel' |
|||
url_opener = urllib2.build_opener(HTTPSHandler()) |
|||
|
|||
try: |
|||
data = self.urlopen(url, opener = url_opener, show_error = False) |
|||
self.api_level = float(json.loads(data).get('apilevel')) |
|||
except URLError, e: |
|||
if hasattr(e, 'code') and e.code == 403: |
|||
log.error('This version of NZBVortex isn\'t supported. Please update to 2.8.6 or higher') |
|||
else: |
|||
log.error('NZBVortex doesn\'t seem to be running or maybe the remote option isn\'t enabled yet: %s', traceback.format_exc(1)) |
|||
|
|||
return self.api_level |
|||
|
|||
|
|||
class HTTPSConnection(httplib.HTTPSConnection): |
|||
def __init__(self, *args, **kwargs): |
|||
httplib.HTTPSConnection.__init__(self, *args, **kwargs) |
|||
|
|||
def connect(self): |
|||
sock = socket.create_connection((self.host, self.port), self.timeout) |
|||
if sys.version_info < (2, 6, 7): |
|||
if hasattr(self, '_tunnel_host'): |
|||
self.sock = sock |
|||
self._tunnel() |
|||
else: |
|||
if self._tunnel_host: |
|||
self.sock = sock |
|||
self._tunnel() |
|||
|
|||
self.sock = ssl.wrap_socket(sock, self.key_file, self.cert_file, ssl_version = ssl.PROTOCOL_TLSv1) |
|||
|
|||
class HTTPSHandler(urllib2.HTTPSHandler): |
|||
def https_open(self, req): |
|||
return self.do_open(HTTPSConnection, req) |
|||
@ -0,0 +1,32 @@ |
|||
from .main import Toasty |
|||
|
|||
def start(): |
|||
return Toasty() |
|||
|
|||
config = [{ |
|||
'name': 'toasty', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'notifications', |
|||
'name': 'toasty', |
|||
'options': [ |
|||
{ |
|||
'name': 'enabled', |
|||
'default': 0, |
|||
'type': 'enabler', |
|||
}, |
|||
{ |
|||
'name': 'api_key', |
|||
'label': 'Device ID', |
|||
}, |
|||
{ |
|||
'name': 'on_snatch', |
|||
'default': 0, |
|||
'type': 'bool', |
|||
'advanced': True, |
|||
'description': 'Also send message when movie is snatched.', |
|||
}, |
|||
], |
|||
} |
|||
], |
|||
}] |
|||
@ -0,0 +1,30 @@ |
|||
from couchpotato.core.helpers.encoding import toUnicode, tryUrlencode |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.notifications.base import Notification |
|||
import traceback |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
class Toasty(Notification): |
|||
|
|||
urls = { |
|||
'api': 'http://api.supertoasty.com/notify/%s?%s' |
|||
} |
|||
|
|||
def notify(self, message = '', data = {}, listener = None): |
|||
if self.isDisabled(): return |
|||
|
|||
data = { |
|||
'title': self.default_title, |
|||
'text': toUnicode(message), |
|||
'sender': toUnicode("CouchPotato"), |
|||
'image': 'https://raw.github.com/RuudBurger/CouchPotatoServer/master/couchpotato/static/images/homescreen.png', |
|||
} |
|||
|
|||
try: |
|||
self.urlopen(self.urls['api'] % (self.conf('api_key'), tryUrlencode(data)), show_error = False) |
|||
return True |
|||
except: |
|||
log.error('Toasty failed: %s', traceback.format_exc()) |
|||
|
|||
return False |
|||
@ -0,0 +1,35 @@ |
|||
from .main import ITunes |
|||
|
|||
def start(): |
|||
return ITunes() |
|||
|
|||
config = [{ |
|||
'name': 'itunes', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'automation', |
|||
'name': 'itunes_automation', |
|||
'label': 'iTunes', |
|||
'description': 'From any <a href="http://itunes.apple.com/rss">iTunes</a> Store feed. Url should be the RSS link. (uses minimal requirements)', |
|||
'options': [ |
|||
{ |
|||
'name': 'automation_enabled', |
|||
'default': False, |
|||
'type': 'enabler', |
|||
}, |
|||
{ |
|||
'name': 'automation_urls_use', |
|||
'label': 'Use', |
|||
'default': ',', |
|||
}, |
|||
{ |
|||
'name': 'automation_urls', |
|||
'label': 'url', |
|||
'type': 'combined', |
|||
'combine': ['automation_urls_use', 'automation_urls'], |
|||
'default': 'https://itunes.apple.com/rss/topmovies/limit=25/xml,', |
|||
}, |
|||
], |
|||
}, |
|||
], |
|||
}] |
|||
@ -0,0 +1,63 @@ |
|||
from couchpotato.core.helpers.rss import RSS |
|||
from couchpotato.core.helpers.variable import md5, splitString, tryInt |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.providers.automation.base import Automation |
|||
from xml.etree.ElementTree import QName |
|||
import datetime |
|||
import traceback |
|||
import xml.etree.ElementTree as XMLTree |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
|
|||
class ITunes(Automation, RSS): |
|||
|
|||
interval = 1800 |
|||
|
|||
def getIMDBids(self): |
|||
|
|||
if self.isDisabled(): |
|||
return |
|||
|
|||
movies = [] |
|||
|
|||
enablers = [tryInt(x) for x in splitString(self.conf('automation_urls_use'))] |
|||
urls = splitString(self.conf('automation_urls')) |
|||
|
|||
namespace = 'http://www.w3.org/2005/Atom' |
|||
namespaceIM = 'http://itunes.apple.com/rss' |
|||
|
|||
index = -1 |
|||
for url in urls: |
|||
|
|||
index += 1 |
|||
if not enablers[index]: |
|||
continue |
|||
|
|||
try: |
|||
cache_key = 'itunes.rss.%s' % md5(url) |
|||
rss_data = self.getCache(cache_key, url) |
|||
|
|||
data = XMLTree.fromstring(rss_data) |
|||
|
|||
if data is not None: |
|||
entry_tag = str(QName(namespace, 'entry')) |
|||
rss_movies = self.getElements(data, entry_tag) |
|||
|
|||
for movie in rss_movies: |
|||
name_tag = str(QName(namespaceIM, 'name')) |
|||
name = self.getTextElement(movie, name_tag) |
|||
|
|||
releaseDate_tag = str(QName(namespaceIM, 'releaseDate')) |
|||
releaseDateText = self.getTextElement(movie, releaseDate_tag) |
|||
year = datetime.datetime.strptime(releaseDateText, '%Y-%m-%dT00:00:00-07:00').strftime("%Y") |
|||
|
|||
imdb = self.search(name, year) |
|||
|
|||
if imdb and self.isMinimalMovie(imdb): |
|||
movies.append(imdb['imdb']) |
|||
|
|||
except: |
|||
log.error('Failed loading iTunes rss feed: %s %s', (url, traceback.format_exc())) |
|||
|
|||
return movies |
|||
@ -0,0 +1,23 @@ |
|||
from .main import Moviemeter |
|||
|
|||
def start(): |
|||
return Moviemeter() |
|||
|
|||
config = [{ |
|||
'name': 'moviemeter', |
|||
'groups': [ |
|||
{ |
|||
'tab': 'automation', |
|||
'name': 'moviemeter_automation', |
|||
'label': 'Moviemeter', |
|||
'description': 'Imports movies from the current top 10 of moviemeter.nl. (uses minimal requirements)', |
|||
'options': [ |
|||
{ |
|||
'name': 'automation_enabled', |
|||
'default': False, |
|||
'type': 'enabler', |
|||
}, |
|||
], |
|||
}, |
|||
], |
|||
}] |
|||
@ -0,0 +1,28 @@ |
|||
from couchpotato.core.event import fireEvent |
|||
from couchpotato.core.helpers.rss import RSS |
|||
from couchpotato.core.logger import CPLog |
|||
from couchpotato.core.providers.automation.base import Automation |
|||
|
|||
log = CPLog(__name__) |
|||
|
|||
|
|||
class Moviemeter(Automation, RSS): |
|||
|
|||
interval = 1800 |
|||
rss_url = 'http://www.moviemeter.nl/rss/cinema' |
|||
|
|||
def getIMDBids(self): |
|||
|
|||
movies = [] |
|||
|
|||
rss_movies = self.getRSSData(self.rss_url) |
|||
|
|||
for movie in rss_movies: |
|||
|
|||
name_year = fireEvent('scanner.name_year', self.getTextElement(movie, 'title'), single = True) |
|||
imdb = self.search(name_year.get('name'), name_year.get('year')) |
|||
|
|||
if imdb and self.isMinimalMovie(imdb): |
|||
movies.append(imdb['imdb']) |
|||
|
|||
return movies |
|||
@ -1,6 +0,0 @@ |
|||
from .main import IMDBAPI |
|||
|
|||
def start(): |
|||
return IMDBAPI() |
|||
|
|||
config = [] |
|||
@ -0,0 +1,6 @@ |
|||
from .main import OMDBAPI |
|||
|
|||
def start(): |
|||
return OMDBAPI() |
|||
|
|||
config = [] |
|||
Loading…
Reference in new issue