NZB/Torrent provider base

couchpotato/core/downloaders/sabnzbd/__init__.py

@@ -26,6 +26,12 @@ config = [{
                     'label': 'Api Key',
                     'description': 'Used for all calls to Sabnzbd.',
                 },
+                {
+                    'advanced': True,
+                    'name': 'pp_directory',
+                    'type': 'directory',
+                    'description': 'Your Post-Processing Script directory, set in Sabnzbd > Config > Directories.',
+                },
             ],
         }
     ],

couchpotato/core/notifications/base.py

@@ -14,12 +14,6 @@ class Notification(Plugin):
     def notify(self, message = '', data = {}):
         pass
 
-    def isDisabled(self):
-        return not self.isEnabled()
-
-    def isEnabled(self):
-        return self.conf('enabled', True)
-
     def test(self):
         success = self.notify(message = self.test_message)
 

couchpotato/core/plugins/base.py

@@ -7,7 +7,10 @@ import os.path
 class Plugin():
 
     def conf(self, attr):
-        return Env.setting(attr, self.__class__.__name__.lower())
+        return Env.setting(attr, self.getName().lower())
+
+    def getName(self):
+        return self.__class__.__name__
 
     def registerStatic(self, file_path):
 
@@ -25,3 +28,9 @@ class Plugin():
         dir = os.path.join(plugin_dir, 'static')
 
         return send_from_directory(dir, file)
+
+    def isDisabled(self):
+        return not self.isEnabled()
+
+    def isEnabled(self):
+        return self.conf('enabled', True)

couchpotato/core/plugins/renamer/__init__.py

@@ -1,33 +1,42 @@
-def start():
+from couchpotato.core.plugins.renamer.main import Renamer
-    pass
 
-#config = [{
+def start():
-#    'name': 'Renamer',
+    return Renamer()
-#    'tab': 'renaming',
+
-#    'options': {
+config = [{
-#        'enabled': {
+    'name': 'renamer',
-#            'default': False,
+    'groups': [
-#            'type': 'bool',
+        {
-#            'description': 'Enable renaming',
+            'tab': 'renamer',
-#        },
+            'name': 'tmdb',
-#        'from': {
+            'label': 'TheMovieDB',
-#            'default': '',
+            'advanced': True,
-#            'type': 'directory',
+            'description': 'Move and rename your downloaded movies to your movie directory.',
-#            'label': 'From',
+            'options': [
-#            'description': 'Folder where the movies are downloaded to.',
+                {
-#        },
+                    'name': 'enabled',
-#        'to': {
+                    'default': False,
-#            'default': '',
+                    'type': 'enabler',
-#            'type': 'directory',
+                },
-#            'label': 'To',
+                {
-#            'description': 'Folder where the movies will be moved to.',
+                    'name': 'from',
-#        },
+                    'type': 'directory',
-#        'run_every': {
+                    'description': 'Folder where the movies are downloaded to.',
-#            'default': 1,
+                },
-#            'type': 'int',
+                {
-#            'unit': 'min(s)',
+                    'name': 'to',
-#            'description': 'Search for new movies inside the folder every X minutes.',
+                    'type': 'directory',
-#        }
+                    'description': 'Folder where the movies will be moved to.',
-#    }
+                },
-#}]
+                {
-config = []
+                    'name': 'run_every',
+                    'label': 'Run every',
+                    'default': 1,
+                    'type': 'int',
+                    'unit': 'min(s)',
+                    'description': 'Search for new movies inside the folder every X minutes.',
+                }
+            ],
+        },
+    ],
+}]

couchpotato/core/plugins/renamer/main.py

@@ -0,0 +1,10 @@
+from couchpotato.core.logger import CPLog
+from couchpotato.core.plugins.base import Plugin
+
+log = CPLog(__name__)
+
+
+class Renamer(Plugin):
+
+    def __init__(self):
+        pass

couchpotato/core/providers/base.py

@@ -8,3 +8,25 @@ class Provider(Plugin):
 
     type = None # movie, nzb, torrent, subtitle, trailer
     timeout = 10 # Default timeout for url requests
+
+
+class MovieProvider(Provider):
+    type = 'movie'
+
+
+class NZBProvider(Provider):
+    type = 'nzb'
+
+    time_between_searches = 10 # Seconds
+
+
+class TorrentProvider(Provider):
+    type = 'torrent'
+
+
+class SubtitleProvider(Provider):
+    type = 'subtitle'
+
+
+class TrailerProvider(Provider):
+    type = 'trailer'

couchpotato/core/providers/imdb/main.py

@@ -1,14 +1,12 @@
 from couchpotato.core.event import addEvent
 from couchpotato.core.logger import CPLog
