From 0c1a4b019397214b3cb6292d13f7707cd5d1a8ee Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Fri, 3 Feb 2012 20:41:21 +0100
Subject: [PATCH] NZBClub provider

---
 .../core/providers/nzb/mysterbin/main.py      |  2 +-
 .../core/providers/nzb/nzbclub/__init__.py    | 22 +++++
 .../core/providers/nzb/nzbclub/main.py        | 92 +++++++++++++++++++
 3 files changed, 115 insertions(+), 1 deletion(-)
 create mode 100644 couchpotato/core/providers/nzb/nzbclub/__init__.py
 create mode 100644 couchpotato/core/providers/nzb/nzbclub/main.py

diff --git a/couchpotato/core/providers/nzb/mysterbin/main.py b/couchpotato/core/providers/nzb/mysterbin/main.py
index 2b59fc57..af6c1eee 100644
--- a/couchpotato/core/providers/nzb/mysterbin/main.py
+++ b/couchpotato/core/providers/nzb/mysterbin/main.py
@@ -92,7 +92,7 @@ class Mysterbin(NZBProvider):
                     except:
                         pass
 
-                return []
+                return results
             except AttributeError:
                 log.debug('No search results found.')
 
diff --git a/couchpotato/core/providers/nzb/nzbclub/__init__.py b/couchpotato/core/providers/nzb/nzbclub/__init__.py
new file mode 100644
index 00000000..18f4e33d
--- /dev/null
+++ b/couchpotato/core/providers/nzb/nzbclub/__init__.py
@@ -0,0 +1,22 @@
+from .main import NZBClub
+
+def start():
+    return NZBClub()
+
+config = [{
+    'name': 'nzbclub',
+    'groups': [
+        {
+            'tab': 'providers',
+            'name': 'NZBClub',
+            'description': '',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                    'default': True,
+                },
+            ],
+        },
+    ],
+}]
diff --git a/couchpotato/core/providers/nzb/nzbclub/main.py b/couchpotato/core/providers/nzb/nzbclub/main.py
new file mode 100644
index 00000000..e45bf9ef
--- /dev/null
+++ b/couchpotato/core/providers/nzb/nzbclub/main.py
@@ -0,0 +1,92 @@
+from BeautifulSoup import BeautifulSoup, SoupStrainer
+from couchpotato.core.event import fireEvent
+from couchpotato.core.helpers.encoding import toUnicode
+from couchpotato.core.helpers.rss import RSS
+from couchpotato.core.helpers.variable import tryInt
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.nzb.base import NZBProvider
+from couchpotato.environment import Env
+from dateutil.parser import parse
+import time
+import urllib
+import xml.etree.ElementTree as XMLTree
+
+log = CPLog(__name__)
+
+
+class NZBClub(NZBProvider, RSS):
+
+    urls = {
+        'search': 'http://www.nzbclub.com/nzbfeed.aspx?%s',
+    }
+
+    http_time_between_calls = 1 #seconds
+
+    def search(self, movie, quality):
+
+        results = []
+        if self.isDisabled() or not self.isAvailable(self.urls['search']):
+            return results
+
+        q = '"%s" %s %s' % (movie['library']['titles'][0]['title'], movie['library']['year'], quality.get('identifier'))
+        for ignored in Env.setting('ignored_words', 'searcher').split(','):
+            q = '%s -%s' % (q, ignored.strip())
+
+        params = {
+            'q': q,
+            'ig': '1',
+            'rpp': 200,
+            'st': 1,
+            'sp': 1,
+            'ns': 1,
+        }
+
+        cache_key = 'nzbclub.%s' % q
+        data = self.getCache(cache_key, self.urls['search'] % urllib.urlencode(params))
+        if data:
+            try:
+                try:
+                    data = XMLTree.fromstring(data)
+                    nzbs = self.getElements(data, 'channel/item')
+                except Exception, e:
+                    log.debug('%s, %s' % (self.getName(), e))
+                    return results
+
+                for nzb in nzbs:
+
+                    nzbclub_id = tryInt(self.getTextElement(nzb, "link").split('/nzb_view/')[1].split('/')[0])
+                    enclosure = self.getElement(nzb, "enclosure").attrib
+                    size = enclosure['length']
+                    date = self.getTextElement(nzb, "pubDate")
+
+                    description = ''
+                    if 'nfo files' in self.getTextElement(nzb, "description"):
+                        description = toUnicode(self.getCache('nzbclub.%s' % nzbclub_id, self.getTextElement(nzb, "link"), timeout = 25920000))
+
+                    new = {
+                        'id': nzbclub_id,
+                        'type': 'nzb',
+                        'provider': self.getName(),
+                        'name': toUnicode(self.getTextElement(nzb, "title")),
+                        'age': self.calculateAge(int(time.mktime(parse(date).timetuple()))),
+                        'size': tryInt(size) / 1024 / 1024,
+                        'url': enclosure['url'],
+                        'download': enclosure['url'],
+                        'detail_url': self.getTextElement(nzb, "link"),
+                        'description': description,
+                    }
+                    new['score'] = fireEvent('score.calculate', new, movie, single = True)
+
+                    is_correct_movie = fireEvent('searcher.correct_movie',
+                                                 nzb = new, movie = movie, quality = quality,
+                                                 imdb_results = False, single_category = False, single = True)
+
+                    if is_correct_movie:
+                        results.append(new)
+                        self.found(new)
+
+                return results
+            except SyntaxError:
+                log.error('Failed to parse XML response from NZBClub')
+
+        return results