Added initial rtorrent downloader, currently testing, possibly has some bugs.

couchpotato/core/downloaders/rtorrent/__init__.py

@@ -0,0 +1,54 @@
+from .main import rTorrent
+
+def start():
+    return rTorrent()
+
+config = [{
+    'name': 'rtorrent',
+    'groups': [
+        {
+            'tab': 'downloaders',
+            'list': 'download_providers',
+            'name': 'rtorrent',
+            'label': 'rTorrent',
+            'description': '',
+            'wizard': True,
+            'options': [
+                {
+                    'name': 'enabled',
+                    'default': 0,
+                    'type': 'enabler',
+                    'radio_group': 'torrent',
+                },
+                {
+                    'name': 'url',
+                    'default': 'http://localhost:80/RPC2',
+                },
+                {
+                    'name': 'username',
+                },
+                {
+                    'name': 'password',
+                    'type': 'password',
+                },
+                {
+                    'name': 'label',
+                    'description': 'Label to add torrent as.',
+                },
+                {
+                    'name': 'paused',
+                    'type': 'bool',
+                    'default': False,
+                    'description': 'Add the torrent paused.',
+                },
+                {
+                    'name': 'manual',
+                    'default': 0,
+                    'type': 'bool',
+                    'advanced': True,
+                    'description': 'Disable this downloader for automated searches, but use it when I manually send a release.',
+                },
+            ],
+        }
+    ],
+}]

couchpotato/core/downloaders/rtorrent/main.py

@@ -0,0 +1,97 @@
+from base64 import b16encode, b32decode
+from bencode import bencode, bdecode
+from couchpotato.core.downloaders.base import Downloader, StatusList
+from couchpotato.core.helpers.encoding import isInt, ss
+from couchpotato.core.logger import CPLog
+from datetime import timedelta
+from hashlib import sha1
+from multipartpost import MultipartPostHandler
+import cookielib
+import httplib
+import json
+import re
+import time
+import urllib
+import urllib2
+from rtorrent import RTorrent
+
+
+log = CPLog(__name__)
+
+
+class rTorrent(Downloader):
+
+    type = ['torrent', 'torrent_magnet']
+    rtorrent_api = None
+
+    def get_conn(self):
+        return RTorrent(
+            self.conf('url'),
+            self.conf('username'),
+            self.conf('password')
+        )
+
+    def download(self, data, movie, filedata=None):
+        log.debug('Sending "%s" (%s) to rTorrent.', (data.get('name'), data.get('type')))
+
+        torrent_params = {}
+        if self.conf('label'):
+            torrent_params['label'] = self.conf('label')
+
+        if not filedata and data.get('type') == 'torrent':
+            log.error('Failed sending torrent, no data')
+            return False
+
+        if data.get('type') == 'torrent_magnet':
+            log.info('magnet torrents are not supported')
+            return False
+
+        info = bdecode(filedata)["info"]
+        torrent_hash = sha1(bencode(info)).hexdigest().upper()
+        torrent_filename = self.createFileName(data, filedata, movie)
+
+        # Convert base 32 to hex
+        if len(torrent_hash) == 32:
+            torrent_hash = b16encode(b32decode(torrent_hash))
+
+        # Send request to rTorrent
+        try:
+            if not self.rtorrent_api:
+                self.rtorrent_api = self.get_conn()
+
+            torrent = self.rtorrent_api.load_torrent(filedata, not self.conf('paused', default=0))
+
+            return self.downloadReturnId(torrent_hash)
+        except Exception, err:
+            log.error('Failed to send torrent to rTorrent: %s', err)
+            return False
+
+
+    def getAllDownloadStatus(self):
+
+        log.debug('Checking rTorrent download status.')
+
+        try:
+            if not self.rtorrent_api:
+                self.rtorrent_api = self.get_conn()
+
+            torrents = self.rtorrent_api.get_torrents()
+
+            statuses = StatusList(self)
+
+            for item in torrents:
+                statuses.append({
+                    'id': item.info_hash,
+                    'name': item.name,
+                    'status': 'completed' if item.complete else 'busy',
+                    'original_status': item.state,
+                    'timeleft': str(timedelta(seconds=float(item.left_bytes) / item.down_rate))
+                                    if item.down_rate > 0 else -1,
+                    'folder': ''
+                })
+
+            return statuses
+
+        except Exception, err:
+            log.error('Failed to send torrent to rTorrent: %s', err)
+            return False