-from couchpotato.core.providers.base import Provider
+from couchpotato.core.providers.base import MovieProvider
 from imdb import IMDb
 
 log = CPLog(__name__)
 
 
-class IMDB(Provider):
+class IMDB(MovieProvider):
-
-    type = 'movie'
 
     def __init__(self):
 

couchpotato/core/providers/newzbin/__init__.py

@@ -0,0 +1,29 @@
+from .main import Newzbin
+
+def start():
+    return Newzbin()
+
+config = [{
+    'name': 'newzbin',
+    'groups': [
+        {
+            'tab': 'providers',
+            'name': 'newzbin',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'username',
+                    'default': '',
+                },
+                {
+                    'name': 'password',
+                    'default': '',
+                    'type': 'password',
+                },
+            ],
+        },
+    ],
+}]

couchpotato/core/providers/newzbin/main.py

@@ -0,0 +1,8 @@
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.base import NZBProvider
+
+log = CPLog(__name__)
+
+
+class Newzbin(NZBProvider):
+    pass

couchpotato/core/providers/newznab/__init__.py

@@ -0,0 +1,30 @@
+from .main import Newznab
+
+def start():
+    return Newznab()
+
+config = [{
+    'name': 'newznab',
+    'groups': [
+        {
+            'tab': 'providers',
+            'name': 'newznab',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'host',
+                    'default': 'http://nzb.su',
+                },
+                {
+                    'name': 'api_key',
+                    'default': '',
+                    'label': 'Api Key',
+                    'description': 'Can be found after login on the "API" page, bottom left. The string after "&apikey=".',
+                },
+            ],
+        },
+    ],
+}]

couchpotato/core/providers/newznab/main.py

@@ -0,0 +1,8 @@
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.base import NZBProvider
+
+log = CPLog(__name__)
+
+
+class Newznab(NZBProvider):
+    pass

couchpotato/core/providers/nzbmatrix/__init__.py

@@ -0,0 +1,29 @@
+from .main import NZBMatrix
+
+def start():
+    return NZBMatrix()
+
+config = [{
+    'name': 'nzbmatrix',
+    'groups': [
+        {
+            'tab': 'providers',
+            'name': 'nzbmatrix',
+            'label': 'NZBMatrix',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'username',
+                },
+                {
+                    'name': 'api_key',
+                    'default': '9b939aee0aaafc12a65bf448e4af9543',
+                    'label': 'Api Key',
+                },
+            ],
+        },
+    ],
+}]

couchpotato/core/providers/nzbmatrix/main.py

@@ -0,0 +1,116 @@
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.base import NZBProvider
+
+log = CPLog(__name__)
+
+
+class NZBMatrix(NZBProvider):
+
+    urls = {
+        'download': 'https://api.nzbmatrix.com/v1.1/download.php?id=%s%s',
+        'detail': 'https://nzbmatrix.com/nzb-details.php?id=%s&hit=1',
+        'search': 'http://rss.nzbmatrix.com/rss.php',
+    }
+
+    cat_ids = [
+        ([42], ['720p', '1080p']),
+        ([2], ['cam', 'ts', 'dvdrip', 'tc', 'r5', 'scr']),
+        ([54], ['brrip']),
+        ([1], ['dvdr']),
+    ]
+    cat_backup_id = 2
+
+    def __init__(self, config):
+        log.info('Using NZBMatrix provider')
+
+        self.config = config
+
+    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'))
+
+    def isEnabled(self):
+        return self.conf('enabled') and self.conf('username') and self.conf('apikey')

couchpotato/core/providers/nzbs/__init__.py

@@ -0,0 +1,30 @@
+from .main import Nzbs
+
+def start():
+    return Nzbs()
+
+config = [{
+    'name': 'nzbs',
+    'groups': [
+        {
+            'tab': 'providers',
+            'name': 'nzbs',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'id',
+                    'label': 'Id',
+                    'description': 'Can be found <a href="http://nzbs.org/index.php?action=rss" target="_blank">here</a>, the number after "&amp;i="',
+                },
+                {
+                    'name': 'api_key',
+                    'label': 'Api Key',
+                    'description': 'Can be found <a href="http://nzbs.org/index.php?action=rss" target="_blank">here</a>, the string after "&amp;h="'
+                },
+            ],
+        },
+    ],
+}]

couchpotato/core/providers/nzbs/main.py

@@ -0,0 +1,8 @@
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.base import NZBProvider
+
+log = CPLog(__name__)
+
+
+class Nzbs(NZBProvider):
+    pass

couchpotato/core/providers/themoviedb/main.py

