Ruud
14 years ago
25 changed files with 1149 additions and 181 deletions
@ -0,0 +1,6 @@ |
|||||
|
from .main import CoreNotifier |
||||
|
|
||||
|
def start(): |
||||
|
return CoreNotifier() |
||||
|
|
||||
|
config = [] |
||||
@ -0,0 +1,49 @@ |
|||||
|
from couchpotato.api import addApiView |
||||
|
from couchpotato.core.event import addEvent, fireEvent |
||||
|
from couchpotato.core.helpers.request import jsonified |
||||
|
from couchpotato.core.logger import CPLog |
||||
|
from couchpotato.core.plugins.base import Plugin |
||||
|
import time |
||||
|
|
||||
|
log = CPLog(__name__) |
||||
|
|
||||
|
|
||||
|
class CoreNotifier(Plugin): |
||||
|
|
||||
|
messages = [] |
||||
|
|
||||
|
def __init__(self): |
||||
|
addEvent('notify', self.notify) |
||||
|
addEvent('notify.core_notifier', self.notify) |
||||
|
addEvent('core_notifier.frontend', self.frontend) |
||||
|
|
||||
|
addApiView('core_notifier.listener', self.listener) |
||||
|
|
||||
|
static = self.registerStatic(__file__) |
||||
|
fireEvent('register_script', static + 'notification.js') |
||||
|
|
||||
|
def notify(self, message = '', data = {}): |
||||
|
self.add(data = { |
||||
|
'message': message, |
||||
|
'raw': data, |
||||
|
}) |
||||
|
|
||||
|
def frontend(self, type = 'notification', data = {}): |
||||
|
|
||||
|
self.messages.append({ |
||||
|
'time': time.time(), |
||||
|
'type': type, |
||||
|
'data': data, |
||||
|
}) |
||||
|
|
||||
|
def listener(self): |
||||
|
|
||||
|
for message in self.messages: |
||||
|
#delete message older then 15s |
||||
|
if message['time'] < (time.time() - 15): |
||||
|
del message |
||||
|
|
||||
|
return jsonified({ |
||||
|
'success': True, |
||||
|
'result': self.messages, |
||||
|
}) |
||||
@ -0,0 +1,42 @@ |
|||||
|
var NotificationBase = new Class({ |
||||
|
|
||||
|
Extends: BlockBase, |
||||
|
Implements: [Options, Events], |
||||
|
|
||||
|
initialize: function(options){ |
||||
|
var self = this; |
||||
|
self.setOptions(options); |
||||
|
|
||||
|
//App.addEvent('load', self.request.bind(self));
|
||||
|
|
||||
|
self.addEvent('notification', self.notify.bind(self)) |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
request: function(){ |
||||
|
var self = this; |
||||
|
|
||||
|
Api.request('core_notifier.listener', { |
||||
|
'initialDelay': 100, |
||||
|
'delay': 3000, |
||||
|
'onComplete': self.processData.bind(self) |
||||
|
}).startTimer() |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
notify: function(data){ |
||||
|
var self = this; |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
processData: function(json){ |
||||
|
var self = this; |
||||
|
|
||||
|
Array.each(json.result, function(result){ |
||||
|
self.fireEvent(result.type, result.data) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
}); |
||||
|
|
||||
|
window.Notification = new NotificationBase(); |
||||
@ -0,0 +1,22 @@ |
|||||
|
from .main import Wizard |
||||
|
|
||||
|
def start(): |
||||
|
return Wizard() |
||||
|
|
||||
|
config = [{ |
||||
|
'name': 'global', |
||||
|
'groups': [ |
||||
|
{ |
||||
|
'tab': 'general', |
||||
|
'name': 'advanced', |
||||
|
'options': [ |
||||
|
{ |
||||
|
'name': 'show_wizard', |
||||
|
'label': 'Run the wizard', |
||||
|
'default': True, |
||||
|
'type': 'bool', |
||||
|
}, |
||||
|
], |
||||
|
}, |
||||
|
], |
||||
|
}] |
||||
@ -0,0 +1,13 @@ |
|||||
|
from couchpotato.core.event import fireEvent |
||||
|
from couchpotato.core.logger import CPLog |
||||
|
from couchpotato.core.plugins.base import Plugin |
||||
|
|
||||
|
log = CPLog(__name__) |
||||
|
|
||||
|
|
||||
|
class Wizard(Plugin): |
||||
|
|
||||
|
def __init__(self): |
||||
|
path = self.registerStatic(__file__) |
||||
|
fireEvent('register_script', path + 'spotlight.js') |
||||
|
fireEvent('register_script', path + 'wizard.js') |
||||
@ -0,0 +1,306 @@ |
|||||
|
/* |
||||
|
--- |
||||
|
description: Fill the empty space around elements, creating a spotlight effect. |
||||
|
|
||||
|
license: GPL v3.0 |
||||
|
|
||||
|
authors: |
||||
|
- Ruud Burger |
||||
|
|
||||
|
requires: |
||||
|
- core/1.3: [Class.Extras, Element.Dimensions] |
||||
|
|
||||
|
provides: [Spotlight] |
||||
|
|
||||
|
... |
||||
|
*/ |
||||
|
|
||||
|
var Spotlight = new Class({ |
||||
|
|
||||
|
Implements: [Options], |
||||
|
|
||||
|
options: { |
||||
|
'fillClass': 'spotlight_fill', |
||||
|
'fillColor': [255,255,255], |
||||
|
'fillOpacity': 1, |
||||
|
'parent': null, |
||||
|
'inject': null, |
||||
|
'soften': 10 |
||||
|
}, |
||||
|
|
||||
|
initialize: function(elements, options){ |
||||
|
var self = this; |
||||
|
self.setOptions(options); |
||||
|
|
||||
|
self.setElements(elements); |
||||
|
self.clean(); |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
clean: function(){ |
||||
|
var self = this; |
||||
|
|
||||
|
self.range = []; self.fills = []; self.edges = []; |
||||
|
|
||||
|
self.vert = []; |
||||
|
self.vert_el = []; |
||||
|
|
||||
|
self.top = []; self.left = []; |
||||
|
self.width = []; self.height = []; |
||||
|
}, |
||||
|
|
||||
|
setElements: function(elements){ |
||||
|
this.elements = elements; |
||||
|
}, |
||||
|
|
||||
|
addElement: function(element){ |
||||
|
this.elements.include(element); |
||||
|
}, |
||||
|
|
||||
|
create: function(){ |
||||
|
var self = this; |
||||
|
|
||||
|
self.destroy(); |
||||
|
|
||||
|
var page_c = $(self.options.parent || window).getScrollSize(); |
||||
|
var soften = self.options.soften; |
||||
|
|
||||
|
// Get the top and bottom of all the elements
|
||||
|
self.elements.each(function(el, nr){ |
||||
|
var c = el.getCoordinates(); |
||||
|
|
||||
|
if(c.top > 0 && nr == 0){ |
||||
|
self.vert.append([0]); |
||||
|
self.vert_el.append([null]); |
||||
|
} |
||||
|
|
||||
|
// Top
|
||||
|
self.vert.append([c.top-soften]); |
||||
|
self.vert_el.append([el]); |
||||
|
|
||||
|
// Bottom
|
||||
|
self.vert.append([c.top+c.height+soften]); |
||||
|
self.vert_el.append([el]); |
||||
|
|
||||
|
// Add it to range, for later calculation from left to right
|
||||
|
self.range.append([{ |
||||
|
'el': el, |
||||
|
'top': c.top-soften, |
||||
|
'bottom': c.top+c.height+soften, |
||||
|
'left': c.left-soften, |
||||
|
'right': c.left+c.width+soften |
||||
|
}]) |
||||
|
|
||||
|
// Create soft edge around element
|
||||
|
self.soften(el); |
||||
|
|
||||
|
}); |
||||
|
|
||||
|
if(self.elements.length == 0){ |
||||
|
self.vert.append([0]); |
||||
|
self.vert_el.append([null]); |
||||
|
} |
||||
|
|
||||
|
// Reorder
|
||||
|
var vert = self.vert.clone().sort(self.numberSort) // Use custom sort function because apparently 100 is less then 20..
|
||||
|
vert_el_new = [], vert_new = []; |
||||
|
vert.each(function(v){ |
||||
|
var old_nr = self.vert.indexOf(v); |
||||
|
vert_el_new.append([self.vert_el[old_nr]]); |
||||
|
vert_new.append([v]); |
||||
|
|
||||
|
}); |
||||
|
self.vert = vert_new; |
||||
|
self.vert_el = vert_el_new; |
||||
|
|
||||
|
// Shorten vars
|
||||
|
var vert = self.vert, |
||||
|
vert_el = self.vert_el; |
||||
|
var t, h, l, w, left, width, |
||||
|
row_el, cursor = 0; |
||||
|
|
||||
|
// Loop over all vertical lines
|
||||
|
vert.each(function(v, nr){ |
||||
|
|
||||
|
// Use defaults if el == null (for first fillblock)
|
||||
|
var c = vert_el[nr] ? vert_el[nr].getCoordinates() : { |
||||
|
'left': 0, |
||||
|
'top': 0, |
||||
|
'width': page_c.x, |
||||
|
'height': 0 |
||||
|
}; |
||||
|
|
||||
|
// Loop till cursor gets to parent_element.width
|
||||
|
var fail_safe = 0; |
||||
|
while (cursor < page_c.x && fail_safe < 10){ |
||||
|
|
||||
|
t = vert[nr]; // Top is the same for every element in a row
|
||||
|
h = (nr == vert.length-1) ? (page_c.y - t) : vert[nr+1] - vert[nr]; // So is hight
|
||||
|
|
||||
|
// First element get special treatment
|
||||
|
if(nr == 0){ |
||||
|
l = 0; |
||||
|
w = c.width+(2*soften); |
||||
|
cursor += w; |
||||
|
} |
||||
|
else { |
||||
|
|
||||
|
row_el = self.firstFromLeft(cursor, t) // First next element
|
||||
|
left = row_el.el ? row_el.left : c.left-soften; |
||||
|
width = row_el.el ? row_el.left - cursor : c.left-soften; |
||||
|
|
||||
|
if(t == c.bottom+soften && !row_el.el) |
||||
|
width = page_c.x; |
||||
|
|
||||
|
l = cursor; |
||||
|
if(cursor < left){ |
||||
|
w = width; |
||||
|
cursor += w+(row_el.right - row_el.left); |
||||
|
} |
||||
|
else { |
||||
|
w = page_c.x-l; |
||||
|
cursor += w; |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
// Add it to the pile!
|
||||
|
if(h > 0 && w > 0){ |
||||
|
self.top.append([t]); self.left.append([l]); |
||||
|
self.width.append([w]); self.height.append([h]); |
||||
|
} |
||||
|
|
||||
|
fail_safe++; |
||||
|
|
||||
|
} |
||||
|
|
||||
|
cursor = 0; // New line, reset cursor position
|
||||
|
fail_safe = 0; |
||||
|
|
||||
|
}); |
||||
|
|
||||
|
// Create the fill blocks
|
||||
|
self.top.each(self.createFillItem.bind(self)); |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
createFillItem: function(top, nr){ |
||||
|
var self = this; |
||||
|
|
||||
|
var fill = new Element('div', { |
||||
|
'class': self.options.fillClass, |
||||
|
'styles': { |
||||
|
'position': 'absolute', |
||||
|
'background-color': 'rgba('+self.options.fillColor.join(',')+', '+self.options.fillOpacity+')', |
||||
|
'display': 'block', |
||||
|
'z-index': 2, |
||||
|
'top': self.top[nr], |
||||
|
'left': self.left[nr], |
||||
|
'height': self.height[nr], |
||||
|
'width': self.width[nr] |
||||
|
} |
||||
|
}).inject(self.options.inject || document.body); |
||||
|
|
||||
|
self.fills.include(fill); |
||||
|
}, |
||||
|
|
||||
|
// Find the first element after x,y coordinates
|
||||
|
firstFromLeft: function(x, y){ |
||||
|
var self = this; |
||||
|
|
||||
|
var lowest_left = null; |
||||
|
var return_data = {}; |
||||
|
|
||||
|
self.range.each(function(range){ |
||||
|
var is_within_height_range = range.top <= y && range.bottom > y, |
||||
|
is_within_width_range = range.left >= x, |
||||
|
more_left_then_previous = range.left < lowest_left || lowest_left == null; |
||||
|
|
||||
|
if(is_within_height_range && is_within_width_range && more_left_then_previous){ |
||||
|
lowest_left = range.left; |
||||
|
return_data = range; |
||||
|
} |
||||
|
}) |
||||
|
|
||||
|
return return_data |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
soften: function(el){ |
||||
|
var self = this; |
||||
|
var soften = self.options.soften; |
||||
|
|
||||
|
var c = el.getCoordinates(); |
||||
|
var from_color = 'rgba('+self.options.fillColor.join(',')+', '+self.options.fillOpacity+')'; |
||||
|
var to_color = 'rgba('+self.options.fillColor.join(',')+', 0)'; |
||||
|
|
||||
|
// Top
|
||||
|
self.createEdge({ |
||||
|
'top': c.top-soften, |
||||
|
'left': c.left-soften, |
||||
|
'width': c.width+(2*soften), |
||||
|
'background': '-webkit-gradient(linear, left top, left bottom, from('+from_color+'), to('+to_color+'))', |
||||
|
'background': '-moz-linear-gradient(top, '+from_color+', '+to_color+')' |
||||
|
}) |
||||
|
|
||||
|
// Right
|
||||
|
self.createEdge({ |
||||
|
'top': c.top-soften, |
||||
|
'left': c.right, |
||||
|
'height': c.height+(2*soften), |
||||
|
'background': '-webkit-gradient(linear, left, right, from('+from_color+'), to('+to_color+'))', |
||||
|
'background': '-moz-linear-gradient(right, '+from_color+', '+to_color+')' |
||||
|
}) |
||||
|
|
||||
|
// Bottom
|
||||
|
self.createEdge({ |
||||
|
'top': c.bottom, |
||||
|
'left': c.left-soften, |
||||
|
'width': c.width+(2*soften), |
||||
|
'background': '-webkit-gradient(linear, left bottom, left top, from('+from_color+'), to('+to_color+'))', |
||||
|
'background': '-moz-linear-gradient(bottom, '+from_color+', '+to_color+')' |
||||
|
}) |
||||
|
|
||||
|
// Left
|
||||
|
self.createEdge({ |
||||
|
'top': c.top-soften, |
||||
|
'left': c.left-soften, |
||||
|
'height': c.height+(2*soften), |
||||
|
'background': '-webkit-gradient(linear, right, left, from('+from_color+'), to('+to_color+'))', |
||||
|
'background': '-moz-linear-gradient(left, '+from_color+', '+to_color+')' |
||||
|
}) |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
createEdge: function(style){ |
||||
|
var self = this; |
||||
|
|
||||
|
var soften = self.options.soften; |
||||
|
var edge = new Element('div', { |
||||
|
'styles': Object.merge({ |
||||
|
'position': 'absolute', |
||||
|
'width': soften, |
||||
|
'height': soften, |
||||
|
}, style) |
||||
|
}).inject(self.options.inject || document.body) |
||||
|
|
||||
|
self.edges.include(edge); |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
destroy: function(){ |
||||
|
var self = this; |
||||
|
self.fills.each(function(fill){ |
||||
|
fill.destroy(); |
||||
|
}) |
||||
|
self.edges.each(function(edge){ |
||||
|
edge.destroy(); |
||||
|
}) |
||||
|
self.clean(); |
||||
|
}, |
||||
|
|
||||
|
numberSort: function (a, b) { |
||||
|
return a - b; |
||||
|
} |
||||
|
|
||||
|
}); |
||||
@ -0,0 +1,76 @@ |
|||||
|
var WizardBase = new Class({ |
||||
|
|
||||
|
Implements: [Options, Events], |
||||
|
|
||||
|
initialize: function(steps){ |
||||
|
var self = this; |
||||
|
|
||||
|
self.steps = steps; |
||||
|
self.start(); |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
start: function(){ |
||||
|
|
||||
|
|
||||
|
|
||||
|
}, |
||||
|
|
||||
|
nextStep: function(){ |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
previousStep: function(){ |
||||
|
|
||||
|
} |
||||
|
|
||||
|
}); |
||||
|
|
||||
|
WizardBase.Screen = new Class({ |
||||
|
|
||||
|
initialize: function(data){ |
||||
|
var self = this; |
||||
|
|
||||
|
self.data = data; |
||||
|
self.create() |
||||
|
|
||||
|
}, |
||||
|
|
||||
|
create: function(){ |
||||
|
var self = this; |
||||
|
|
||||
|
self.el = new Element('div.') |
||||
|
|
||||
|
|
||||
|
}, |
||||
|
|
||||
|
destroy: function(){ |
||||
|
this.el.destroy(); |
||||
|
|
||||
|
return this |
||||
|
} |
||||
|
|
||||
|
}) |
||||
|
|
||||
|
window.Wizard = new WizardBase([ |
||||
|
{ |
||||
|
'title': 'Fill in your username and password', |
||||
|
'Description': 'Outside blabla', |
||||
|
'tab': 'general', |
||||
|
'fields': ['username', 'password'] |
||||
|
}, |
||||
|
{ |
||||
|
'title': 'What do you use to download your movies', |
||||
|
'answers': [ |
||||
|
{'name': 'nzb', 'label': 'Usenet'}, |
||||
|
{'name': 'torrent', 'label': 'Torrents'} |
||||
|
] |
||||
|
}, |
||||
|
{ |
||||
|
'title': 'Do you have a login for any of the following sites', |
||||
|
'tab': 'providers', |
||||
|
'needs': function(){ |
||||
|
return self.config_nzb || self.config_torrent |
||||
|
} |
||||
|
} |
||||
|
]) |
||||
@ -1,8 +1,138 @@ |
|||||
|
from couchpotato.core.event import addEvent |
||||
from couchpotato.core.logger import CPLog |
from couchpotato.core.logger import CPLog |
||||
from couchpotato.core.providers.base import NZBProvider |
from couchpotato.core.providers.base import NZBProvider |
||||
|
from dateutil.parser import parse |
||||
|
from urllib import urlencode |
||||
|
from urllib2 import URLError |
||||
|
import time |
||||
|
|
||||
log = CPLog(__name__) |
log = CPLog(__name__) |
||||
|
|
||||
|
|
||||
class Newzbin(NZBProvider): |
class Newzbin(NZBProvider): |
||||
pass |
searchUrl = 'https://www.newzbin.com/search/' |
||||
|
|
||||
|
formatIds = { |
||||
|
2: ['scr'], |
||||
|
1: ['cam'], |
||||
|
4: ['tc'], |
||||
|
8: ['ts'], |
||||
|
1024: ['r5'], |
||||
|
} |
||||
|
cat_ids = [ |
||||
|
([2097152], ['1080p']), |
||||
|
([524288], ['720p']), |
||||
|
([262144], ['brrip']), |
||||
|
([2], ['dvdr']), |
||||
|
] |
||||
|
cat_backup_id = -1 |
||||
|
|
||||
|
def __init__(self): |
||||
|
addEvent('provider.nzb.search', self.search) |
||||
|
|
||||
|
def search(self, movie, quality): |
||||
|
|
||||
|
self.cleanCache(); |
||||
|
|
||||
|
results = [] |
||||
|
if not self.enabled() or not self.isAvailable(self.searchUrl): |
||||
|
return results |
||||
|
|
||||
|
formatId = self.getFormatId(type) |
||||
|
catId = self.getCatId(type) |
||||
|
|
||||
|
arguments = urlencode({ |
||||
|
'searchaction': 'Search', |
||||
|
'u_url_posts_only': '0', |
||||
|
'u_show_passworded': '0', |
||||
|
'q_url': 'imdb.com/title/' + movie.imdb, |
||||
|
'sort': 'ps_totalsize', |
||||
|
'order': 'asc', |
||||
|
'u_post_results_amt': '100', |
||||
|
'feed': 'rss', |
||||
|
'category': '6', |
||||
|
'ps_rb_video_format': str(catId), |
||||
|
'ps_rb_source': str(formatId), |
||||
|
}) |
||||
|
|
||||
|
url = "%s?%s" % (self.searchUrl, arguments) |
||||
|
cacheId = str('%s %s %s' % (movie.imdb, str(formatId), str(catId))) |
||||
|
singleCat = True |
||||
|
|
||||
|
try: |
||||
|
cached = False |
||||
|
if(self.cache.get(cacheId)): |
||||
|
data = True |
||||
|
cached = True |
||||
|
log.info('Getting RSS from cache: %s.' % cacheId) |
||||
|
else: |
||||
|
log.info('Searching: %s' % url) |
||||
|
data = self.urlopen(url, username = self.conf('username'), password = self.conf('password')) |
||||
|
self.cache[cacheId] = { |
||||
|
'time': time.time() |
||||
|
} |
||||
|
|
||||
|
except (IOError, URLError): |
||||
|
log.error('Failed to open %s.' % url) |
||||
|
return results |
||||
|
|
||||
|
if data: |
||||
|
try: |
||||
|
try: |
||||
|
if cached: |
||||
|
xml = self.cache[cacheId]['xml'] |
||||
|
else: |
||||
|
xml = self.getItems(data) |
||||
|
self.cache[cacheId]['xml'] = xml |
||||
|
except: |
||||
|
log.debug('No valid xml or to many requests.. You never know with %s.' % self.name) |
||||
|
return results |
||||
|
|
||||
|
for item in xml: |
||||
|
|
||||
|
title = self.gettextelement(item, "title") |
||||
|
if 'error' in title.lower(): continue |
||||
|
|
||||
|
REPORT_NS = 'http://www.newzbin.com/DTD/2007/feeds/report/'; |
||||
|
|
||||
|
# Add attributes to name |
||||
|
for attr in item.find('{%s}attributes' % REPORT_NS): |
||||
|
title += ' ' + attr.text |
||||
|
|
||||
|
id = int(self.gettextelement(item, '{%s}id' % REPORT_NS)) |
||||
|
size = str(int(self.gettextelement(item, '{%s}size' % REPORT_NS)) / 1024 / 1024) + ' mb' |
||||
|
date = str(self.gettextelement(item, '{%s}postdate' % REPORT_NS)) |
||||
|
|
||||
|
new = self.feedItem() |
||||
|
new.id = id |
||||
|
new.type = 'nzb' |
||||
|
new.name = title |
||||
|
new.date = int(time.mktime(parse(date).timetuple())) |
||||
|
new.size = self.parseSize(size) |
||||
|
new.url = str(self.gettextelement(item, '{%s}nzb' % REPORT_NS)) |
||||
|
new.detailUrl = str(self.gettextelement(item, 'link')) |
||||
|
new.content = self.gettextelement(item, "description") |
||||
|
new.score = self.calcScore(new, movie) |
||||
|
new.addbyid = True |
||||
|
new.checkNZB = False |
||||
|
|
||||
|
if new.date > time.time() - (int(self.config.get('NZB', 'retention')) * 24 * 60 * 60) and self.isCorrectMovie(new, movie, type, imdbResults = True, singleCategory = singleCat): |
||||
|
results.append(new) |
||||
|
log.info('Found: %s' % new.name) |
||||
|
|
||||
|
return results |
||||
|
except SyntaxError: |
||||
|
log.error('Failed to parse XML response from newzbin.com') |
||||
|
|
||||
|
return results |
||||
|
|
||||
|
def getFormatId(self, format): |
||||
|
for id, quality in self.formatIds.iteritems(): |
||||
|
for q in quality: |
||||
|
if q == format: |
||||
|
return id |
||||
|
|
||||
|
return self.catBackupId |
||||
|
|
||||
|
def isEnabled(self): |
||||
|
return NZBProvider.isEnabled(self) and self.conf('enabled') and self.conf('username') and self.conf('password') |
||||
|
|||||
@ -1,8 +1,126 @@ |
|||||
|
from couchpotato.core.event import addEvent |
||||
|
from couchpotato.core.helpers.variable import cleanHost |
||||
from couchpotato.core.logger import CPLog |
from couchpotato.core.logger import CPLog |
||||
from couchpotato.core.providers.base import NZBProvider |
from couchpotato.core.providers.base import NZBProvider |
||||
|
from dateutil.parser import parse |
||||
|
from urllib import urlencode |
||||
|
from urllib2 import URLError |
||||
|
import time |
||||
|
|
||||
log = CPLog(__name__) |
log = CPLog(__name__) |
||||
|
|
||||
|
|
||||
class Newznab(NZBProvider): |
class Newznab(NZBProvider): |
||||
pass |
|
||||
|
urls = { |
||||
|
'download': 'get&id=%s%s', |
||||
|
'detail': 'details&id=%s', |
||||
|
} |
||||
|
|
||||
|
cat_ids = [ |
||||
|
([2000], ['brrip']), |
||||
|
([2010], ['dvdr']), |
||||
|
([2030], ['cam', 'ts', 'dvdrip', 'tc', 'r5', 'scr']), |
||||
|
([2040], ['720p', '1080p']), |
||||
|
] |
||||
|
cat_backup_id = 2000 |
||||
|
|
||||
|
time_between_searches = 1 # Seconds |
||||
|
|
||||
|
def __init__(self): |
||||
|
addEvent('provider.nzb.search', self.search) |
||||
|
|
||||
|
def getUrl(self, type): |
||||
|
return cleanHost(self.conf('host')) + 'api?t=' + type |
||||
|
|
||||
|
def search(self, movie, quality): |
||||
|
|
||||
|
self.cleanCache(); |
||||
|
|
||||
|
results = [] |
||||
|
if not self.enabled() or not self.isAvailable(self.getUrl(self.searchUrl)): |
||||
|
return results |
||||
|
|
||||
|
catId = self.getCatId(type) |
||||
|
arguments = urlencode({ |
||||
|
'imdbid': movie.imdb.replace('tt', ''), |
||||
|
'cat': catId, |
||||
|
'apikey': self.conf('apikey'), |
||||
|
't': self.searchUrl, |
||||
|
'extended': 1 |
||||
|
}) |
||||
|
url = "%s&%s" % (self.getUrl(self.searchUrl), arguments) |
||||
|
cacheId = str(movie.imdb) + '-' + str(catId) |
||||
|
singleCat = (len(self.catIds.get(catId)) == 1 and catId != self.catBackupId) |
||||
|
|
||||
|
try: |
||||
|
cached = False |
||||
|
if(self.cache.get(cacheId)): |
||||
|
data = True |
||||
|
cached = True |
||||
|
log.info('Getting RSS from cache: %s.' % cacheId) |
||||
|
else: |
||||
|
log.info('Searching: %s' % url) |
||||
|
data = self.urlopen(url) |
||||
|
self.cache[cacheId] = { |
||||
|
'time': time.time() |
||||
|
} |
||||
|
|
||||
|
except (IOError, URLError): |
||||
|
log.error('Failed to open %s.' % url) |
||||
|
return results |
||||
|
|
||||
|
if data: |
||||
|
try: |
||||
|
try: |
||||
|
if cached: |
||||
|
xml = self.cache[cacheId]['xml'] |
||||
|
else: |
||||
|
xml = self.getItems(data) |
||||
|
self.cache[cacheId]['xml'] = xml |
||||
|
except: |
||||
|
log.debug('No valid xml or to many requests.' % self.name) |
||||
|
return results |
||||
|
|
||||
|
results = [] |
||||
|
for nzb in xml: |
||||
|
|
||||
|
for item in nzb: |
||||
|
if item.attrib.get('name') == 'size': |
||||
|
size = item.attrib.get('value') |
||||
|
elif item.attrib.get('name') == 'usenetdate': |
||||
|
date = item.attrib.get('value') |
||||
|
|
||||
|
new = self.feedItem() |
||||
|
new.id = self.gettextelement(nzb, "guid").split('/')[-1:].pop() |
||||
|
new.type = 'nzb' |
||||
|
new.name = self.gettextelement(nzb, "title") |
||||
|
new.date = int(time.mktime(parse(date).timetuple())) |
||||
|
new.size = int(size) / 1024 / 1024 |
||||
|
new.url = self.downloadLink(new.id) |
||||
|
new.detailUrl = self.detailLink(new.id) |
||||
|
new.content = self.gettextelement(nzb, "description") |
||||
|
new.score = self.calcScore(new, movie) |
||||
|
|
||||
|
if new.date > time.time() - (int(self.config.get('NZB', 'retention')) * 24 * 60 * 60) and self.isCorrectMovie(new, movie, type, imdbResults = True, singleCategory = singleCat): |
||||
|
results.append(new) |
||||
|
log.info('Found: %s' % new.name) |
||||
|
|
||||
|
return results |
||||
|
except SyntaxError: |
||||
|
log.error('Failed to parse XML response from Newznab') |
||||
|
return False |
||||
|
|
||||
|
return results |
||||
|
|
||||
|
def isEnabled(self): |
||||
|
return NZBProvider.isEnabled(self) and self.conf('enabled') and self.conf('host') and self.conf('apikey') |
||||
|
|
||||
|
def getApiExt(self): |
||||
|
return '&apikey=%s' % self.conf('apikey') |
||||
|
|
||||
|
def downloadLink(self, id): |
||||
|
return self.getUrl(self.downloadUrl) % (id, self.getApiExt()) |
||||
|
|
||||
|
def detailLink(self, id): |
||||
|
return self.getUrl(self.detailUrl) % id |
||||
|
|||||
@ -1,124 +0,0 @@ |
|||||
from app.config.cplog import CPLog |
|
||||
from app.lib.provider.yarr.base import nzbBase |
|
||||
from dateutil.parser import parse |
|
||||
from urllib import urlencode |
|
||||
from urllib2 import URLError |
|
||||
import time |
|
||||
|
|
||||
log = CPLog(__name__) |
|
||||
|
|
||||
class nzbMatrix(nzbBase): |
|
||||
"""Api for NZBMatrix""" |
|
||||
|
|
||||
name = 'NZBMatrix' |
|
||||
downloadUrl = 'https://api.nzbmatrix.com/v1.1/download.php?id=%s%s' |
|
||||
detailUrl = 'https://nzbmatrix.com/nzb-details.php?id=%s&hit=1' |
|
||||
searchUrl = 'http://rss.nzbmatrix.com/rss.php' |
|
||||
|
|
||||
catIds = { |
|
||||
42: ['720p', '1080p'], |
|
||||
2: ['cam', 'ts', 'dvdrip', 'tc', 'r5', 'scr'], |
|
||||
54: ['brrip'], |
|
||||
1: ['dvdr'] |
|
||||
} |
|
||||
catBackupId = 2 |
|
||||
|
|
||||
timeBetween = 10 # Seconds |
|
||||
|
|
||||
def __init__(self, config): |
|
||||
log.info('Using NZBMatrix provider') |
|
||||
|
|
||||
self.config = config |
|
||||
|
|
||||
def conf(self, option): |
|
||||
return self.config.get('NZBMatrix', option) |
|
||||
|
|
||||
def enabled(self): |
|
||||
return self.conf('enabled') and self.config.get('NZB', 'enabled') and self.conf('username') and self.conf('apikey') |
|
||||
|
|
||||
def find(self, movie, quality, type, retry = False): |
|
||||
|
|
||||
self.cleanCache(); |
|
||||
|
|
||||
results = [] |
|
||||
if not self.enabled() or not self.isAvailable(self.searchUrl): |
|
||||
return results |
|
||||
|
|
||||
catId = self.getCatId(type) |
|
||||
arguments = urlencode({ |
|
||||
'term': movie.imdb, |
|
||||
'subcat': catId, |
|
||||
'username': self.conf('username'), |
|
||||
'apikey': self.conf('apikey'), |
|
||||
'searchin': 'weblink', |
|
||||
'english': 1 if self.conf('english') else 0, |
|
||||
}) |
|
||||
url = "%s?%s" % (self.searchUrl, arguments) |
|
||||
cacheId = str(movie.imdb) + '-' + str(catId) |
|
||||
singleCat = (len(self.catIds.get(catId)) == 1 and catId != self.catBackupId) |
|
||||
|
|
||||
try: |
|
||||
cached = False |
|
||||
if(self.cache.get(cacheId)): |
|
||||
data = True |
|
||||
cached = True |
|
||||
log.info('Getting RSS from cache: %s.' % cacheId) |
|
||||
else: |
|
||||
log.info('Searching: %s' % url) |
|
||||
data = self.urlopen(url) |
|
||||
self.cache[cacheId] = { |
|
||||
'time': time.time() |
|
||||
} |
|
||||
|
|
||||
except (IOError, URLError): |
|
||||
log.error('Failed to open %s.' % url) |
|
||||
return results |
|
||||
|
|
||||
if data: |
|
||||
try: |
|
||||
try: |
|
||||
if cached: |
|
||||
xml = self.cache[cacheId]['xml'] |
|
||||
else: |
|
||||
xml = self.getItems(data) |
|
||||
self.cache[cacheId]['xml'] = xml |
|
||||
except: |
|
||||
log.debug('No valid xml or to many requests.. You never know with %s.' % self.name) |
|
||||
return results |
|
||||
|
|
||||
for nzb in xml: |
|
||||
|
|
||||
title = self.gettextelement(nzb, "title") |
|
||||
if 'error' in title.lower(): continue |
|
||||
|
|
||||
id = int(self.gettextelement(nzb, "link").split('&')[0].partition('id=')[2]) |
|
||||
size = self.gettextelement(nzb, "description").split('<br /><b>')[2].split('> ')[1] |
|
||||
date = str(self.gettextelement(nzb, "description").split('<br /><b>')[3].partition('Added:</b> ')[2]) |
|
||||
|
|
||||
new = self.feedItem() |
|
||||
new.id = id |
|
||||
new.type = 'nzb' |
|
||||
new.name = title |
|
||||
new.date = int(time.mktime(parse(date).timetuple())) |
|
||||
new.size = self.parseSize(size) |
|
||||
new.url = self.downloadLink(id) |
|
||||
new.detailUrl = self.detailLink(id) |
|
||||
new.content = self.gettextelement(nzb, "description") |
|
||||
new.score = self.calcScore(new, movie) |
|
||||
new.checkNZB = True |
|
||||
|
|
||||
if new.date > time.time() - (int(self.config.get('NZB', 'retention')) * 24 * 60 * 60): |
|
||||
if self.isCorrectMovie(new, movie, type, imdbResults = True, singleCategory = singleCat): |
|
||||
results.append(new) |
|
||||
log.info('Found: %s' % new.name) |
|
||||
else: |
|
||||
log.info('Found outside retention: %s' % new.name) |
|
||||
|
|
||||
return results |
|
||||
except SyntaxError: |
|
||||
log.error('Failed to parse XML response from NZBMatrix.com') |
|
||||
|
|
||||
return results |
|
||||
|
|
||||
def getApiExt(self): |
|
||||
return '&username=%s&apikey=%s' % (self.conf('username'), self.conf('apikey')) |
|
||||
@ -1,8 +1,122 @@ |
|||||
|
from couchpotato.core.event import addEvent |
||||
from couchpotato.core.logger import CPLog |
from couchpotato.core.logger import CPLog |
||||
from couchpotato.core.providers.base import NZBProvider |
from couchpotato.core.providers.base import NZBProvider |
||||
|
from dateutil.parser import parse |
||||
|
from urllib import urlencode |
||||
|
from urllib2 import URLError |
||||
|
import time |
||||
|
|
||||
log = CPLog(__name__) |
log = CPLog(__name__) |
||||
|
|
||||
|
|
||||
class Nzbs(NZBProvider): |
class Nzbs(NZBProvider): |
||||
pass |
|
||||
|
urls = { |
||||
|
'download': 'http://nzbs.org/index.php?action=getnzb&nzbid=%s%s', |
||||
|
'nfo': 'http://nzbs.org/index.php?action=view&nzbid=%s&nfo=1', |
||||
|
'detail': 'http://nzbs.org/index.php?action=view&nzbid=%s', |
||||
|
'api': 'http://nzbs.org/rss.php', |
||||
|
} |
||||
|
|
||||
|
cat_ids = [ |
||||
|
([4], ['720p', '1080p']), |
||||
|
([2], ['cam', 'ts', 'dvdrip', 'tc', 'brrip', 'r5', 'scr']), |
||||
|
([9], ['dvdr']), |
||||
|
] |
||||
|
cat_backup_id = 't2' |
||||
|
|
||||
|
time_between_searches = 3 # Seconds |
||||
|
|
||||
|
def __init__(self): |
||||
|
addEvent('provider.nzb.search', self.search) |
||||
|
|
||||
|
def search(self, movie, quality): |
||||
|
|
||||
|
self.cleanCache(); |
||||
|
|
||||
|
results = [] |
||||
|
if not self.enabled() or not self.isAvailable(self.apiUrl + '?test' + self.getApiExt()): |
||||
|
return results |
||||
|
|
||||
|
catId = self.getCatId(type) |
||||
|
arguments = urlencode({ |
||||
|
'action':'search', |
||||
|
'q': self.toSearchString(movie.name), |
||||
|
'catid': catId, |
||||
|
'i': self.conf('id'), |
||||
|
'h': self.conf('key'), |
||||
|
'age': self.config.get('NZB', 'retention') |
||||
|
}) |
||||
|
url = "%s?%s" % (self.apiUrl, arguments) |
||||
|
cacheId = str(movie.imdb) + '-' + str(catId) |
||||
|
singleCat = (len(self.catIds.get(catId)) == 1 and catId != self.catBackupId) |
||||
|
|
||||
|
try: |
||||
|
cached = False |
||||
|
if(self.cache.get(cacheId)): |
||||
|
data = True |
||||
|
cached = True |
||||
|
log.info('Getting RSS from cache: %s.' % cacheId) |
||||
|
else: |
||||
|
log.info('Searching: %s' % url) |
||||
|
data = self.urlopen(url) |
||||
|
self.cache[cacheId] = { |
||||
|
'time': time.time() |
||||
|
} |
||||
|
except (IOError, URLError): |
||||
|
log.error('Failed to open %s.' % url) |
||||
|
return results |
||||
|
|
||||
|
if data: |
||||
|
log.debug('Parsing NZBs.org RSS.') |
||||
|
try: |
||||
|
try: |
||||
|
if cached: |
||||
|
xml = self.cache[cacheId]['xml'] |
||||
|
else: |
||||
|
xml = self.getItems(data) |
||||
|
self.cache[cacheId]['xml'] = xml |
||||
|
except: |
||||
|
retry = False |
||||
|
if retry == False: |
||||
|
log.error('No valid xml, to many requests? Try again in 15sec.') |
||||
|
time.sleep(15) |
||||
|
return self.find(movie, quality, type, retry = True) |
||||
|
else: |
||||
|
log.error('Failed again.. disable %s for 15min.' % self.name) |
||||
|
self.available = False |
||||
|
return results |
||||
|
|
||||
|
for nzb in xml: |
||||
|
|
||||
|
id = int(self.gettextelement(nzb, "link").partition('nzbid=')[2]) |
||||
|
|
||||
|
size = self.gettextelement(nzb, "description").split('</a><br />')[1].split('">')[1] |
||||
|
|
||||
|
new = self.feedItem() |
||||
|
new.id = id |
||||
|
new.type = 'nzb' |
||||
|
new.name = self.gettextelement(nzb, "title") |
||||
|
new.date = int(time.mktime(parse(self.gettextelement(nzb, "pubDate")).timetuple())) |
||||
|
new.size = self.parseSize(size) |
||||
|
new.url = self.downloadLink(id) |
||||
|
new.detailUrl = self.detailLink(id) |
||||
|
new.content = self.gettextelement(nzb, "description") |
||||
|
new.score = self.calcScore(new, movie) |
||||
|
|
||||
|
if self.isCorrectMovie(new, movie, type, singleCategory = singleCat): |
||||
|
results.append(new) |
||||
|
log.info('Found: %s' % new.name) |
||||
|
|
||||
|
return results |
||||
|
except SyntaxError: |
||||
|
log.error('Failed to parse XML response from NZBs.org') |
||||
|
return False |
||||
|
|
||||
|
return results |
||||
|
|
||||
|
def isEnabled(self): |
||||
|
return NZBProvider.isEnabled(self) and self.conf('enabled') and self.conf('id') and self.conf('key') |
||||
|
|
||||
|
def getApiExt(self): |
||||
|
return '&i=%s&h=%s' % (self.conf('id'), self.conf('key')) |
||||
|
|||||
Loading…
Reference in new issue