From 50b9c734da3da2fe806d05a740eb894530d74ba6 Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Sat, 20 Aug 2011 23:21:27 +0200
Subject: [PATCH] NZBGet support

---
 .../core/downloaders/nzbget/__init__.py       | 37 ++++++++++++
 couchpotato/core/downloaders/nzbget/main.py   | 60 +++++++++++++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 couchpotato/core/downloaders/nzbget/__init__.py
 create mode 100644 couchpotato/core/downloaders/nzbget/main.py

diff --git a/couchpotato/core/downloaders/nzbget/__init__.py b/couchpotato/core/downloaders/nzbget/__init__.py
new file mode 100644
index 00000000..1acf097d
--- /dev/null
+++ b/couchpotato/core/downloaders/nzbget/__init__.py
@@ -0,0 +1,37 @@
+from .main import NZBGet
+
+def start():
+    return NZBGet()
+
+config = [{
+    'name': 'nzbget',
+    'groups': [
+        {
+            'tab': 'downloaders',
+            'name': 'nzbget',
+            'label': 'NZBGet',
+            'description': 'Send NZBs to your NZBGet installation.',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'default': 0,
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'host',
+                    'default': 'localhost:6789',
+                    'description': 'Hostname with port. Usually <strong>localhost:6789</strong>',
+                },
+                {
+                    'name': 'password',
+                    'description': 'Default NZBGet password is <i>tegbzn6789</i>',
+                },
+                {
+                    'name': 'category',
+                    'default': 'Movies',
+                    'description': 'The category CP places the nzb in. Like <strong>movies</strong> or <strong>couchpotato</strong>',
+                },
+            ],
+        }
+    ],
+}]
diff --git a/couchpotato/core/downloaders/nzbget/main.py b/couchpotato/core/downloaders/nzbget/main.py
new file mode 100644
index 00000000..b7843b27
--- /dev/null
+++ b/couchpotato/core/downloaders/nzbget/main.py
@@ -0,0 +1,60 @@
+from base64 import standard_b64encode
+from couchpotato.core.downloaders.base import Downloader
+from couchpotato.core.logger import CPLog
+from inspect import isfunction
+import socket
+import xmlrpclib
+
+log = CPLog(__name__)
+
+class NZBGet(Downloader):
+
+    type = ['nzb']
+
+    url = 'http://nzbget:%(password)s@%(host)s/xmlrpc'
+
+    def download(self, data = {}, movie = {}):
+
+        if self.isDisabled() or not self.isCorrectType(data.get('type')):
+            return
+
+        log.info('Sending "%s" to NZBGet.' % data.get('name'))
+
+        url = self.url % {'host': self.conf('host'), 'password': self.conf('password')}
+        nzb_name = data.get('name') + '.nzb'
+
+        rpc = xmlrpclib.ServerProxy(url)
+        try:
+            if rpc.writelog('INFO', 'CouchPotato connected to drop off %s.' % nzb_name):
+                log.info('Successfully connected to NZBGet')
+            else:
+                log.info('Successfully connected to NZBGet, but unable to send a message')
+        except socket.error:
+            log.error('NZBGet is not responding. Please ensure that NZBGet is running and host setting is correct.')
+            return False
+        except xmlrpclib.ProtocolError, e:
+            if e.errcode == 401:
+                log.error('Password is incorrect.')
+            else:
+                log.error('Protocol Error: %s' % e)
+            return False
+
+        try:
+            if isfunction(data.get('download')):
+                file = data.get('download')()
+                if not file:
+                    log.error('Failed download file: %s' % nzb_name)
+                    return False
+            else:
+                log.info('Downloading: %s' % data.get('url'))
+                file = self.urlopen(data.get('url'))
+        except Exception, e:
+            log.error('Unable to get NZB file: %s' % e)
+            return False
+
+        if rpc.append(nzb_name, self.conf('category'), False, standard_b64encode(file.strip())):
+            log.info('NZB sent successfully to NZBGet')
+            return True
+        else:
+            log.error('NZBGet could not add %s to the queue.' % nzb_name)
+            return False