From f1a1db8d5bf1018e8a22c703a5191d235a658989 Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Sun, 9 Dec 2012 21:50:07 +0100
Subject: [PATCH] Revert "Remove NZBsRus"

This reverts commit f515cd2477843e9c613ff01adc98d7ff240f6493.
---
 .../core/providers/nzb/nzbsrus/__init__.py    |  40 +++++++
 .../core/providers/nzb/nzbsrus/main.py        | 101 ++++++++++++++++++
 2 files changed, 141 insertions(+)
 create mode 100644 couchpotato/core/providers/nzb/nzbsrus/__init__.py
 create mode 100644 couchpotato/core/providers/nzb/nzbsrus/main.py

diff --git a/couchpotato/core/providers/nzb/nzbsrus/__init__.py b/couchpotato/core/providers/nzb/nzbsrus/__init__.py
new file mode 100644
index 00000000..cd4d6691
--- /dev/null
+++ b/couchpotato/core/providers/nzb/nzbsrus/__init__.py
@@ -0,0 +1,40 @@
+from .main import Nzbsrus
+
+def start():
+    return Nzbsrus()
+
+config = [{
+    'name': 'nzbsrus',
+    'groups': [
+        {
+            'tab': 'searcher',
+            'subtab': 'nzb_providers',
+            'name': 'nzbsrus',
+            'label': 'Nzbsrus',
+            'description': 'See <a href="https://www.nzbsrus.com/">NZBsRus</a>',
+            'wizard': True,
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'userid',
+                    'label': 'User ID',
+                },
+                {
+                    'name': 'api_key',
+                    'default': '',
+                    'label': 'Api Key',
+                },
+                {
+                    'name': 'english_only',
+                    'default': 1,
+                    'type': 'bool',
+                    'label': 'English only',
+                    'description': 'Only search for English spoken movies on Nzbsrus',
+                },
+            ],
+        },
+    ],
+}]
diff --git a/couchpotato/core/providers/nzb/nzbsrus/main.py b/couchpotato/core/providers/nzb/nzbsrus/main.py
new file mode 100644
index 00000000..b92d6774
--- /dev/null
+++ b/couchpotato/core/providers/nzb/nzbsrus/main.py
@@ -0,0 +1,101 @@
+from couchpotato.core.event import fireEvent
+from couchpotato.core.helpers.encoding import tryUrlencode
+from couchpotato.core.helpers.rss import RSS
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.nzb.base import NZBProvider
+from couchpotato.environment import Env
+import time
+import xml.etree.ElementTree as XMLTree
+
+log = CPLog(__name__)
+
+class Nzbsrus(NZBProvider, RSS):
+
+    urls = {
+        'download': 'https://www.nzbsrus.com/nzbdownload_rss.php/%s',
+        'detail': 'https://www.nzbsrus.com/nzbdetails.php?id=%s',
+        'search': 'https://www.nzbsrus.com/api.php?extended=1&xml=1&listname={date,grabs}',
+    }
+
+    cat_ids = [
+        ([90, 45, 51], ['720p', '1080p', 'brrip', 'bd50', 'dvdr']),
+        ([48, 51], ['cam', 'ts', 'dvdrip', 'tc', 'r5', 'scr']),
+    ]
+    cat_backup_id = 240
+
+    def search(self, movie, quality):
+
+        results = []
+
+        if self.isDisabled():
+            return results
+
+        cat_id_string = '&'.join(['c%s=1' % x for x in self.getCatId(quality.get('identifier'))])
+
+        arguments = tryUrlencode({
+            'searchtext': 'imdb:' + movie['library']['identifier'][2:],
+            'uid': self.conf('userid'),
+            'key': self.conf('api_key'),
+            'age': Env.setting('retention', section = 'nzb'),
+
+        })
+
+        # check for english_only
+        if self.conf('english_only'):
+            arguments += "&lang0=1&lang3=1&lang1=1"
+
+        url = "%s&%s&%s" % (self.urls['search'], arguments , cat_id_string)
+
+        cache_key = 'nzbsrus_1.%s.%s' % (movie['library'].get('identifier'), cat_id_string)
+        single_cat = True
+
+        data = self.getCache(cache_key, url, cache_timeout = 1800, headers = {'User-Agent': Env.getIdentifier()})
+        if data:
+            try:
+                try:
+                    data = XMLTree.fromstring(data)
+                    nzbs = self.getElements(data, 'results/result')
+                except Exception, e:
+                    log.debug('%s, %s', (self.getName(), e))
+                    return results
+
+                for nzb in nzbs:
+
+                    title = self.getTextElement(nzb, "name")
+                    if 'error' in title.lower(): continue
+
+                    id = self.getTextElement(nzb, "id")
+                    size = int(round(int(self.getTextElement(nzb, "size")) / 1048576))
+                    age = int(round((time.time() - int(self.getTextElement(nzb, "postdate"))) / 86400))
+
+                    new = {
+                        'id': id,
+                        'type': 'nzb',
+                        'provider': self.getName(),
+                        'name': title,
+                        'age': age,
+                        'size': size,
+                        'url': self.urls['download'] % id + self.getApiExt() + self.getTextElement(nzb, "key"),
+                        'download': self.download,
+                        'detail_url': self.urls['detail'] % id,
+                        'description': self.getTextElement(nzb, "addtext"),
+                        'check_nzb': True,
+                    }
+
+                    is_correct_movie = fireEvent('searcher.correct_movie',
+                                                 nzb = new, movie = movie, quality = quality,
+                                                 imdb_results = True, single = True)
+
+                    if is_correct_movie:
+                        new['score'] = fireEvent('score.calculate', new, movie, single = True)
+                        results.append(new)
+                        self.found(new)
+
+                return results
+            except SyntaxError:
+                log.error('Failed to parse XML response from Nzbsrus.com')
+
+        return results
+
+    def getApiExt(self):
+        return '/%s/' % (self.conf('userid'))