@@ -2,16 +2,15 @@ from __future__ import with_statement
 from couchpotato.core.event import addEvent
 from couchpotato.core.helpers.encoding import simplifyString, toUnicode
 from couchpotato.core.logger import CPLog
-from couchpotato.core.providers.base import Provider
+from couchpotato.core.providers.base import MovieProvider
 from libs.themoviedb import tmdb
 
 log = CPLog(__name__)
 
 
-class TheMovieDb(Provider):
+class TheMovieDb(MovieProvider):
     """Api for theMovieDb"""
 
-    type = 'movie'
     apiUrl = 'http://api.themoviedb.org/2.1'
     imageUrl = 'http://hwcdn.themoviedb.org'
 

couchpotato/core/providers/thepiratebay/__init__.py

@@ -0,0 +1,24 @@
+from .main import ThePirateBay
+
+def start():
+    return ThePirateBay()
+
+config = [{
+    'name': 'themoviedb',
+    'groups': [
+        {
+            'tab': 'providers',
+            'name': 'tmdb',
+            'label': 'TheMovieDB',
+            'advanced': True,
+            'description': 'Used for all calls to TheMovieDB.',
+            'options': [
+                {
+                    'name': 'api_key',
+                    'default': '9b939aee0aaafc12a65bf448e4af9543',
+                    'label': 'Api Key',
+                },
+            ],
+        },
+    ],
+}]

couchpotato/core/providers/thepiratebay/main.py

@@ -0,0 +1,8 @@
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.base import TorrentProvider
+
+log = CPLog(__name__)
+
+
+class ThePirateBay(TorrentProvider):
+    pass

couchpotato/static/scripts/couchpotato.js

@@ -92,6 +92,7 @@ var CouchPotato = new Class({
 	getPage: function(name){
 		return this.pages[name]
 	}
+
 });
 window.App = new CouchPotato();
 

couchpotato/static/scripts/page.js

@@ -8,18 +8,17 @@ var PageBase = new Class({
 
 	has_tab: true,
 
-	initialize: function(parent, options) {
+	initialize: function(options) {
 		var self = this;
 
 		self.setOptions(options)
-		self.setParent(parent)
 
 		// Create main page container
 		self.el = new Element('div.page.'+self.name);
 
 		// Create tab for page
 		if(self.has_tab){
-			var nav = self.getParent().getBlock('navigation');
+			var nav = App.getBlock('navigation');
 			self.tab = nav.addTab({
 				'href': '/'+self.name,
 				'title': self.title,
@@ -49,14 +48,6 @@ var PageBase = new Class({
 	getName: function(){
 		return this.name
 	},
-
-	setParent: function(parent){
-		this.app = parent
-	},
-
-	getParent: function(){
-		return this.app
-	},
 	
 	show: function(){
 		this.el.addClass('active');

couchpotato/static/scripts/page/settings.js

@@ -6,18 +6,10 @@ Page.Settings = new Class({
 	title: 'Change settings.',
 
 	tabs: {
-		'general': {
+		'general': {},
-			'label': 'General'
+		'providers': {},
-		},
+		'downloaders': {},
-		'providers': {
+		'notifications': {}
-			'label': 'Providers'
-		},
-		'downloaders': {
-			'label': 'Downloaders'
-		},
-		'notifications': {
-			'label': 'Notifications'
-		}
 	},
 
 	open: function(action, params){
@@ -150,16 +142,17 @@ Page.Settings = new Class({
 		if(self.tabs[tab_name] && self.tabs[tab_name].tab)
 			return self.tabs[tab_name].tab
 
+		var label = (tab.label || tab.name).capitalize()
 		var tab_el = new Element('li').adopt(
 			new Element('a', {
 				'href': '/'+self.name+'/'+tab_name+'/',
-				'text': (tab.label || tab.name).capitalize()
+				'text': label
 			})
 		).inject(self.tabs_container);
 
 		if(!self.tabs[tab_name])
 			self.tabs[tab_name] = {
-				'label': tab.label || tab.name
+				'label': label
 			}
 
 		self.tabs[tab_name] = Object.merge(self.tabs[tab_name], {
@@ -179,7 +172,7 @@ Page.Settings = new Class({
 			'class': group.advanced ? 'inlineLabels advanced' : 'inlineLabels'
 		}).adopt(
 			new Element('h2', {
-				'text': group.label || group.name.capitalize()
+				'text': (group.label || group.name).capitalize()
 			}).adopt(
 				new Element('span.hint', {
 					'text': group.description
@@ -236,7 +229,7 @@ var OptionBase = new Class({
 	createLabel: function(){
 		var self = this;
 		return new Element('label', {
-			'text': self.options.label || self.options.name.capitalize()
+			'text': (self.options.label || self.options.name).capitalize()
 		})
 	},